chore_: fix test

Author: kaichaosun

kvstore/database_test.go

@@ -14,13 +14,27 @@ type DbInitializer struct {
 }
 
 func (a DbInitializer) Initialize(path, password string, kdfIterationsNumber int) (*sql.DB, error) {
-	return sqlite.OpenDB(path, password, kdfIterationsNumber)
+	db, err := sqlite.OpenDB(path, password, kdfIterationsNumber)
+	if err != nil {
+		return nil, err
+	}
+
+	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS kv_store (
+		key TEXT PRIMARY KEY,
+		value BLOB
+	);`)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return db, nil
 }
 
 func setupTestDB(t *testing.T) (*Database, func()) {
-	db, cleanup, err := helpers.SetupTestSQLDB(DbInitializer{}, "kvstore-tests")
+	db, err := helpers.SetupTestMemorySQLDB(DbInitializer{})
 	require.NoError(t, err)
-	return NewDB(db), func() { require.NoError(t, cleanup()) }
+	return NewDB(db), func() { require.NoError(t, db.Close()) }
 }
 
 func TestSetGetDelete(t *testing.T) {

protocol/messenger_base_test.go

@@ -8,10 +8,12 @@ import (
 
 	"go.uber.org/zap"
 
+	"github.com/status-im/status-go/appdatabase"
 	"github.com/status-im/status-go/eth-node/crypto"
 	"github.com/status-im/status-go/multiaccounts/settings"
 	"github.com/status-im/status-go/params"
 	"github.com/status-im/status-go/protocol/tt"
+	"github.com/status-im/status-go/t/helpers"
 	"github.com/status-im/status-go/wakuv2"
 
 	wakutypes "github.com/status-im/status-go/waku/types"
@@ -85,11 +87,15 @@ func newMessengerWithKey(shh wakutypes.Waku, privateKey *ecdsa.PrivateKey, logge
 }
 
 func newTestWakuNode(logger *zap.Logger) (wakutypes.Waku, error) {
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	if err != nil {
+		return nil, err
+	}
 	return wakuv2.New(
 		nil,
 		&wakuv2.DefaultConfig,
 		logger,
-		nil,
+		db,
 		nil,
 		func([]byte, peer.AddrInfo, error) {},
 		nil,

server/pairing/sync_device_test.go

@@ -20,6 +20,7 @@ import (
 	"github.com/stretchr/testify/suite"
 
 	"github.com/status-im/status-go/api"
+	"github.com/status-im/status-go/appdatabase"
 	"github.com/status-im/status-go/common/dbsetup"
 	"github.com/status-im/status-go/eth-node/crypto"
 	"github.com/status-im/status-go/eth-node/types"
@@ -33,6 +34,7 @@ import (
 	"github.com/status-im/status-go/protocol/tt"
 	accservice "github.com/status-im/status-go/services/accounts"
 	"github.com/status-im/status-go/services/browsers"
+	"github.com/status-im/status-go/t/helpers"
 	"github.com/status-im/status-go/wakuv2"
 )
 
@@ -85,8 +87,10 @@ func (s *SyncDeviceSuite) SetupTest() {
 		DefaultShardPubsubTopic:                wakuv2.DefaultShardPubsubTopic(),
 		EnableStoreConfirmationForMessagesSent: false,
 	}
-	var err error
-	s.pxBootNode, err = wakuv2.New(nil, exchangeNodeConfig, s.logger.Named("pxServerNode"), nil, nil, nil, nil)
+
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	s.Require().NoError(err)
+	s.pxBootNode, err = wakuv2.New(nil, exchangeNodeConfig, s.logger.Named("pxServerNode"), db, nil, nil, nil)
 	s.Require().NoError(err)
 	s.Require().NoError(s.pxBootNode.Start())
 

wakuv2/waku_test.go

@@ -66,7 +66,9 @@ func TestDiscoveryV5(t *testing.T) {
 	setDefaultConfig(config, false)
 	config.DiscV5BootstrapNodes = []string{testStoreENRBootstrap}
 	config.DiscoveryLimit = 20
-	w, err := New(nil, config, nil, nil, nil, nil, nil)
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	require.NoError(t, err)
+	w, err := New(nil, config, nil, db, nil, nil, nil)
 	require.NoError(t, err)
 
 	require.NoError(t, w.Start())
@@ -92,7 +94,9 @@ func TestRestartDiscoveryV5(t *testing.T) {
 	config.DiscoveryLimit = 20
 	config.UDPPort = 10002
 	config.ClusterID = 16
-	w, err := New(nil, config, nil, nil, nil, nil, nil)
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	require.NoError(t, err)
+	w, err := New(nil, config, nil, db, nil, nil, nil)
 	require.NoError(t, err)
 
 	require.NoError(t, w.Start())
@@ -135,8 +139,10 @@ func TestRelayPeers(t *testing.T) {
 	config := &Config{
 		EnableMissingMessageVerification: true,
 	}
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	require.NoError(t, err)
 	setDefaultConfig(config, false)
-	w, err := New(nil, config, nil, nil, nil, nil, nil)
+	w, err := New(nil, config, nil, db, nil, nil, nil)
 	require.NoError(t, err)
 	require.NoError(t, w.Start())
 	_, err = w.RelayPeersByTopic(config.DefaultShardPubsubTopic)
@@ -146,7 +152,7 @@ func TestRelayPeers(t *testing.T) {
 	config = &Config{}
 	config.ClusterID = 16
 	config.LightClient = true
-	w, err = New(nil, config, nil, nil, nil, nil, nil)
+	w, err = New(nil, config, nil, db, nil, nil, nil)
 	require.NoError(t, err)
 	require.NoError(t, w.Start())
 	_, err = w.RelayPeersByTopic(config.DefaultShardPubsubTopic)
@@ -194,7 +200,9 @@ func TestBasicWakuV2(t *testing.T) {
 	config.DiscV5BootstrapNodes = []string{enrTreeAddress}
 	config.DiscoveryLimit = 20
 	config.WakuNodes = []string{enrTreeAddress}
-	w, err := New(nil, config, nil, nil, nil, nil, nil)
+	db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
+	require.NoError(t, err)
+	w, err := New(nil, config, nil, db, nil, nil, nil)
 	require.NoError(t, err)
 	require.NoError(t, w.Start())