frostfs-sdk-go/policy/query_test.go

package policy

import (
	"errors"
	"fmt"
	"math"
	"testing"

	"github.com/nspcc-dev/neofs-sdk-go/netmap"
	"github.com/stretchr/testify/require"
)

func TestSimple(t *testing.T) {
	q := `REP 3`
	expected := new(netmap.PlacementPolicy)
	expected.SetFilters([]netmap.Filter{}...)
	expected.SetSelectors([]netmap.Selector{}...)
	expected.SetReplicas(newReplica("", 3))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestSimpleWithHRWB(t *testing.T) {
	q := `REP 3 CBF 4`
	expected := new(netmap.PlacementPolicy)
	expected.SetFilters([]netmap.Filter{}...)
	expected.SetSelectors([]netmap.Selector{}...)
	expected.SetReplicas(newReplica("", 3))
	expected.SetContainerBackupFactor(4)

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestFromSelect(t *testing.T) {
	q := `REP 1 IN SPB
SELECT 1 IN City FROM * AS SPB`
	expected := new(netmap.PlacementPolicy)
	expected.SetFilters([]netmap.Filter{}...)
	expected.SetSelectors(newSelector(1, netmap.ClauseUnspecified, "City", "*", "SPB"))
	expected.SetReplicas(newReplica("SPB", 1))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

// https://github.com/nspcc-dev/neofs-node/issues/46
func TestFromSelectNoAttribute(t *testing.T) {
	t.Run("Simple", func(t *testing.T) {
		q := `REP 2
		SELECT 6 FROM *`

		expected := new(netmap.PlacementPolicy)
		expected.SetFilters([]netmap.Filter{}...)
		expected.SetSelectors(newSelector(6, netmap.ClauseUnspecified, "", "*", ""))
		expected.SetReplicas(newReplica("", 2))

		r, err := Parse(q)
		require.NoError(t, err)
		require.EqualValues(t, expected, r)
	})

	t.Run("with filter", func(t *testing.T) {
		q := `REP 2
		SELECT 6 FROM F
		FILTER StorageType EQ SSD AS F`

		expected := new(netmap.PlacementPolicy)
		expected.SetFilters(newFilter("F", "StorageType", "SSD", netmap.OpEQ))
		expected.SetSelectors(newSelector(6, netmap.ClauseUnspecified, "", "F", ""))
		expected.SetReplicas(newReplica("", 2))

		r, err := Parse(q)
		require.NoError(t, err)
		require.EqualValues(t, expected, r)
	})
}

func TestString(t *testing.T) {
	qTemplate := `REP 1
SELECT 1 IN City FROM Filt
FILTER Property EQ %s AND Something NE 7 AS Filt`

	testCases := []string{
		`"double-quoted"`,
		`"with ' single"`,
		`'single-quoted'`,
		`'with " double'`,
	}

	for _, s := range testCases {
		t.Run(s, func(t *testing.T) {
			q := fmt.Sprintf(qTemplate, s)
			r, err := Parse(q)
			require.NoError(t, err)

			expected := newFilter("Filt", "", "", netmap.OpAND,
				newFilter("", "Property", s[1:len(s)-1], netmap.OpEQ),
				newFilter("", "Something", "7", netmap.OpNE))
			require.EqualValues(t, []netmap.Filter{expected}, r.Filters())
		})
	}
}

func TestFromSelectClause(t *testing.T) {
	q := `REP 4
SELECT 3 IN Country FROM *
SELECT 2 IN SAME City FROM *
SELECT 1 IN DISTINCT Continent FROM *`
	expected := new(netmap.PlacementPolicy)
	expected.SetFilters([]netmap.Filter{}...)
	expected.SetSelectors(
		newSelector(3, netmap.ClauseUnspecified, "Country", "*", ""),
		newSelector(2, netmap.ClauseSame, "City", "*", ""),
		newSelector(1, netmap.ClauseDistinct, "Continent", "*", ""))
	expected.SetReplicas(newReplica("", 4))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestSimpleFilter(t *testing.T) {
	q := `REP 1
SELECT 1 IN City FROM Good
FILTER Rating GT 7 AS Good`
	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("", 1))
	expected.SetSelectors(
		newSelector(1, netmap.ClauseUnspecified, "City", "Good", ""))
	expected.SetFilters(newFilter("Good", "Rating", "7", netmap.OpGT))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestFilterReference(t *testing.T) {
	q := `REP 1
SELECT 2 IN City FROM Good
FILTER Country EQ "RU" AS FromRU
FILTER @FromRU AND Rating GT 7 AS Good`
	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("", 1))
	expected.SetSelectors(
		newSelector(2, netmap.ClauseUnspecified, "City", "Good", ""))
	expected.SetFilters(
		newFilter("FromRU", "Country", "RU", netmap.OpEQ),
		newFilter("Good", "", "", netmap.OpAND,
			newFilter("FromRU", "", "", 0),
			newFilter("", "Rating", "7", netmap.OpGT)),
	)

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestFilterOps(t *testing.T) {
	q := `REP 1
SELECT 2 IN City FROM Good
FILTER A GT 1 AND B GE 2 AND C LT 3 AND D LE 4
  AND E EQ 5 AND F NE 6 AS Good`
	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("", 1))
	expected.SetSelectors(
		newSelector(2, netmap.ClauseUnspecified, "City", "Good", ""))
	expected.SetFilters(newFilter("Good", "", "", netmap.OpAND,
		newFilter("", "A", "1", netmap.OpGT),
		newFilter("", "B", "2", netmap.OpGE),
		newFilter("", "C", "3", netmap.OpLT),
		newFilter("", "D", "4", netmap.OpLE),
		newFilter("", "E", "5", netmap.OpEQ),
		newFilter("", "F", "6", netmap.OpNE)))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestWithFilterPrecedence(t *testing.T) {
	q := `REP 7 IN SPB
SELECT 1 IN City FROM SPBSSD AS SPB
FILTER City EQ "SPB" AND SSD EQ true OR City EQ "SPB" AND Rating GE 5 AS SPBSSD`
	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("SPB", 7))
	expected.SetSelectors(
		newSelector(1, netmap.ClauseUnspecified, "City", "SPBSSD", "SPB"))
	expected.SetFilters(
		newFilter("SPBSSD", "", "", netmap.OpOR,
			newFilter("", "", "", netmap.OpAND,
				newFilter("", "City", "SPB", netmap.OpEQ),
				newFilter("", "SSD", "true", netmap.OpEQ)),
			newFilter("", "", "", netmap.OpAND,
				newFilter("", "City", "SPB", netmap.OpEQ),
				newFilter("", "Rating", "5", netmap.OpGE))))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestBrackets(t *testing.T) {
	q := `REP 7 IN SPB
SELECT 1 IN City FROM SPBSSD AS SPB
FILTER ( City EQ "SPB" OR SSD EQ true ) AND (City EQ "SPB" OR Rating GE 5) AS SPBSSD`

	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("SPB", 7))
	expected.SetSelectors(
		newSelector(1, netmap.ClauseUnspecified, "City", "SPBSSD", "SPB"))
	expected.SetFilters(
		newFilter("SPBSSD", "", "", netmap.OpAND,
			newFilter("", "", "", netmap.OpOR,
				newFilter("", "City", "SPB", netmap.OpEQ),
				newFilter("", "SSD", "true", netmap.OpEQ)),
			newFilter("", "", "", netmap.OpOR,
				newFilter("", "City", "SPB", netmap.OpEQ),
				newFilter("", "Rating", "5", netmap.OpGE))))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func TestValidation(t *testing.T) {
	t.Run("MissingSelector", func(t *testing.T) {
		q := `REP 3 IN RU`
		_, err := Parse(q)
		require.True(t, errors.Is(err, ErrUnknownSelector), "got: %v", err)
	})
	t.Run("MissingFilter", func(t *testing.T) {
		q := `REP 3
              SELECT 1 IN City FROM MissingFilter`
		_, err := Parse(q)
		require.True(t, errors.Is(err, ErrUnknownFilter), "got: %v", err)
	})
	t.Run("UnknownOp", func(t *testing.T) {
		q := `REP 3
              SELECT 1 IN City FROM F
			  FILTER Country KEK RU AS F`
		_, err := Parse(q)
		require.True(t, errors.Is(err, ErrSyntaxError), "got: %v", err)
	})
	t.Run("TypoInREP", func(t *testing.T) {
		q := `REK 3`
		_, err := Parse(q)
		require.True(t, errors.Is(err, ErrSyntaxError))
	})
	t.Run("InvalidFilterName", func(t *testing.T) {
		q := `REP 3
			  SELECT 1 IN City FROM F
			  FILTER Good AND Country EQ RU AS F
			  FILTER Rating EQ 5 AS Good`
		_, err := Parse(q)
		require.Error(t, err)
	})
}

// Checks that an error is returned in cases when positive 32-bit integer is expected.
func TestInvalidNumbers(t *testing.T) {
	tmpls := []string{
		"REP %d",
		"REP 1 CBF %d",
		"REP 1 SELECT %d FROM *",
	}
	for i := range tmpls {
		zero := fmt.Sprintf(tmpls[i], 0)
		t.Run(zero, func(t *testing.T) {
			_, err := Parse(zero)
			require.Error(t, err)
		})

		big := fmt.Sprintf(tmpls[i], int64(math.MaxUint32)+1)
		t.Run(big, func(t *testing.T) {
			_, err := Parse(big)
			require.Error(t, err)
		})
	}
}

func TestFilterStringSymbols(t *testing.T) {
	q := `REP 1 IN S
SELECT 1 FROM F AS S
FILTER "UN-LOCODE" EQ "RU LED" AS F`

	expected := new(netmap.PlacementPolicy)
	expected.SetReplicas(newReplica("S", 1))
	expected.SetSelectors(
		newSelector(1, netmap.ClauseUnspecified, "", "F", "S"))
	expected.SetFilters(newFilter("F", "UN-LOCODE", "RU LED", netmap.OpEQ))

	r, err := Parse(q)
	require.NoError(t, err)
	require.EqualValues(t, expected, r)
}

func newFilter(name, key, value string, op netmap.Operation, sub ...netmap.Filter) (f netmap.Filter) {
	f.SetName(name)
	f.SetKey(key)
	f.SetValue(value)
	f.SetOperation(op)
	f.SetInnerFilters(sub...)
	return f
}

func newReplica(s string, c uint32) (r netmap.Replica) {
	r.SetSelector(s)
	r.SetCount(c)
	return r
}

func newSelector(count uint32, c netmap.Clause, attr, f, name string) (s netmap.Selector) {
	s.SetCount(count)
	s.SetClause(c)
	s.SetAttribute(attr)
	s.SetFilter(f)
	s.SetName(name)
	return s
}