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

schema/edge: fix edge annotations with to/from builders (#679)

Browse Source
This commit is contained in:
Ariel Mashraki
2020-08-20 12:15:43 +03:00
committed by GitHub
parent b18716931d
commit 5e56543133
4 changed files with 46 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
entc/gen/graph.go
View File

@@ -190,7 +190,7 @@ func (g *Graph) addEdges(schema *load.Schema) {
Unique: ref.Unique,
Optional: !ref.Required,
StructTag: ref.Tag,
Annotations: e.Annotations,
Annotations: ref.Annotations,
})
default:
panic(graphError{"edge must be either an assoc or inverse edge"})

6
entc/gen/graph_test.go
View File

@@ -57,6 +57,7 @@ var (
{Name: "t1_o2m", Type: "T1", RefName: "t2_m2o", Inverse: true},
{Name: "t1_m2o", Type: "T1", RefName: "t2_o2m", Unique: true, Inverse: true},
{Name: "t1_m2m", Type: "T1", RefName: "t2_m2m", Inverse: true},
{Name: "t2_m2m_from", Type: "T2", Ref: &load.Edge{Name: "t2_m2m_to", Type: "T2", Annotations: dict("GQL", map[string]string{"Name": "To"})}, Inverse: true, Annotations: dict("GQL", map[string]string{"Name": "From"})},
},
}
)
@@ -118,6 +119,11 @@ func TestNewGraph(t *testing.T) {
require.True(e1.IsInverse())
require.Equal("t2", e1.Inverse)
require.Equal(graph.Nodes[0], e1.Type)
require.Equal("t2_m2m_from", t2.Edges[5].Name)
require.Equal("t2_m2m_to", t2.Edges[6].Name)
require.Equal(map[string]string{"Name": "From"}, t2.Edges[5].Annotations["GQL"])
require.Equal(map[string]string{"Name": "To"}, t2.Edges[6].Annotations["GQL"])
}
func TestNewGraphRequiredLoop(t *testing.T) {