golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-04-29 22:20:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/sqlgraph: better support for update nodes with predicates (#2574)

Browse Source
This commit is contained in:
Ariel Mashraki
2022-05-29 16:23:52 +03:00
committed by GitHub
parent 0917701f91
commit 5b81d7d832
7 changed files with 116 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1572,7 +1572,7 @@ func TestUpdateNode(t *testing.T) {
wantUser: &user{name: "Ariel", age: 30, id: 1},
},
{
name: "fields/add_clear",
name: "fields/add_set_clear",
spec: &UpdateSpec{
Node: &NodeSpec{
Table: "users",
@@ -1586,6 +1586,9 @@ func TestUpdateNode(t *testing.T) {
Add: []*FieldSpec{
{Column: "age", Type: field.TypeInt, Value: 1},
},
Set: []*FieldSpec{
{Column: "deleted", Type: field.TypeBool, Value: true},
},
Clear: []*FieldSpec{
{Column: "name", Type: field.TypeString},
},
@@ -1593,10 +1596,10 @@ func TestUpdateNode(t *testing.T) {
},
prepare: func(mock sqlmock.Sqlmock) {
mock.ExpectBegin()
mock.ExpectExec(escape("UPDATE `users` SET `name` = NULL, `age` = COALESCE(`users`.`age`, 0) + ? WHERE `id` = ? AND NOT `deleted`")).
WithArgs(1, 1).
WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectQuery(escape("SELECT `id`, `name`, `age` FROM `users` WHERE `id` = ? AND NOT `deleted`")).
mock.ExpectExec(escape("UPDATE `users` SET `name` = NULL, `deleted` = ?, `age` = COALESCE(`users`.`age`, 0) + ? WHERE `id` = ? AND NOT `deleted`")).
WithArgs(true, 1, 1).
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectQuery(escape("SELECT `id`, `name`, `age` FROM `users` WHERE `id` = ?")).
WithArgs(1).
WillReturnRows(sqlmock.NewRows([]string{"id", "age", "name"}).
AddRow(1, 31, nil))
@@ -1604,6 +1607,43 @@ func TestUpdateNode(t *testing.T) {
},
wantUser: &user{age: 31, id: 1},
},
{
name: "fields/ensure_exists",
spec: &UpdateSpec{
Node: &NodeSpec{
Table: "users",
Columns: []string{"id", "name", "age"},
ID: &FieldSpec{Column: "id", Type: field.TypeInt, Value: 1},
},
Predicate: func(s *sql.Selector) {
s.Where(sql.EQ("deleted", false))
},
Fields: FieldMut{
Add: []*FieldSpec{
{Column: "age", Type: field.TypeInt, Value: 1},
},
Set: []*FieldSpec{
{Column: "deleted", Type: field.TypeBool, Value: true},
},
Clear: []*FieldSpec{
{Column: "name", Type: field.TypeString},
},
},
},
prepare: func(mock sqlmock.Sqlmock) {
mock.ExpectBegin()
mock.ExpectExec(escape("UPDATE `users` SET `name` = NULL, `deleted` = ?, `age` = COALESCE(`users`.`age`, 0) + ? WHERE `id` = ? AND NOT `deleted`")).
WithArgs(true, 1, 1).
WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectQuery(escape("SELECT EXISTS (SELECT * FROM `users` WHERE `id` = ? AND NOT `deleted`)")).
WithArgs(1).
WillReturnRows(sqlmock.NewRows([]string{"exists"}).
AddRow(false))
mock.ExpectRollback()
},
wantErr: true,
wantUser: &user{},
},
{
name: "edges/o2o_non_inverse and m2o",
spec: &UpdateSpec{