sqlboiler/templates_test/relationship_to_many.tpl

101 lines
3.1 KiB
Smarty
Raw Normal View History

2016-07-13 05:05:33 +02:00
{{- if .Table.IsJoinTable -}}
{{- else -}}
{{- $dot := . }}
{{- $table := .Table }}
{{- range .Table.ToManyRelationships -}}
{{- if (and .ForeignColumnUnique (not .ToJoinTable)) -}}
{{- template "relationship_to_one_test_helper" (textsFromOneToOneRelationship $dot.PkgName $dot.Tables $table .) -}}
{{- else -}}
{{- $rel := textsFromRelationship $dot.Tables $table . -}}
func test{{$rel.LocalTable.NameGo}}ToMany{{$rel.Function.Name}}(t *testing.T) {
2016-07-15 08:15:35 +02:00
var err error
tx := MustTx(boil.Begin())
defer tx.Rollback()
var a {{$rel.LocalTable.NameGo}}
var b, c {{$rel.ForeignTable.NameGo}}
2016-07-13 05:05:33 +02:00
2016-08-01 07:10:10 +02:00
if err := a.Insert(tx); err != nil {
2016-07-13 05:05:33 +02:00
t.Fatal(err)
}
2016-08-18 09:06:28 +02:00
seed := randomize.NewSeed()
randomize.Struct(seed, &b, {{$rel.ForeignTable.NameSingular | camelCase}}DBTypes, false, "{{.ForeignColumn}}")
randomize.Struct(seed, &c, {{$rel.ForeignTable.NameSingular | camelCase}}DBTypes, false, "{{.ForeignColumn}}")
{{if .Nullable -}}
a.{{.Column | titleCase}}.Valid = true
{{- end}}
{{- if .ForeignColumnNullable -}}
b.{{.ForeignColumn | titleCase}}.Valid = true
c.{{.ForeignColumn | titleCase}}.Valid = true
{{- end}}
{{if not .ToJoinTable -}}
b.{{$rel.Function.ForeignAssignment}} = a.{{$rel.Function.LocalAssignment}}
c.{{$rel.Function.ForeignAssignment}} = a.{{$rel.Function.LocalAssignment}}
{{- end}}
2016-08-01 07:10:10 +02:00
if err = b.Insert(tx); err != nil {
2016-07-13 05:05:33 +02:00
t.Fatal(err)
}
2016-08-01 07:10:10 +02:00
if err = c.Insert(tx); err != nil {
2016-07-13 05:05:33 +02:00
t.Fatal(err)
}
{{if .ToJoinTable -}}
2016-08-27 21:00:42 +02:00
_, err = tx.Exec(`insert into "{{.JoinTable}}" ({{.JoinLocalColumn}}, {{.JoinForeignColumn}}) values ($1, $2)`, a.{{$rel.LocalTable.ColumnNameGo}}, b.{{$rel.ForeignTable.ColumnNameGo}})
if err != nil {
t.Fatal(err)
}
2016-08-27 21:00:42 +02:00
_, err = tx.Exec(`insert into "{{.JoinTable}}" ({{.JoinLocalColumn}}, {{.JoinForeignColumn}}) values ($1, $2)`, a.{{$rel.LocalTable.ColumnNameGo}}, c.{{$rel.ForeignTable.ColumnNameGo}})
if err != nil {
t.Fatal(err)
}
{{end}}
2016-08-14 09:43:30 +02:00
{{$varname := .ForeignTable | singular | camelCase -}}
{{$varname}}, err := a.{{$rel.Function.Name}}(tx).All()
if err != nil {
t.Fatal(err)
}
bFound, cFound := false, false
for _, v := range {{$varname}} {
if v.{{$rel.Function.ForeignAssignment}} == b.{{$rel.Function.ForeignAssignment}} {
bFound = true
}
if v.{{$rel.Function.ForeignAssignment}} == c.{{$rel.Function.ForeignAssignment}} {
cFound = true
}
}
if !bFound {
t.Error("expected to find b")
}
if !cFound {
t.Error("expected to find c")
}
2016-08-24 07:07:51 +02:00
slice := {{$rel.LocalTable.NameGo}}Slice{&a}
2016-08-27 06:40:11 +02:00
if err = a.R.Load{{$rel.Function.Name}}(tx, false, &slice); err != nil {
t.Fatal(err)
2016-08-16 09:30:08 +02:00
}
2016-08-27 06:40:11 +02:00
if got := len(a.R.{{$rel.Function.Name}}); got != 2 {
t.Error("number of eager loaded records wrong, got:", got)
2016-08-16 09:30:08 +02:00
}
2016-08-27 06:40:11 +02:00
a.R.{{$rel.Function.Name}} = nil
if err = a.R.Load{{$rel.Function.Name}}(tx, true, &a); err != nil {
t.Fatal(err)
2016-08-16 09:30:08 +02:00
}
2016-08-27 06:40:11 +02:00
if got := len(a.R.{{$rel.Function.Name}}); got != 2 {
t.Error("number of eager loaded records wrong, got:", got)
2016-08-16 09:30:08 +02:00
}
if t.Failed() {
t.Logf("%#v", {{$varname}})
}
2016-07-13 05:05:33 +02:00
}
{{end -}}{{- /* if unique */ -}}
{{- end -}}{{- /* range */ -}}
2016-07-13 05:05:33 +02:00
{{- end -}}{{- /* outer if join table */ -}}