mirror of
https://github.com/status-im/status-go.git
synced 2025-02-23 20:28:27 +00:00
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package connector
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/mock/gomock"
|
|
|
|
"github.com/status-im/status-go/appdatabase"
|
|
"github.com/status-im/status-go/multiaccounts/settings"
|
|
"github.com/status-im/status-go/params"
|
|
mock_rpcclient "github.com/status-im/status-go/rpc/mock/client"
|
|
"github.com/status-im/status-go/rpc/network"
|
|
network_testutil "github.com/status-im/status-go/rpc/network/testutil"
|
|
walletCommon "github.com/status-im/status-go/services/wallet/common"
|
|
"github.com/status-im/status-go/t/helpers"
|
|
"github.com/status-im/status-go/walletdatabase"
|
|
)
|
|
|
|
type testState struct {
|
|
ctx context.Context
|
|
db *sql.DB
|
|
walletDb *sql.DB
|
|
mockCtrl *gomock.Controller
|
|
rpcClient *mock_rpcclient.MockClientInterface
|
|
service *Service
|
|
api *API
|
|
}
|
|
|
|
func createDB(t *testing.T) (*sql.DB, func()) {
|
|
db, cleanup, err := helpers.SetupTestSQLDB(appdatabase.DbInitializer{}, "browser-connect-tests-")
|
|
require.NoError(t, err)
|
|
return db, func() { require.NoError(t, cleanup()) }
|
|
}
|
|
|
|
func createWalletDB(t *testing.T) (db *sql.DB, close func()) {
|
|
db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
|
|
require.NoError(t, err)
|
|
return db, func() {
|
|
require.NoError(t, db.Close())
|
|
}
|
|
}
|
|
|
|
func setupTests(t *testing.T) (state testState, close func()) {
|
|
state.ctx = context.Background()
|
|
|
|
var (
|
|
closeDb func()
|
|
closeWalletDb func()
|
|
)
|
|
state.db, closeDb = createDB(t)
|
|
state.walletDb, closeWalletDb = createWalletDB(t)
|
|
|
|
config := params.NodeConfig{
|
|
NetworkID: 10,
|
|
DataDir: "test",
|
|
}
|
|
networks := json.RawMessage("{}")
|
|
settingsObj := settings.Settings{
|
|
Networks: &networks,
|
|
}
|
|
|
|
settDb, err := settings.MakeNewDB(state.db)
|
|
require.NoError(t, err)
|
|
err = settDb.CreateSettings(settingsObj, config)
|
|
require.NoError(t, err)
|
|
|
|
state.mockCtrl = gomock.NewController(t)
|
|
state.rpcClient = mock_rpcclient.NewMockClientInterface(state.mockCtrl)
|
|
|
|
networkManager := network.NewManager(state.db, nil, nil, nil)
|
|
require.NotNil(t, networkManager)
|
|
|
|
initNetworks := []params.Network{
|
|
*network_testutil.CreateNetwork(walletCommon.EthereumMainnet, "Ethereum Mainnet", []params.RpcProvider{
|
|
network_testutil.CreateProvider(walletCommon.EthereumMainnet, "Infura Mainnet", params.EmbeddedProxyProviderType, true, "https://mainnet.infura.io"),
|
|
}),
|
|
*network_testutil.CreateNetwork(walletCommon.OptimismMainnet, "Optimism Mainnet", []params.RpcProvider{
|
|
network_testutil.CreateProvider(walletCommon.OptimismMainnet, "Optimism Mainnet", params.EmbeddedProxyProviderType, true, "https://mainnet.optimism.io"),
|
|
}),
|
|
}
|
|
err = networkManager.InitEmbeddedNetworks(initNetworks)
|
|
require.NoError(t, err)
|
|
|
|
state.rpcClient.EXPECT().GetNetworkManager().AnyTimes().Return(networkManager)
|
|
|
|
state.service = NewService(state.walletDb, state.rpcClient, state.rpcClient.GetNetworkManager())
|
|
|
|
state.api = NewAPI(state.service)
|
|
|
|
return state, func() {
|
|
closeDb()
|
|
closeWalletDb()
|
|
}
|
|
}
|