status-go/services/wallet/currency/currency_db_test.go

87 lines
1.6 KiB
Go

package currency
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/status-im/status-go/t/helpers"
"github.com/status-im/status-go/walletdatabase"
)
func setupTestCurrencyDB(t *testing.T) (*DB, func()) {
db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
require.NoError(t, err)
return NewCurrencyDB(db), func() {
require.NoError(t, db.Close())
}
}
func TestCurrencyFormats(t *testing.T) {
db, stop := setupTestCurrencyDB(t)
defer stop()
rst, err := db.GetCachedFormats()
require.NoError(t, err)
require.Empty(t, rst)
pr1 := FormatPerSymbol{
"A": {
Symbol: "A",
DisplayDecimals: 1,
StripTrailingZeroes: false,
},
"B": {
Symbol: "B",
DisplayDecimals: 2,
StripTrailingZeroes: true,
},
}
err = db.UpdateCachedFormats(pr1)
require.NoError(t, err)
rst, err = db.GetCachedFormats()
require.NoError(t, err)
require.Equal(t, rst, pr1)
pr2 := FormatPerSymbol{
"B": {
Symbol: "B",
DisplayDecimals: 3,
StripTrailingZeroes: true,
},
"C": {
Symbol: "C",
DisplayDecimals: 4,
StripTrailingZeroes: false,
},
}
err = db.UpdateCachedFormats(pr2)
require.NoError(t, err)
rst, err = db.GetCachedFormats()
require.NoError(t, err)
expected := FormatPerSymbol{
"A": {
Symbol: "A",
DisplayDecimals: 1,
StripTrailingZeroes: false,
},
"B": {
Symbol: "B",
DisplayDecimals: 3,
StripTrailingZeroes: true,
},
"C": {
Symbol: "C",
DisplayDecimals: 4,
StripTrailingZeroes: false,
},
}
require.Equal(t, rst, expected)
}