1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright 2019-present Facebook Inc. All rights reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
package entql_test
import (
"strconv"
"testing"
"github.com/facebook/ent/entql"
"github.com/stretchr/testify/assert"
)
func TestPString(t *testing.T) {
tests := []struct {
P entql.P
S string
}{
{
P: entql.And(
entql.FieldEQ("name", "a8m"),
entql.FieldIn("org", "fb", "ent"),
),
S: `name == "a8m" && org in ["fb","ent"]`,
},
{
P: entql.Or(
entql.Not(entql.FieldEQ("name", "mashraki")),
entql.FieldIn("org", "fb", "ent"),
),
S: `!(name == "mashraki") || org in ["fb","ent"]`,
},
{
P: entql.HasEdgeWith(
"groups",
entql.HasEdgeWith(
"admins",
entql.Not(entql.FieldEQ("name", "a8m")),
),
),
S: `has_edge(groups, has_edge(admins, !(name == "a8m")))`,
},
{
P: entql.And(
entql.FieldGT("age", 30),
entql.FieldContains("workplace", "fb"),
),
S: `age > 30 && contains(workplace, "fb")`,
},
{
P: entql.Not(entql.FieldLT("score", 32.23)),
S: `!(score < 32.23)`,
},
{
P: entql.And(
entql.FieldNil("active"),
entql.FieldNotNil("name"),
),
S: `active == nil && name != nil`,
},
{
P: entql.Or(
entql.FieldNotIn("id", 1, 2, 3),
entql.FieldHasSuffix("name", "admin"),
),
S: `id not in [1,2,3] || has_suffix(name, "admin")`,
},
{
P: entql.EQ(entql.F("current"), entql.F("total")).Negate(),
S: `!(current == total)`,
},
}
for i := range tests {
t.Run(strconv.Itoa(i), func(t *testing.T) {
s := tests[i].P.String()
assert.Equal(t, tests[i].S, s)
})
}
}
|
