package netmap

import (
	"encoding/hex"
	"testing"

	netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc"
	crypto "github.com/nspcc-dev/neofs-crypto"
	"github.com/nspcc-dev/neofs-node/pkg/util/test"
	"github.com/stretchr/testify/require"
)

func TestCleanupTable(t *testing.T) {
	infos := []netmap.NodeInfo{
		{PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(1).PublicKey)},
		{PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(2).PublicKey)},
		{PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(3).PublicKey)},
	}

	mapInfos := map[string]struct{}{
		hex.EncodeToString(infos[0].PublicKey): {},
		hex.EncodeToString(infos[1].PublicKey): {},
		hex.EncodeToString(infos[2].PublicKey): {},
	}

	t.Run("update", func(t *testing.T) {
		c := newCleanupTable(true, 1)
		c.update(infos, 1)
		require.Len(t, c.lastAccess, len(infos))

		for k, v := range c.lastAccess {
			require.EqualValues(t, 1, v.epoch)
			require.False(t, v.removeFlag)

			_, ok := mapInfos[k]
			require.True(t, ok)
		}

		t.Run("update with flagged", func(t *testing.T) {
			key := hex.EncodeToString(infos[0].PublicKey)
			c.flag(key)

			c.update(infos, 2)
			require.EqualValues(t, 1, c.lastAccess[key].epoch)
			require.False(t, c.lastAccess[key].removeFlag)
		})
	})

	t.Run("touch", func(t *testing.T) {
		c := newCleanupTable(true, 1)
		c.update(infos, 1)

		key := hex.EncodeToString(infos[1].PublicKey)
		require.True(t, c.touch(key, 11))
		require.EqualValues(t, 11, c.lastAccess[key].epoch)

		require.False(t, c.touch(key+"x", 12))
		require.EqualValues(t, 12, c.lastAccess[key+"x"].epoch)
	})

	t.Run("flag", func(t *testing.T) {
		c := newCleanupTable(true, 1)
		c.update(infos, 1)

		key := hex.EncodeToString(infos[1].PublicKey)
		c.flag(key)
		require.True(t, c.lastAccess[key].removeFlag)

		require.False(t, c.touch(key, 2))
		require.False(t, c.lastAccess[key].removeFlag)
	})

	t.Run("iterator", func(t *testing.T) {
		c := newCleanupTable(true, 2)
		c.update(infos, 1)

		t.Run("no nodes to remove", func(t *testing.T) {
			cnt := 0
			require.NoError(t,
				c.forEachRemoveCandidate(2, func(_ string) error {
					cnt++
					return nil
				}))
			require.EqualValues(t, 0, cnt)
		})

		t.Run("all nodes to remove", func(t *testing.T) {
			cnt := 0
			require.NoError(t,
				c.forEachRemoveCandidate(4, func(s string) error {
					cnt++
					_, ok := mapInfos[s]
					require.True(t, ok)
					return nil
				}))
			require.EqualValues(t, len(infos), cnt)
		})

		t.Run("some nodes to remove", func(t *testing.T) {
			cnt := 0
			key := hex.EncodeToString(infos[1].PublicKey)

			require.False(t, c.touch(key, 4)) // one node was updated

			require.NoError(t,
				c.forEachRemoveCandidate(4, func(s string) error {
					cnt++
					require.NotEqual(t, s, key)
					return nil
				}))
			require.EqualValues(t, len(infos)-1, cnt)
		})
	})
}