323 lines
11 KiB
Go
323 lines
11 KiB
Go
|
package appdatabase
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/rand"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"math"
|
||
|
|
"math/big"
|
||
|
|
"os"
|
||
|
|
"sort"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/p2p/discv5"
|
||
|
|
|
||
|
|
"github.com/status-im/status-go/eth-node/crypto"
|
||
|
|
"github.com/status-im/status-go/nodecfg"
|
||
|
|
"github.com/status-im/status-go/params"
|
||
|
|
"github.com/status-im/status-go/protocol/pushnotificationserver"
|
||
|
|
"github.com/status-im/status-go/sqlite"
|
||
|
|
)
|
||
|
|
|
||
|
|
func setupTestDB(t *testing.T) (*sql.DB, func()) {
|
||
|
|
tmpfile, err := ioutil.TempFile("", "settings-tests-")
|
||
|
|
require.NoError(t, err)
|
||
|
|
db, err := InitializeDB(tmpfile.Name(), "settings-tests")
|
||
|
|
require.NoError(t, err)
|
||
|
|
return db, func() {
|
||
|
|
require.NoError(t, db.Close())
|
||
|
|
require.NoError(t, os.Remove(tmpfile.Name()))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetNodeConfig(t *testing.T) {
|
||
|
|
db, stop := setupTestDB(t)
|
||
|
|
defer stop()
|
||
|
|
|
||
|
|
nodeConfig := randomNodeConfig()
|
||
|
|
require.NoError(t, nodecfg.SaveNodeConfig(db, nodeConfig))
|
||
|
|
|
||
|
|
dbNodeConfig, err := nodecfg.GetNodeConfig(db)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, nodeConfig, dbNodeConfig)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSaveNodeConfig(t *testing.T) {
|
||
|
|
db, stop := setupTestDB(t)
|
||
|
|
defer stop()
|
||
|
|
|
||
|
|
require.NoError(t, nodecfg.SaveNodeConfig(db, randomNodeConfig()))
|
||
|
|
|
||
|
|
newNodeConfig := randomNodeConfig()
|
||
|
|
require.NoError(t, nodecfg.SaveNodeConfig(db, newNodeConfig))
|
||
|
|
|
||
|
|
dbNodeConfig, err := nodecfg.GetNodeConfig(db)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, *newNodeConfig, *dbNodeConfig)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMigrateNodeConfig(t *testing.T) {
|
||
|
|
db, stop := setupTestDB(t)
|
||
|
|
defer stop()
|
||
|
|
|
||
|
|
nodeConfig := ¶ms.NodeConfig{
|
||
|
|
NetworkID: uint64(randomInt(10)),
|
||
|
|
DataDir: randomString(),
|
||
|
|
ListenAddr: randomString(),
|
||
|
|
Rendezvous: true,
|
||
|
|
ShhextConfig: params.ShhextConfig{
|
||
|
|
PFSEnabled: true,
|
||
|
|
},
|
||
|
|
WakuV2Config: params.WakuV2Config{
|
||
|
|
CustomNodes: make(map[string]string),
|
||
|
|
},
|
||
|
|
RequireTopics: make(map[discv5.Topic]params.Limits),
|
||
|
|
}
|
||
|
|
value := &sqlite.JSONBlob{Data: nodeConfig}
|
||
|
|
|
||
|
|
fmt.Println(value)
|
||
|
|
update, err := db.Prepare("INSERT INTO settings(node_config, address, current_network, dapps_address, installation_id, key_uid, name, networks, photo_path, public_key, signing_phrase, wallet_root_address, synthetic_id) VALUES(?, '', 0, '', '', '', '', '', '', '', '', '', 'id')")
|
||
|
|
require.NoError(t, err)
|
||
|
|
b, err := update.Exec(value)
|
||
|
|
require.NoError(t, err)
|
||
|
|
fmt.Println(b.RowsAffected())
|
||
|
|
// Forcing deletion of node_config data to be able to run migration
|
||
|
|
_, err = db.Exec("DELETE FROM node_config WHERE synthetic_id = 'id'")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
err = nodecfg.MigrateNodeConfig(db)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
dbNodeConfig, err := nodecfg.GetNodeConfig(db)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, nodeConfig, dbNodeConfig)
|
||
|
|
|
||
|
|
// node_config column should be empty
|
||
|
|
var result string
|
||
|
|
err = db.QueryRow("SELECT COALESCE(NULL, 'empty')").Scan(&result)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "empty", result)
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomString() string {
|
||
|
|
b := make([]byte, 10)
|
||
|
|
_, _ = rand.Read(b)
|
||
|
|
return fmt.Sprintf("%x", b)[:10]
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomBool() bool {
|
||
|
|
return randomInt(2) == 1
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomInt(max int64) int {
|
||
|
|
r, _ := rand.Int(rand.Reader, big.NewInt(max))
|
||
|
|
return int(r.Int64())
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomFloat(max int64) float64 {
|
||
|
|
r, _ := rand.Int(rand.Reader, big.NewInt(max))
|
||
|
|
return float64(r.Int64()) / (1 << 63)
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomStringSlice() []string {
|
||
|
|
m := randomInt(7)
|
||
|
|
var result []string
|
||
|
|
for i := 0; i < m; i++ {
|
||
|
|
result = append(result, randomString())
|
||
|
|
}
|
||
|
|
sort.Strings(result)
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomTopicSlice() []discv5.Topic {
|
||
|
|
randomValues := randomStringSlice()
|
||
|
|
var result []discv5.Topic
|
||
|
|
for _, v := range randomValues {
|
||
|
|
result = append(result, discv5.Topic(v))
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomTopicLimits() map[discv5.Topic]params.Limits {
|
||
|
|
result := make(map[discv5.Topic]params.Limits)
|
||
|
|
m := randomInt(7) + 1
|
||
|
|
for i := 0; i < m; i++ {
|
||
|
|
result[discv5.Topic(fmt.Sprint(i))] = params.Limits{Min: randomInt(2), Max: randomInt(10)}
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomCustomNodes() map[string]string {
|
||
|
|
result := make(map[string]string)
|
||
|
|
m := randomInt(7)
|
||
|
|
for i := 0; i < m; i++ {
|
||
|
|
result[randomString()] = randomString()
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomNetworkSlice() []params.Network {
|
||
|
|
m := randomInt(7) + 1
|
||
|
|
var result []params.Network
|
||
|
|
for i := 0; i < m; i++ {
|
||
|
|
n := params.Network{
|
||
|
|
ChainID: uint64(i),
|
||
|
|
ChainName: randomString(),
|
||
|
|
RPCURL: randomString(),
|
||
|
|
BlockExplorerURL: randomString(),
|
||
|
|
IconURL: randomString(),
|
||
|
|
NativeCurrencyName: randomString(),
|
||
|
|
NativeCurrencySymbol: randomString(),
|
||
|
|
NativeCurrencyDecimals: uint64(int64(randomInt(math.MaxInt64))),
|
||
|
|
IsTest: randomBool(),
|
||
|
|
Layer: uint64(int64(randomInt(math.MaxInt64))),
|
||
|
|
Enabled: randomBool(),
|
||
|
|
}
|
||
|
|
result = append(result, n)
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func randomNodeConfig() *params.NodeConfig {
|
||
|
|
privK, _ := crypto.GenerateKey()
|
||
|
|
|
||
|
|
return ¶ms.NodeConfig{
|
||
|
|
NetworkID: uint64(int64(randomInt(math.MaxInt64))),
|
||
|
|
DataDir: randomString(),
|
||
|
|
KeyStoreDir: randomString(),
|
||
|
|
NodeKey: randomString(),
|
||
|
|
NoDiscovery: randomBool(),
|
||
|
|
Rendezvous: randomBool(),
|
||
|
|
ListenAddr: randomString(),
|
||
|
|
AdvertiseAddr: randomString(),
|
||
|
|
Name: randomString(),
|
||
|
|
Version: randomString(),
|
||
|
|
APIModules: randomString(),
|
||
|
|
TLSEnabled: randomBool(),
|
||
|
|
MaxPeers: randomInt(math.MaxInt64),
|
||
|
|
MaxPendingPeers: randomInt(math.MaxInt64),
|
||
|
|
EnableStatusService: randomBool(),
|
||
|
|
EnableNTPSync: randomBool(),
|
||
|
|
BridgeConfig: params.BridgeConfig{Enabled: randomBool()},
|
||
|
|
WalletConfig: params.WalletConfig{Enabled: randomBool()},
|
||
|
|
LocalNotificationsConfig: params.LocalNotificationsConfig{Enabled: randomBool()},
|
||
|
|
BrowsersConfig: params.BrowsersConfig{Enabled: randomBool()},
|
||
|
|
PermissionsConfig: params.PermissionsConfig{Enabled: randomBool()},
|
||
|
|
MailserversConfig: params.MailserversConfig{Enabled: randomBool()},
|
||
|
|
Web3ProviderConfig: params.Web3ProviderConfig{Enabled: randomBool()},
|
||
|
|
SwarmConfig: params.SwarmConfig{Enabled: randomBool()},
|
||
|
|
MailServerRegistryAddress: randomString(),
|
||
|
|
HTTPEnabled: randomBool(),
|
||
|
|
HTTPHost: randomString(),
|
||
|
|
HTTPPort: randomInt(math.MaxInt64),
|
||
|
|
HTTPVirtualHosts: randomStringSlice(),
|
||
|
|
HTTPCors: randomStringSlice(),
|
||
|
|
IPCEnabled: randomBool(),
|
||
|
|
IPCFile: randomString(),
|
||
|
|
LogEnabled: randomBool(),
|
||
|
|
LogMobileSystem: randomBool(),
|
||
|
|
LogDir: randomString(),
|
||
|
|
LogFile: randomString(),
|
||
|
|
LogLevel: randomString(),
|
||
|
|
LogMaxBackups: randomInt(math.MaxInt64),
|
||
|
|
LogMaxSize: randomInt(math.MaxInt64),
|
||
|
|
LogCompressRotated: randomBool(),
|
||
|
|
LogToStderr: randomBool(),
|
||
|
|
UpstreamConfig: params.UpstreamRPCConfig{Enabled: randomBool(), URL: randomString()},
|
||
|
|
Networks: randomNetworkSlice(),
|
||
|
|
ClusterConfig: params.ClusterConfig{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
Fleet: randomString(),
|
||
|
|
StaticNodes: randomStringSlice(),
|
||
|
|
BootNodes: randomStringSlice(),
|
||
|
|
},
|
||
|
|
LightEthConfig: params.LightEthConfig{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
DatabaseCache: randomInt(math.MaxInt64),
|
||
|
|
TrustedNodes: randomStringSlice(),
|
||
|
|
MinTrustedFraction: randomInt(math.MaxInt64),
|
||
|
|
},
|
||
|
|
RegisterTopics: randomTopicSlice(),
|
||
|
|
RequireTopics: randomTopicLimits(),
|
||
|
|
PushNotificationServerConfig: pushnotificationserver.Config{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
GorushURL: randomString(),
|
||
|
|
Identity: privK,
|
||
|
|
},
|
||
|
|
ShhextConfig: params.ShhextConfig{
|
||
|
|
PFSEnabled: randomBool(),
|
||
|
|
BackupDisabledDataDir: randomString(),
|
||
|
|
InstallationID: randomString(),
|
||
|
|
MailServerConfirmations: randomBool(),
|
||
|
|
EnableConnectionManager: randomBool(),
|
||
|
|
EnableLastUsedMonitor: randomBool(),
|
||
|
|
ConnectionTarget: randomInt(math.MaxInt64),
|
||
|
|
RequestsDelay: time.Duration(randomInt(math.MaxInt64)),
|
||
|
|
MaxServerFailures: randomInt(math.MaxInt64),
|
||
|
|
MaxMessageDeliveryAttempts: randomInt(math.MaxInt64),
|
||
|
|
WhisperCacheDir: randomString(),
|
||
|
|
DisableGenericDiscoveryTopic: randomBool(),
|
||
|
|
SendV1Messages: randomBool(),
|
||
|
|
DataSyncEnabled: randomBool(),
|
||
|
|
VerifyTransactionURL: randomString(),
|
||
|
|
VerifyENSURL: randomString(),
|
||
|
|
VerifyENSContractAddress: randomString(),
|
||
|
|
VerifyTransactionChainID: int64(randomInt(math.MaxInt64)),
|
||
|
|
AnonMetricsSendID: randomString(),
|
||
|
|
AnonMetricsServerEnabled: randomBool(),
|
||
|
|
AnonMetricsServerPostgresURI: randomString(),
|
||
|
|
BandwidthStatsEnabled: randomBool(),
|
||
|
|
},
|
||
|
|
WakuV2Config: params.WakuV2Config{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
Host: randomString(),
|
||
|
|
Port: randomInt(math.MaxInt64),
|
||
|
|
KeepAliveInterval: randomInt(math.MaxInt64),
|
||
|
|
LightClient: randomBool(),
|
||
|
|
FullNode: randomBool(),
|
||
|
|
DiscoveryLimit: randomInt(math.MaxInt64),
|
||
|
|
PersistPeers: randomBool(),
|
||
|
|
DataDir: randomString(),
|
||
|
|
MaxMessageSize: uint32(randomInt(math.MaxInt64)),
|
||
|
|
EnableConfirmations: randomBool(),
|
||
|
|
CustomNodes: randomCustomNodes(),
|
||
|
|
PeerExchange: randomBool(),
|
||
|
|
EnableDiscV5: randomBool(),
|
||
|
|
UDPPort: randomInt(math.MaxInt64),
|
||
|
|
AutoUpdate: randomBool(),
|
||
|
|
},
|
||
|
|
WakuConfig: params.WakuConfig{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
LightClient: randomBool(),
|
||
|
|
FullNode: randomBool(),
|
||
|
|
EnableMailServer: randomBool(),
|
||
|
|
DataDir: randomString(),
|
||
|
|
MinimumPoW: randomFloat(math.MaxInt64),
|
||
|
|
MailServerPassword: randomString(),
|
||
|
|
MailServerRateLimit: randomInt(math.MaxInt64),
|
||
|
|
MailServerDataRetention: randomInt(math.MaxInt64),
|
||
|
|
TTL: randomInt(math.MaxInt64),
|
||
|
|
MaxMessageSize: uint32(randomInt(math.MaxInt64)),
|
||
|
|
DatabaseConfig: params.DatabaseConfig{
|
||
|
|
PGConfig: params.PGConfig{
|
||
|
|
Enabled: randomBool(),
|
||
|
|
URI: randomString(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
EnableRateLimiter: randomBool(),
|
||
|
|
PacketRateLimitIP: int64(randomInt(math.MaxInt64)),
|
||
|
|
PacketRateLimitPeerID: int64(randomInt(math.MaxInt64)),
|
||
|
|
BytesRateLimitIP: int64(randomInt(math.MaxInt64)),
|
||
|
|
BytesRateLimitPeerID: int64(randomInt(math.MaxInt64)),
|
||
|
|
RateLimitTolerance: int64(randomInt(math.MaxInt64)),
|
||
|
|
BloomFilterMode: randomBool(),
|
||
|
|
SoftBlacklistedPeerIDs: randomStringSlice(),
|
||
|
|
EnableConfirmations: randomBool(),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|