frostfs-node/pkg/util/attributes/parser_test.go

package attributes_test

import (
	"testing"

	"github.com/TrueCloudLab/frostfs-node/pkg/util/attributes"
	"github.com/TrueCloudLab/frostfs-sdk-go/netmap"
	"github.com/stretchr/testify/require"
)

func testAttributeMap(t *testing.T, mSrc, mExp map[string]string) {
	var node netmap.NodeInfo

	s := make([]string, 0, len(mSrc))
	for k, v := range mSrc {
		s = append(s, k+":"+v)
	}

	err := attributes.ReadNodeAttributes(&node, s)
	require.NoError(t, err)

	if mExp == nil {
		mExp = mSrc
	}

	node.IterateAttributes(func(key, value string) {
		v, ok := mExp[key]
		require.True(t, ok)
		require.Equal(t, value, v)
		delete(mExp, key)
	})

	require.Empty(t, mExp)
}

func TestParseV2Attributes(t *testing.T) {
	t.Run("empty", func(t *testing.T) {
		var node netmap.NodeInfo
		err := attributes.ReadNodeAttributes(&node, nil)
		require.NoError(t, err)
		require.Zero(t, node.NumberOfAttributes())
	})

	t.Run("empty key and/or value", func(t *testing.T) {
		var node netmap.NodeInfo
		err := attributes.ReadNodeAttributes(&node, []string{
			":HDD",
		})
		require.Error(t, err)

		err = attributes.ReadNodeAttributes(&node, []string{
			"StorageType:",
		})
		require.Error(t, err)

		err = attributes.ReadNodeAttributes(&node, []string{
			":",
		})
		require.Error(t, err)
	})

	t.Run("non-unique keys", func(t *testing.T) {
		var node netmap.NodeInfo
		err := attributes.ReadNodeAttributes(&node, []string{
			"StorageType:HDD",
			"StorageType:HDD",
		})
		require.Error(t, err)
	})

	t.Run("malformed", func(t *testing.T) {
		var node netmap.NodeInfo
		err := attributes.ReadNodeAttributes(&node, []string{"..."})
		require.Error(t, err)

		err = attributes.ReadNodeAttributes(&node, []string{"a:b", ""})
		require.Error(t, err)

		err = attributes.ReadNodeAttributes(&node, []string{"//"})
		require.Error(t, err)
	})

	t.Run("correct", func(t *testing.T) {
		testAttributeMap(t, map[string]string{
			"Location":    "Europe",
			"StorageType": "HDD",
		}, nil)
	})

	t.Run("escape characters", func(t *testing.T) {
		testAttributeMap(t, map[string]string{
			`K\:ey1`: `V\/alue`,
			`Ke\/y2`: `Va\:lue`,
		}, map[string]string{
			`K:ey1`:  `V\/alue`,
			`Ke\/y2`: `Va:lue`,
		})
	})
}