package chainmanager_test

import (
	"context"
	"errors"
	"github.com/brianvoe/gofakeit/v6"
	"github.com/richardwilkes/toolbox/collection"
	. "github.com/stretchr/testify/assert"
	"github.com/synapsecns/sanguine/core/metrics"
	"github.com/synapsecns/sanguine/services/omnirpc/chainmanager"
	"github.com/synapsecns/sanguine/services/omnirpc/metadata"
	"github.com/synapsecns/sanguine/services/omnirpc/rpcinfo"
	"sort"
	"testing"
	"time"
)

func TestRefreshRPCInfoNil(t *testing.T) {
	nullHandler, err := metrics.NewByType(context.Background(), metadata.BuildInfo(), metrics.Null)
	NoError(t, err)

	cm := chainmanager.NewChainManager(nullHandler)

	// make sure we don't panic if the chain is not nil
	NotPanics(t, func() {
		cm.RefreshRPCInfo(context.Background(), gofakeit.Uint32())
	})
}

func TestSortInfoList(t *testing.T) {
	// errors first
	infoList := []rpcinfo.Result{
		{
			URL:      "err",
			HasError: true,
			Error:    errors.New(gofakeit.Sentence(10)),
		},
		{
			URL:      "err",
			HasError: true,
			Error:    errors.New(gofakeit.Sentence(10)),
		},
		{
			URL:      "laggiest",
			BlockAge: time.Second * 20,
			Latency:  time.Second * 20,
		},
		{
			URL:      "laggier",
			BlockAge: time.Second * 20,
			Latency:  time.Second * 10,
		},
		{
			URL:     "slowest",
			Latency: time.Second * 20,
		},
		{
			URL:     "slower",
			Latency: time.Second * 10,
		},
	}

	sortedInfoList := chainmanager.SortInfoList(infoList)
	Equal(t, infoList, sortedInfoList)
}

func TestGetChainIDs(t *testing.T) {
	nullHandler, err := metrics.NewByType(context.Background(), metadata.BuildInfo(), metrics.Null)
	NoError(t, err)

	cm := chainmanager.NewChainManager(nullHandler)

	chainIDs := collection.Set[uint32]{}

	for i := 0; i < 40; i++ {
		testChainID := gofakeit.Uint32()

		chainIDs.Add(testChainID)
		cm.PutChain(testChainID, []string{gofakeit.URL(), gofakeit.URL()}, gofakeit.Uint16())
	}

	// sort both slices to assert equality
	Equal(t, sortSlice(cm.GetChainIDs()), sortSlice(chainIDs.Values()))
}

// sortSlice sorts a uint 32 slice.
func sortSlice(res []uint32) []uint32 {
	sort.Slice(res, func(i, j int) bool {
		return res[i] > res[j]
	})
	return res
}