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/sqltest: ensure predicates can be reused

Browse Source
This commit is contained in:
Ariel Mashraki
2021-11-02 22:25:51 +02:00
committed by Ariel Mashraki
parent 924de2d341
commit 02203c3f91
2 changed files with 70 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
dialect/sql/builder_test.go
View File

@@ -1874,3 +1874,60 @@ func TestEscapePatterns(t *testing.T) {
require.Equal(t, "UPDATE `users` SET `name` = NULL WHERE `nickname` LIKE ? ESCAPE ? OR `nickname` LIKE ? ESCAPE ? OR `nickname` LIKE ? ESCAPE ? OR LOWER(`nickname`) LIKE ? ESCAPE ?", q)
require.Equal(t, []interface{}{"\\%a8m\\%%", "\\", "%\\_alexsn\\_", "\\", "%\\\\pedro\\\\%", "\\", "%\\%abcd\\%efg%", "\\"}, args)
}
func TestReusePredicates(t *testing.T) {
tests := []struct {
p *Predicate
wantQuery string
wantArgs []interface{}
}{
{
p: EQ("active", false),
wantQuery: "SELECT * FROM `users` WHERE `active` = ?",
wantArgs: []interface{}{false},
},
{
p: Or(
EQ("a", "a"),
EQ("b", "b"),
),
wantQuery: "SELECT * FROM `users` WHERE `a` = ? OR `b` = ?",
wantArgs: []interface{}{"a", "b"},
},
{
p: And(
EQ("active", true),
HasPrefix("name", "foo"),
HasSuffix("name", "bar"),
Or(
In("id", Select("oid").From(Table("audit"))),
In("id", Select("oid").From(Table("history"))),
),
),
wantQuery: "SELECT * FROM `users` WHERE `active` = ? AND `name` LIKE ? AND `name` LIKE ? AND (`id` IN (SELECT `oid` FROM `audit`) OR `id` IN (SELECT `oid` FROM `history`))",
wantArgs: []interface{}{true, "foo%", "%bar"},
},
{
p: func() *Predicate {
t1 := Table("groups")
pivot := Table("user_groups")
matches := Select(pivot.C("user_id")).
From(pivot).
Join(t1).
On(pivot.C("group_id"), t1.C("id")).
Where(EQ(t1.C("name"), "ent"))
return In("id", matches)
}(),
wantQuery: "SELECT * FROM `users` WHERE `id` IN (SELECT `user_groups`.`user_id` FROM `user_groups` JOIN `groups` AS `t1` ON `user_groups`.`group_id` = `t1`.`id` WHERE `t1`.`name` = ?)",
wantArgs: []interface{}{"ent"},
},
}
for _, tt := range tests {
query, args := Select().From(Table("users")).Where(tt.p).Query()
require.Equal(t, tt.wantQuery, query)
require.Equal(t, tt.wantArgs, args)
query, args = Select().From(Table("users")).Where(tt.p).Query()
require.Equal(t, tt.wantQuery, query)
require.Equal(t, tt.wantArgs, args)
}
}

22
dialect/sql/sqlgraph/graph_test.go
View File

@@ -625,10 +625,12 @@ func TestHasNeighbors(t *testing.T) {
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
HasNeighbors(tt.selector, tt.step)
query, args := tt.selector.Query()
require.Equal(t, tt.wantQuery, query)
require.Empty(t, args)
for _, s := range []*sql.Selector{tt.selector, tt.selector.Clone()} {
HasNeighbors(s, tt.step)
query, args := s.Query()
require.Equal(t, tt.wantQuery, query)
require.Empty(t, args)
}
})
}
}
@@ -874,11 +876,13 @@ WHERE "s1"."users"."id" IN
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
HasNeighborsWith(tt.selector, tt.step, tt.predicate)
query, args := tt.selector.Query()
tt.wantQuery = strings.Join(strings.Fields(tt.wantQuery), " ")
require.Equal(t, tt.wantQuery, query)
require.Equal(t, tt.wantArgs, args)
for _, s := range []*sql.Selector{tt.selector, tt.selector.Clone()} {
HasNeighborsWith(s, tt.step, tt.predicate)
query, args := s.Query()
tt.wantQuery = strings.Join(strings.Fields(tt.wantQuery), " ")
require.Equal(t, tt.wantQuery, query)
require.Equal(t, tt.wantArgs, args)
}
})
}
}