85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
|
package cryptocompare_tests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/status-im/status-go/appdatabase"
|
||
|
|
"github.com/status-im/status-go/params"
|
||
|
|
mock_network "github.com/status-im/status-go/rpc/network/mock"
|
||
|
|
w_common "github.com/status-im/status-go/services/wallet/common"
|
||
|
|
"github.com/status-im/status-go/services/wallet/thirdparty/cryptocompare"
|
||
|
|
"github.com/status-im/status-go/services/wallet/token"
|
||
|
|
"github.com/status-im/status-go/t/helpers"
|
||
|
|
"github.com/status-im/status-go/walletdatabase"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"go.uber.org/mock/gomock"
|
||
|
|
)
|
||
|
|
|
||
|
|
func getTokenSymbols(t *testing.T) []string {
|
||
|
|
appDB, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
walletDB, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
networksList := []params.Network{
|
||
|
|
{
|
||
|
|
ChainID: w_common.EthereumMainnet,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ChainID: w_common.OptimismMainnet,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ChainID: w_common.ArbitrumMainnet,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
ptrNetworkList := make([]*params.Network, 0, len(networksList))
|
||
|
|
for i := range networksList {
|
||
|
|
ptrNetworkList = append(ptrNetworkList, &networksList[i])
|
||
|
|
}
|
||
|
|
|
||
|
|
ctrl := gomock.NewController(t)
|
||
|
|
defer ctrl.Finish()
|
||
|
|
networkManager := mock_network.NewMockManagerInterface(ctrl)
|
||
|
|
networkManager.EXPECT().Get(gomock.Any()).Return(ptrNetworkList, nil).AnyTimes()
|
||
|
|
networkManager.EXPECT().GetAll().Return(ptrNetworkList, nil).AnyTimes()
|
||
|
|
networkManager.EXPECT().GetConfiguredNetworks().Return(networksList).AnyTimes()
|
||
|
|
|
||
|
|
// Skeleton token store to get full list of tokens
|
||
|
|
tm := token.NewTokenManager(walletDB, nil, nil, networkManager, appDB, nil, nil, nil, nil, nil)
|
||
|
|
|
||
|
|
tokens, err := tm.GetAllTokens()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
symbolsMap := make(map[string]bool)
|
||
|
|
for _, token := range tokens {
|
||
|
|
symbolsMap[token.Symbol] = true
|
||
|
|
}
|
||
|
|
|
||
|
|
symbols := make([]string, 0, len(symbolsMap))
|
||
|
|
for symbol := range symbolsMap {
|
||
|
|
symbols = append(symbols, symbol)
|
||
|
|
}
|
||
|
|
|
||
|
|
return symbols
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFetchPrices(t *testing.T) {
|
||
|
|
symbols := getTokenSymbols(t)
|
||
|
|
|
||
|
|
stdClient := cryptocompare.NewClient()
|
||
|
|
_, err := stdClient.FetchPrices(symbols, []string{"USD"})
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFetchTokenMarketValues(t *testing.T) {
|
||
|
|
symbols := getTokenSymbols(t)
|
||
|
|
|
||
|
|
stdClient := cryptocompare.NewClient()
|
||
|
|
_, err := stdClient.FetchTokenMarketValues(symbols, "USD")
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|