package netmap import ( "strings" "testing" "github.com/stretchr/testify/require" ) func TestDecodeString(t *testing.T) { testCases := []string{ `REP 2 CBF 2 SELECT 2 FROM *`, `REP 1 IN X CBF 1 SELECT 2 IN SAME Location FROM * AS X`, `REP 1 IN X REP 2 IN Y CBF 1 SELECT 2 FROM * AS X SELECT 3 FROM * AS Y`, `REP 1 IN X SELECT 2 IN City FROM Good AS X FILTER Country EQ RU AS FromRU FILTER Country EQ EN AS FromEN FILTER @FromRU AND @FromEN AND Rating GT 7 AS Good`, `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`, `REP 7 IN SPB SELECT 1 IN City FROM SPBSSD AS SPB FILTER NOT (NOT (City EQ SPB) AND SSD EQ true OR City EQ SPB AND Rating GE 5) AS SPBSSD`, `REP 1 IN FNODE CBF 1 SELECT 1 FROM F AS FNODE FILTER Node EQ '10.78.8.11' AS F`, `UNIQUE REP 1 REP 1`, } var p PlacementPolicy for _, testCase := range testCases { require.NoError(t, p.DecodeString(testCase), "unable parse %s", testCase) var b strings.Builder require.NoError(t, p.WriteStringTo(&b)) require.Equal(t, testCase, b.String()) } invalidTestCases := map[string]error{ `?REP 1`: errSyntaxError, `REP 1 trailing garbage`: errSyntaxError, `REP 1 REP 1 SELECT 4 FROM *`: errUnnamedSelector, `REP 1 SELECT 4 FROM * SELECT 1 FROM *`: errUnnamedSelector, `REP 1 IN X SELECT 4 FROM *`: errUnknownSelector, `REP 1 IN X REP 2 SELECT 4 FROM * AS X FILTER 'UN-LOCODE' EQ 'RU LED' AS F`: errRedundantFilter, } for i := range invalidTestCases { require.ErrorIs(t, p.DecodeString(i), invalidTestCases[i], "#%s", i) } }