woop/p2p/stream/protocols/sync/protocol_test.go

package sync

import (
	"context"
	"sync"
	"testing"
	"time"

	"github.com/libp2p/go-libp2p/core/discovery"
	libp2p_peer "github.com/libp2p/go-libp2p/core/peer"
)

func TestProtocol_Match(t *testing.T) {
	tests := []struct {
		targetID string
		exp      bool
	}{
		{"harmony/sync/unitest/0/1.0.1/1", true},
		{"harmony/sync/unitest/0/1.0.1/0", true},
		{"h123456", false},
		{"harmony/sync/unitest/0/0.9.9/1", false},
		{"harmony/epoch/unitest/0/1.0.1/1", false},
		{"harmony/sync/mainnet/0/1.0.1/1", false},
		{"harmony/sync/unitest/1/1.0.1/1", false},
	}

	for i, test := range tests {
		p := &Protocol{
			beaconNode: true,
			config: Config{
				Network: "unitest",
				ShardID: 0,
			},
		}

		res := p.Match(test.targetID)

		if res != test.exp {
			t.Errorf("Test %v: unexpected result %v / %v", i, res, test.exp)
		}
	}
}

func TestProtocol_advertiseLoop(t *testing.T) {
	disc := newTestDiscovery(100 * time.Millisecond)
	p := &Protocol{
		disc:   disc,
		closeC: make(chan struct{}),
	}

	go p.advertiseLoop()

	time.Sleep(150 * time.Millisecond)
	close(p.closeC)

	advCnt := disc.Extract()
	if len(advCnt) != len(p.supportedVersions()) {
		t.Errorf("unexpected advertise topic count: %v / %v", len(advCnt),
			len(p.supportedVersions()))
	}
	for _, cnt := range advCnt {
		if cnt < 1 {
			t.Errorf("unexpected discovery count: %v", cnt)
		}
	}
}

type testDiscovery struct {
	advCnt map[string]int
	sleep  time.Duration
	mu     sync.Mutex
}

func newTestDiscovery(discInterval time.Duration) *testDiscovery {
	return &testDiscovery{
		advCnt: make(map[string]int),
		sleep:  discInterval,
	}
}

func (disc *testDiscovery) Start() error {
	return nil
}

func (disc *testDiscovery) Close() error {
	return nil
}

func (disc *testDiscovery) Advertise(ctx context.Context, ns string) (time.Duration, error) {
	disc.mu.Lock()
	defer disc.mu.Unlock()
	disc.advCnt[ns]++
	return disc.sleep, nil
}

func (disc *testDiscovery) Extract() map[string]int {
	disc.mu.Lock()
	defer disc.mu.Unlock()
	var out map[string]int
	out, disc.advCnt = disc.advCnt, make(map[string]int)
	return out
}

func (disc *testDiscovery) FindPeers(ctx context.Context, ns string, peerLimit int) (<-chan libp2p_peer.AddrInfo, error) {
	return nil, nil
}

func (disc *testDiscovery) GetRawDiscovery() discovery.Discovery {
	return nil
}