80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package storenodes
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/multiformats/go-multiaddr"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
)
|
|
|
|
var (
|
|
communityID1 = types.HexBytes("community001")
|
|
communityID2 = types.HexBytes("community002")
|
|
)
|
|
|
|
func TestSyncSave(t *testing.T) {
|
|
db, close := setupTestDB(t, communityID1)
|
|
defer close()
|
|
|
|
maddr, err := multiaddr.NewMultiaddr("/dns4/test.net/tcp/30303/p2p/16Uiu2HAmMELCo218hncCtTvC2Dwbej3rbyHQcR8erXNnKGei7WPZ")
|
|
require.NoError(t, err)
|
|
|
|
snodes := []Storenode{
|
|
{
|
|
CommunityID: communityID1,
|
|
StorenodeID: "storenode001",
|
|
Name: "My Mailserver",
|
|
Address: maddr,
|
|
Fleet: "prod",
|
|
Version: 2,
|
|
},
|
|
}
|
|
|
|
// ========
|
|
// Save
|
|
|
|
err = db.syncSave(communityID1, snodes, 0)
|
|
require.NoError(t, err)
|
|
|
|
dbNodes, err := db.getByCommunityID(communityID1)
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, dbNodes, 1)
|
|
require.ElementsMatch(t, dbNodes, snodes)
|
|
|
|
// ========
|
|
// Update
|
|
|
|
updated := []Storenode{
|
|
{
|
|
CommunityID: communityID1,
|
|
StorenodeID: "storenode001",
|
|
Name: "My Mailserver 2",
|
|
Address: maddr,
|
|
Fleet: "prod",
|
|
Version: 2,
|
|
},
|
|
}
|
|
err = db.syncSave(communityID1, updated, 0)
|
|
require.NoError(t, err)
|
|
|
|
dbNodes, err = db.getByCommunityID(communityID1)
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, dbNodes, 1)
|
|
require.ElementsMatch(t, dbNodes, updated)
|
|
|
|
// ========
|
|
// Remove
|
|
|
|
err = db.syncSave(communityID1, []Storenode{}, 0)
|
|
require.NoError(t, err)
|
|
|
|
dbNodes, err = db.getByCommunityID(communityID1)
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, dbNodes, 0)
|
|
}
|