package server

import (
	"testing"
	"time"

	"go.miragespace.co/specter/spec/protocol"
	"go.miragespace.co/specter/spec/tun"

	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"
)

func TestIdentitiesRoutine(t *testing.T) {
	as := require.New(t)

	_, node, clientT, chordT, serv := getFixture(t, as)
	_, cht, tn := getIdentities()

	ctx := t.Context()

	chordT.On("Identity").Return(cht)
	clientT.On("Identity").Return(tn)

	// On start up -> publish
	node.On("Put", mock.Anything, mock.MatchedBy(func(k []byte) bool {
		exp := [][]byte{
			[]byte(tun.DestinationByChordKey(cht)),
			[]byte(tun.DestinationByTunnelKey(tn)),
		}
		return assertBytes(k, exp...)
	}), mock.MatchedBy(func(v []byte) bool {
		pair := &protocol.TunnelDestination{}
		err := pair.UnmarshalVT(v)
		if err != nil {
			return false
		}
		if pair.GetChord().GetId() != cht.GetId() {
			return false
		}
		if pair.GetTunnel().GetId() != tn.GetId() {
			return false
		}
		return true
	})).Return(nil)

	// On stop -> unpublish
	node.On("Delete", mock.Anything, mock.MatchedBy(func(k []byte) bool {
		exp := [][]byte{
			[]byte(tun.DestinationByChordKey(cht)),
			[]byte(tun.DestinationByTunnelKey(tn)),
		}
		return assertBytes(k, exp...)
	})).Return(nil)

	serv.MustRegister(ctx)

	<-time.After(time.Millisecond * 100)

	serv.Stop()

	<-time.After(time.Millisecond * 100)

	node.AssertExpectations(t)
	clientT.AssertExpectations(t)
	chordT.AssertExpectations(t)
}