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

dialect/sql: add option for getting the neighbors of set of vertices (#157)

Browse Source 
Summary:
Pull Request resolved: https://github.com/facebookincubator/ent/pull/157

It should be used by the query builders for stepping to neighbors of the matches.

Reviewed By: alexsn

Differential Revision: D18448517

fbshipit-source-id: bf3aeea8c90a4e9b090e9298a20600b8f0b5ace3
This commit is contained in:
Ariel Mashraki
2019-11-12 10:35:50 -08:00
committed by Facebook Github Bot
parent aa57d732c1
commit aece79f8a8
2 changed files with 148 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
dialect/sql/graph_test.go
View File

@@ -5,6 +5,7 @@
package sql
import (
"strings"
"testing"
"github.com/stretchr/testify/require"
@@ -204,3 +205,109 @@ func TestNeighbors(t *testing.T) {
})
}
}
func TestSetNeighbors(t *testing.T) {
tests := []struct {
name string
input *Step
wantQuery string
wantArgs []interface{}
}{
{
name: "O2M/2types",
input: func() *Step {
step := &Step{}
step.From.V = Select().From(Table("users")).Where(EQ("name", "a8m"))
step.From.Table = "users"
step.From.Column = "id"
step.To.Table = "pets"
step.To.Column = "id"
step.Edge.Rel = O2M
step.Edge.Table = "pets"
step.Edge.Columns = []string{"owner_id"}
return step
}(),
wantQuery: `SELECT * FROM "pets" JOIN (SELECT "id" FROM "users" WHERE "name" = $1) AS "t1" ON "pets"."owner_id" = "t1"."id"`,
wantArgs: []interface{}{"a8m"},
},
{
name: "M2O/2types",
input: func() *Step {
step := &Step{}
step.From.V = Select().From(Table("pets")).Where(EQ("name", "pedro"))
step.From.Table = "pets"
step.From.Column = "id"
step.To.Table = "users"
step.To.Column = "id"
step.Edge.Rel = M2O
step.Edge.Table = "pets"
step.Edge.Columns = []string{"owner_id"}
return step
}(),
wantQuery: `SELECT * FROM "users" JOIN (SELECT "owner_id" FROM "pets" WHERE "name" = $1) AS "t1" ON "users"."id" = "t1"."owner_id"`,
wantArgs: []interface{}{"pedro"},
},
{
name: "M2M/2types",
input: func() *Step {
step := &Step{}
step.From.V = Select().From(Table("users")).Where(EQ("name", "a8m"))
step.From.Table = "users"
step.From.Column = "id"
step.To.Table = "groups"
step.To.Column = "id"
step.Edge.Rel = M2M
step.Edge.Table = "user_groups"
step.Edge.Columns = []string{"user_id", "group_id"}
return step
}(),
wantQuery: `
SELECT *
FROM "groups"
JOIN
(SELECT "user_groups"."group_id"
FROM "user_groups"
JOIN
(SELECT "id"
FROM "users"
WHERE "name" = $1) AS "t1" ON "user_groups"."user_id" = "t1"."id") AS "t1" ON "groups"."id" = "t1"."group_id"`,
wantArgs: []interface{}{"a8m"},
},
{
name: "M2M/2types/inverse",
input: func() *Step {
step := &Step{}
step.From.V = Select().From(Table("groups")).Where(EQ("name", "GitHub"))
step.From.Table = "groups"
step.From.Column = "id"
step.To.Table = "users"
step.To.Column = "id"
step.Edge.Rel = M2M
step.Edge.Inverse = true
step.Edge.Table = "user_groups"
step.Edge.Columns = []string{"user_id", "group_id"}
return step
}(),
wantQuery: `
SELECT *
FROM "users"
JOIN
(SELECT "user_groups"."user_id"
FROM "user_groups"
JOIN
(SELECT "id"
FROM "groups"
WHERE "name" = $1) AS "t1" ON "user_groups"."group_id" = "t1"."id") AS "t1" ON "users"."id" = "t1"."user_id"`,
wantArgs: []interface{}{"GitHub"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
selector := SetNeighbors("postgres", tt.input)
query, args := selector.Query()
tt.wantQuery = strings.Join(strings.Fields(tt.wantQuery), " ")
require.Equal(t, tt.wantQuery, query)
require.Equal(t, tt.wantArgs, args)
})
}
}