frostfs-node/pkg/services/object_manager/placement/cache_test.go

package placement_test

import (
	"strconv"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object_manager/placement"
	cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
	cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
	netmapSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
	"github.com/stretchr/testify/require"
)

func TestContainerNodesCache(t *testing.T) {
	const size = 3

	nodes := [6]netmapSDK.NodeInfo{}
	for i := range nodes {
		nodes[i].SetAttribute("ATTR", strconv.Itoa(i))
	}

	nm := func(epoch uint64, nodes []netmapSDK.NodeInfo) *netmapSDK.NetMap {
		var nm netmapSDK.NetMap
		nm.SetEpoch(epoch)
		nm.SetNodes(nodes)
		return &nm
	}

	var pp netmapSDK.PlacementPolicy
	require.NoError(t, pp.DecodeString("REP 1"))

	t.Run("update netmap on the new epoch", func(t *testing.T) {
		c := placement.NewContainerNodesCache(size)

		cnr := cidtest.ID()
		res, err := c.ContainerNodes(nm(1, nodes[0:1]), cnr, pp)
		require.NoError(t, err)

		// Use other nodes in the argument to ensure the result is taken from cache.
		resCached, err := c.ContainerNodes(nm(1, nodes[1:2]), cnr, pp)
		require.NoError(t, err)
		require.Equal(t, res, resCached)

		// Update epoch, netmap should be purged.
		resCached, err = c.ContainerNodes(nm(2, nodes[2:3]), cnr, pp)
		require.NoError(t, err)
		require.NotEqual(t, res, resCached)
	})
	t.Run("cache uses container as a key", func(t *testing.T) {
		c := placement.NewContainerNodesCache(size)

		res1, err := c.ContainerNodes(nm(1, nodes[0:1]), cidtest.ID(), pp)
		require.NoError(t, err)

		res2, err := c.ContainerNodes(nm(1, nodes[1:2]), cidtest.ID(), pp)
		require.NoError(t, err)

		require.NotEqual(t, res1, res2)
	})
	t.Run("cache respects size parameter", func(t *testing.T) {
		c := placement.NewContainerNodesCache(size)

		nm1 := nm(1, nodes[0:1])
		nm2 := nm(1, nodes[1:2])
		cnr := [size * 2]cid.ID{}
		res := [size * 2][][]netmapSDK.NodeInfo{}
		for i := range size * 2 {
			cnr[i] = cidtest.ID()

			var err error
			res[i], err = c.ContainerNodes(nm1, cnr[i], pp)
			require.NoError(t, err)
		}

		for i := size; i < size*2; i++ {
			r, err := c.ContainerNodes(nm2, cnr[i], pp)
			require.NoError(t, err)
			require.Equal(t, res[i], r)
		}
		for i := range size {
			r, err := c.ContainerNodes(nm2, cnr[i], pp)
			require.NoError(t, err)
			require.NotEqual(t, res[i], r)
		}
	})
	t.Run("the error is propagated", func(t *testing.T) {
		var pp netmapSDK.PlacementPolicy
		r := netmapSDK.ReplicaDescriptor{}
		r.SetNumberOfObjects(1)
		r.SetSelectorName("Missing")
		pp.AddReplicas(r)

		c := placement.NewContainerNodesCache(size)
		_, err := c.ContainerNodes(nm(1, nodes[0:1]), cidtest.ID(), pp)
		require.Error(t, err)
	})
}