105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package netmap_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
. "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
netmaptest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEncode(t *testing.T) {
|
|
testCases := []string{
|
|
`REP 1 IN X
|
|
CBF 1
|
|
SELECT 2 IN SAME Location FROM * AS X`,
|
|
|
|
`REP 1
|
|
SELECT 2 IN City FROM Good
|
|
FILTER Country EQ RU AS FromRU
|
|
FILTER @FromRU 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`,
|
|
|
|
`UNIQUE
|
|
REP 1
|
|
REP 1`,
|
|
|
|
`REP 1 IN X
|
|
SELECT 1 FROM F AS X
|
|
FILTER 'UN-LOCODE' EQ 'RU LED' AS F`,
|
|
}
|
|
|
|
var p PlacementPolicy
|
|
|
|
for _, testCase := range testCases {
|
|
require.NoError(t, p.DecodeString(testCase))
|
|
|
|
var b strings.Builder
|
|
require.NoError(t, p.WriteStringTo(&b))
|
|
|
|
require.Equal(t, testCase, b.String())
|
|
}
|
|
|
|
invalidTestCases := []string{
|
|
`?REP 1`,
|
|
`REP 1 trailing garbage`,
|
|
}
|
|
|
|
for i := range invalidTestCases {
|
|
require.Error(t, p.DecodeString(invalidTestCases[i]), "#%d", i)
|
|
}
|
|
}
|
|
|
|
func TestPlacementPolicyEncoding(t *testing.T) {
|
|
v := netmaptest.PlacementPolicy()
|
|
|
|
t.Run("binary", func(t *testing.T) {
|
|
var v2 PlacementPolicy
|
|
require.NoError(t, v2.Unmarshal(v.Marshal()))
|
|
|
|
require.Equal(t, v, v2)
|
|
})
|
|
|
|
t.Run("json", func(t *testing.T) {
|
|
data, err := v.MarshalJSON()
|
|
require.NoError(t, err)
|
|
|
|
var v2 PlacementPolicy
|
|
require.NoError(t, v2.UnmarshalJSON(data))
|
|
|
|
require.Equal(t, v, v2)
|
|
})
|
|
}
|
|
|
|
func TestDecodeSelectFilterExpr(t *testing.T) {
|
|
for _, s := range []string{
|
|
"SELECT 1 FROM *",
|
|
"FILTER Color EQ 'Red' AS RedNode",
|
|
`
|
|
FILTER Color EQ 'Red' AS RedNode
|
|
FILTER @RedNode AND Shape EQ 'Cirle' AS RedCircleNode
|
|
`,
|
|
`
|
|
SELECT 1 FROM RedCircleNode
|
|
FILTER Color EQ 'Red' AS RedNode
|
|
FILTER @RedNode AND Shape EQ 'Cirle' AS RedCircleNode
|
|
`,
|
|
`
|
|
CBF 1
|
|
SELECT 1 FROM RedCircleNode
|
|
FILTER Color EQ 'Red' AS RedNode
|
|
FILTER @RedNode AND Shape EQ 'Cirle' AS RedCircleNode
|
|
`,
|
|
} {
|
|
_, err := DecodeSelectFilterString(s)
|
|
require.NoError(t, err)
|
|
}
|
|
}
|