Compare commits
No commits in common. "2ace3606bc4045998c4ad0524c595d9b386983bd" and "c407dd1702ffc229b9b68a3e66ecc831f48b5383" have entirely different histories.
2ace3606bc
...
c407dd1702
@ -1,15 +1,10 @@
|
|||||||
package store
|
package store
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
visedb "git.defalsify.org/vise.git/db"
|
|
||||||
memdb "git.defalsify.org/vise.git/db/mem"
|
|
||||||
storedb "git.grassecon.net/grassrootseconomics/sarafu-vise/store/db"
|
storedb "git.grassecon.net/grassrootseconomics/sarafu-vise/store/db"
|
||||||
"github.com/alecthomas/assert/v2"
|
"github.com/alecthomas/assert/v2"
|
||||||
dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestReadSwapData(t *testing.T) {
|
func TestReadSwapData(t *testing.T) {
|
||||||
@ -93,77 +88,3 @@ func TestReadSwapPreviewData(t *testing.T) {
|
|||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.Equal(t, expectedResult, data)
|
assert.Equal(t, expectedResult, data)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestGetSwapFromVoucherData(t *testing.T) {
|
|
||||||
ctx := context.Background()
|
|
||||||
|
|
||||||
db := memdb.NewMemDb()
|
|
||||||
err := db.Connect(ctx, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
prefix := storedb.ToBytes(visedb.DATATYPE_USERDATA)
|
|
||||||
spdb := storedb.NewSubPrefixDb(db, prefix)
|
|
||||||
|
|
||||||
// Test pool swap data
|
|
||||||
mockData := map[storedb.DataTyp][]byte{
|
|
||||||
storedb.DATA_POOL_FROM_SYMBOLS: []byte("1:AMANI\n2:AMUA"),
|
|
||||||
storedb.DATA_POOL_FROM_BALANCES: []byte("1:\n2:"),
|
|
||||||
storedb.DATA_POOL_FROM_DECIMALS: []byte("1:6\n2:4"),
|
|
||||||
storedb.DATA_POOL_FROM_ADDRESSES: []byte("1:0xc7B78Ac9ACB9E025C8234621FC515bC58179dEAe\n2:0xF0C3C7581b8b96B59a97daEc8Bd48247cE078674"),
|
|
||||||
}
|
|
||||||
|
|
||||||
// Put the data
|
|
||||||
for key, value := range mockData {
|
|
||||||
err = spdb.Put(ctx, []byte(storedb.ToBytes(key)), []byte(value))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
result, err := GetSwapFromVoucherData(ctx, spdb, "1")
|
|
||||||
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, "AMANI", result.TokenSymbol)
|
|
||||||
assert.Equal(t, "", result.Balance)
|
|
||||||
assert.Equal(t, "6", result.TokenDecimals)
|
|
||||||
assert.Equal(t, "0xc7B78Ac9ACB9E025C8234621FC515bC58179dEAe", result.ContractAddress)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpdateSwapFromVoucherData(t *testing.T) {
|
|
||||||
ctx, store := InitializeTestDb(t)
|
|
||||||
sessionId := "session123"
|
|
||||||
|
|
||||||
// New swap from voucher data
|
|
||||||
newData := &dataserviceapi.TokenHoldings{
|
|
||||||
TokenSymbol: "AMANI",
|
|
||||||
TokenDecimals: "6",
|
|
||||||
ContractAddress: "0xc7B78Ac9ACB9E025C8234621FC515bC58179dEAe",
|
|
||||||
}
|
|
||||||
|
|
||||||
// Old temporary data
|
|
||||||
tempData := &dataserviceapi.TokenHoldings{
|
|
||||||
TokenSymbol: "OLD",
|
|
||||||
TokenDecimals: "8",
|
|
||||||
ContractAddress: "0xold",
|
|
||||||
}
|
|
||||||
require.NoError(t, StoreTemporaryVoucher(ctx, store, sessionId, tempData))
|
|
||||||
|
|
||||||
// Execute update
|
|
||||||
err := UpdateSwapFromVoucherData(ctx, store, sessionId, newData)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
// Verify active swap from data was stored correctly
|
|
||||||
activeEntries := map[storedb.DataTyp][]byte{
|
|
||||||
storedb.DATA_ACTIVE_SWAP_FROM_SYM: []byte(newData.TokenSymbol),
|
|
||||||
storedb.DATA_ACTIVE_SWAP_FROM_DECIMAL: []byte(newData.TokenDecimals),
|
|
||||||
storedb.DATA_ACTIVE_SWAP_FROM_ADDRESS: []byte(newData.ContractAddress),
|
|
||||||
}
|
|
||||||
|
|
||||||
for key, expectedValue := range activeEntries {
|
|
||||||
storedValue, err := store.ReadEntry(ctx, sessionId, key)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, expectedValue, storedValue, "Active swap from data mismatch for key %v", key)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user