golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-04-28 13:40:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

ent: initial support for edge schemas (#2560)

Browse Source
This commit is contained in:
Ariel Mashraki
2022-05-25 15:46:00 +03:00
committed by GitHub
parent 1e05ccc284
commit e1c5277483
306 changed files with 30969 additions and 1233 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1135,6 +1135,29 @@ func TestCreateNode(t *testing.T) {
m.ExpectCommit()
},
},
{
name: "edges/m2m/fields",
spec: &CreateSpec{
Table: "groups",
ID: &FieldSpec{Column: "id", Type: field.TypeInt},
Fields: []*FieldSpec{
{Column: "name", Type: field.TypeString, Value: "GitHub"},
},
Edges: []*EdgeSpec{
{Rel: M2M, Table: "group_users", Columns: []string{"group_id", "user_id"}, Target: &EdgeTarget{Nodes: []driver.Value{2}, IDSpec: &FieldSpec{Column: "id"}, Fields: []*FieldSpec{{Column: "ts", Type: field.TypeInt, Value: 3}}}},
},
},
expect: func(m sqlmock.Sqlmock) {
m.ExpectBegin()
m.ExpectExec(escape("INSERT INTO `groups` (`name`) VALUES (?)")).
WithArgs("GitHub").
WillReturnResult(sqlmock.NewResult(1, 1))
m.ExpectExec(escape("INSERT INTO `group_users` (`group_id`, `user_id`, `ts`) VALUES (?, ?, ?)")).
WithArgs(1, 2, 3).
WillReturnResult(sqlmock.NewResult(1, 1))
m.ExpectCommit()
},
},
{
name: "edges/m2m/inverse",
spec: &CreateSpec{
@@ -1181,6 +1204,29 @@ func TestCreateNode(t *testing.T) {
m.ExpectCommit()
},
},
{
name: "edges/m2m/bidi/fields",
spec: &CreateSpec{
Table: "users",
ID: &FieldSpec{Column: "id", Type: field.TypeInt},
Fields: []*FieldSpec{
{Column: "name", Type: field.TypeString, Value: "mashraki"},
},
Edges: []*EdgeSpec{
{Rel: M2M, Bidi: true, Table: "user_friends", Columns: []string{"user_id", "friend_id"}, Target: &EdgeTarget{Nodes: []driver.Value{2}, IDSpec: &FieldSpec{Column: "id"}, Fields: []*FieldSpec{{Column: "ts", Type: field.TypeInt, Value: 3}}}},
},
},
expect: func(m sqlmock.Sqlmock) {
m.ExpectBegin()
m.ExpectExec(escape("INSERT INTO `users` (`name`) VALUES (?)")).
WithArgs("mashraki").
WillReturnResult(sqlmock.NewResult(1, 1))
m.ExpectExec(escape("INSERT INTO `user_friends` (`user_id`, `friend_id`, `ts`) VALUES (?, ?, ?), (?, ?, ?)")).
WithArgs(1, 2, 3, 2, 1, 3).
WillReturnResult(sqlmock.NewResult(1, 1))
m.ExpectCommit()
},
},
{
name: "edges/m2m/bidi/batch",
spec: &CreateSpec{