golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/sqljson: use the builtin string predciates in sqljson and improve tests

Browse Source
This commit is contained in:
Ariel Mashraki
2021-11-16 10:07:34 +02:00
parent 12feb4301d f9e2609e95
commit 77701d4db6
3 changed files with 159 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
entc/integration/json/json_test.go
View File

@@ -300,6 +300,7 @@ func Predicates(t *testing.T, client *ent.Client) {
s.Where(sqljson.ValueContains(user.FieldInts, 3))
}).OnlyX(ctx)
require.Contains(t, r.Ints, 3)
r = client.User.Query().Where(func(s *sql.Selector) {
s.Where(sqljson.ValueContains(user.FieldT, 3, sqljson.Path("li")))
}).OnlyX(ctx)
@@ -334,4 +335,58 @@ func Predicates(t *testing.T, client *ent.Client) {
}).CountX(ctx)
require.Equal(t, 2, n, "both u1 and u2 have a 'User' key")
})
t.Run("Strings", func(t *testing.T) {
client.User.Delete().ExecX(ctx)
u, err := url.Parse("https://github.com/a8m")
require.NoError(t, err)
dirs := []http.Dir{"/dev/null"}
_, err = client.User.CreateBulk(
client.User.Create().SetURL(u),
client.User.Create().SetDirs(dirs),
client.User.Create().SetT(&schema.T{S: "foobar", Ls: []string{"foo", "bar"}}),
).Save(ctx)
require.NoError(t, err)
ps := []*sql.Predicate{
sqljson.StringContains(user.FieldDirs, "dev", sqljson.Path("[0]")),
sqljson.StringHasPrefix(user.FieldDirs, "/dev", sqljson.Path("[0]")),
sqljson.StringHasSuffix(user.FieldDirs, "/null", sqljson.Path("[0]")),
}
for _, p := range ps {
r = client.User.Query().Where(func(s *sql.Selector) { s.Where(p) }).OnlyX(ctx)
require.Equal(t, dirs, r.Dirs)
}
r = client.User.Query().Where(func(s *sql.Selector) { s.Where(sql.And(ps...)) }).OnlyX(ctx)
require.Equal(t, dirs, r.Dirs)
ps = []*sql.Predicate{
sqljson.StringContains(user.FieldURL, "hub", sqljson.Path("Host")),
sqljson.StringHasPrefix(user.FieldURL, "github", sqljson.Path("Host")),
sqljson.StringHasSuffix(user.FieldURL, "hub.com", sqljson.Path("Host")),
}
for _, p := range ps {
r = client.User.Query().Where(func(s *sql.Selector) { s.Where(p) }).OnlyX(ctx)
require.Equal(t, u, r.URL)
}
ps = []*sql.Predicate{
sqljson.StringHasPrefix(user.FieldT, "foo", sqljson.Path("ls", "[0]")),
sqljson.StringHasSuffix(user.FieldT, "bar", sqljson.DotPath("ls[1]")),
sql.And(
sql.Or(
sqljson.StringContains(user.FieldT, "foo", sqljson.DotPath("ls[0]")),
sqljson.StringContains(user.FieldT, "foo", sqljson.DotPath("ls[1]")),
),
sql.Or(
sqljson.StringContains(user.FieldT, "bar", sqljson.DotPath("ls[0]")),
sqljson.StringContains(user.FieldT, "bar", sqljson.DotPath("ls[1]")),
),
),
}
for _, p := range ps {
r = client.User.Query().Where(func(s *sql.Selector) { s.Where(p) }).OnlyX(ctx)
require.Equal(t, []string{"foo", "bar"}, r.T.Ls)
}
})
}