83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package storenodes
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/multiformats/go-multiaddr"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/status-im/status-go/services/mailservers"
|
|
)
|
|
|
|
func TestSerialization(t *testing.T) {
|
|
maddr, err := multiaddr.NewMultiaddr("/dns4/test.net/tcp/30303/p2p/16Uiu2HAmMELCo218hncCtTvC2Dwbej3rbyHQcR8erXNnKGei7WPZ")
|
|
require.NoError(t, err)
|
|
snodes := Storenodes{
|
|
{
|
|
CommunityID: communityID1,
|
|
StorenodeID: "storenode001",
|
|
Name: "My Mailserver",
|
|
Address: maddr,
|
|
Fleet: "prod",
|
|
Version: 2,
|
|
},
|
|
}
|
|
|
|
snodesProtobuf := snodes.ToProtobuf()
|
|
|
|
snodes2 := FromProtobuf(snodesProtobuf, 0)
|
|
|
|
require.Equal(t, snodes[0].Address.String(), snodes2[0].Address.String())
|
|
}
|
|
|
|
func TestUpdateStorenodesInDB(t *testing.T) {
|
|
db, close := setupTestDB(t, communityID1, communityID2)
|
|
defer close()
|
|
|
|
maddr, err := multiaddr.NewMultiaddr("/dns4/test.net/tcp/30303/p2p/16Uiu2HAmMELCo218hncCtTvC2Dwbej3rbyHQcR8erXNnKGei7WPZ")
|
|
require.NoError(t, err)
|
|
|
|
csn := NewCommunityStorenodes(db, nil)
|
|
snodes1 := []Storenode{
|
|
{
|
|
CommunityID: communityID1,
|
|
StorenodeID: "storenode001",
|
|
Name: "My Mailserver",
|
|
Address: maddr,
|
|
Fleet: "prod",
|
|
Version: 2,
|
|
},
|
|
}
|
|
snodes2 := []Storenode{
|
|
{
|
|
CommunityID: communityID2,
|
|
StorenodeID: "storenode002",
|
|
Name: "My Mailserver",
|
|
Address: maddr,
|
|
Fleet: "prod",
|
|
Version: 2,
|
|
},
|
|
}
|
|
// populate db
|
|
err = csn.UpdateStorenodesInDB(communityID1, snodes1, 0)
|
|
require.NoError(t, err)
|
|
err = csn.UpdateStorenodesInDB(communityID2, snodes2, 0)
|
|
require.NoError(t, err)
|
|
|
|
// check if storenodes are loaded
|
|
ms1, err := csn.GetStorenodeByCommunityID(communityID1.String())
|
|
require.NoError(t, err)
|
|
matchStoreNode(t, snodes1[0], ms1)
|
|
|
|
ms2, err := csn.GetStorenodeByCommunityID(communityID2.String())
|
|
require.NoError(t, err)
|
|
matchStoreNode(t, snodes2[0], ms2)
|
|
}
|
|
|
|
func matchStoreNode(t *testing.T, sn Storenode, ms mailservers.Mailserver) {
|
|
require.Equal(t, sn.StorenodeID, ms.ID)
|
|
require.Equal(t, sn.Name, ms.Name)
|
|
require.Equal(t, sn.Address.String(), (*ms.Addr).String())
|
|
require.Equal(t, sn.Fleet, ms.Fleet)
|
|
}
|