any-sync/commonspace/sync/sync_test.go

package sync

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/anyproto/any-sync/app"
	"github.com/anyproto/any-sync/commonspace/sync/synctest"
	"github.com/anyproto/any-sync/net/rpc/rpctest"
	"github.com/anyproto/any-sync/net/streampool"
)

var ctx = context.Background()

func TestNewSyncService(t *testing.T) {
	peerPool := synctest.NewPeerGlobalPool([]string{"first", "second", "third"})
	peerPool.MakePeers()
	var (
		firstApp  = newFixture(t, "first", counterFixtureParams{peerPool: peerPool, start: 0, delta: 3})
		secondApp = newFixture(t, "second", counterFixtureParams{peerPool: peerPool, start: 1, delta: 3})
		thirdApp  = newFixture(t, "third", counterFixtureParams{peerPool: peerPool, start: 2, delta: 3})
	)
	require.NoError(t, firstApp.a.Start(ctx))
	require.NoError(t, secondApp.a.Start(ctx))
	require.NoError(t, thirdApp.a.Start(ctx))
	time.Sleep(100 * time.Second)
	firstApp.a.Close(context.Background())
	secondApp.a.Close(context.Background())
}

type counterFixture struct {
	a *app.App
}

type counterFixtureParams struct {
	peerPool *synctest.PeerGlobalPool
	start    int32
	delta    int32
}

func newFixture(t *testing.T, peerId string, params counterFixtureParams) *counterFixture {
	a := &app.App{}
	a.Register(params.peerPool).
		Register(synctest.NewConfig()).
		Register(rpctest.NewTestServer()).
		Register(synctest.NewCounterStreamOpener()).
		Register(synctest.NewPeerProvider(peerId)).
		Register(synctest.NewCounter(params.start, params.delta)).
		Register(streampool.NewStreamPool()).
		Register(synctest.NewCounterSyncHandler()).
		Register(NewSyncService()).
		Register(synctest.NewCounterGenerator()).
		Register(synctest.NewRpcServer())
	return &counterFixture{a: a}
}