status-go/protocol/storenodes/storenodes_test.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)
}