status-go/protocol/waku_builder_test.go

package protocol

import (
	"database/sql"

	"github.com/stretchr/testify/suite"
	"go.uber.org/zap"

	"github.com/status-im/status-go/appdatabase"
	gethbridge "github.com/status-im/status-go/eth-node/bridge/geth"
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/protocol/common/shard"
	"github.com/status-im/status-go/t/helpers"
	waku2 "github.com/status-im/status-go/wakuv2"
)

type testWakuV2Config struct {
	logger                 *zap.Logger
	enableStore            bool
	useShardAsDefaultTopic bool
	clusterID              uint16
}

func NewTestWakuV2(s *suite.Suite, cfg testWakuV2Config) *waku2.Waku {
	wakuConfig := &waku2.Config{
		UseShardAsDefaultTopic:   cfg.useShardAsDefaultTopic,
		ClusterID:                cfg.clusterID,
		LightClient:              false,
		EnablePeerExchangeServer: true,
		EnablePeerExchangeClient: false,
		EnableDiscV5:             true,
	}

	var db *sql.DB

	if cfg.enableStore {
		var err error
		db, err = helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
		s.Require().NoError(err)

		wakuConfig.EnableStore = true
		wakuConfig.StoreCapacity = 200
		wakuConfig.StoreSeconds = 200
	}

	wakuNode, err := waku2.New(
		"",
		"",
		wakuConfig,
		cfg.logger,
		db,
		nil,
		nil,
		nil)

	s.Require().NoError(err)

	err = wakuNode.Start()
	s.Require().NoError(err)

	return wakuNode
}

func CreateWakuV2Network(s *suite.Suite, parentLogger *zap.Logger, useShardAsDefaultTopic bool, nodeNames []string) []types.Waku {
	nodes := make([]*waku2.Waku, len(nodeNames))
	wrappers := make([]types.Waku, len(nodes))

	for i, name := range nodeNames {
		nodes[i] = NewTestWakuV2(s, testWakuV2Config{
			logger:                 parentLogger.Named("waku-" + name),
			enableStore:            false,
			useShardAsDefaultTopic: useShardAsDefaultTopic,
			clusterID:              shard.UndefinedShardValue, // FIXME: why it was 0 here?
		})
	}

	// Setup local network graph
	for i := 0; i < len(nodes); i++ {
		for j := 0; j < len(nodes); j++ {
			if i == j {
				continue
			}

			addrs := nodes[j].ListenAddresses()
			s.Require().Greater(len(addrs), 0)
			_, err := nodes[i].AddRelayPeer(addrs[0])
			s.Require().NoError(err)
			err = nodes[i].DialPeer(addrs[0])
			s.Require().NoError(err)
		}
	}
	for i, n := range nodes {
		wrappers[i] = gethbridge.NewGethWakuV2Wrapper(n)
	}
	return wrappers
}