50 lines
1.7 KiB
Smarty
50 lines
1.7 KiB
Smarty
|
{{- $tableNameSingular := titleCaseSingular .Table.Name -}}
|
||
|
{{- $dbName := singular .Table.Name -}}
|
||
|
{{- $tableNamePlural := titleCasePlural .Table.Name -}}
|
||
|
{{- $varNamePlural := camelCasePlural .Table.Name -}}
|
||
|
{{- $varNameSingular := camelCaseSingular .Table.Name -}}
|
||
|
func Test{{$tableNamePlural}}InPrimaryKeyArgs(t *testing.T) {
|
||
|
var err error
|
||
|
var o {{$tableNameSingular}}
|
||
|
o = {{$tableNameSingular}}{}
|
||
|
|
||
|
if err = boil.RandomizeStruct(&o); err != nil {
|
||
|
t.Errorf("Could not randomize struct: %s", err)
|
||
|
}
|
||
|
|
||
|
args := o.inPrimaryKeyArgs()
|
||
|
|
||
|
if len(args) != len({{$varNameSingular}}PrimaryKeyColumns) {
|
||
|
t.Errorf("Expected args to be len %d, but got %d", len({{$varNameSingular}}PrimaryKeyColumns), len(args))
|
||
|
}
|
||
|
|
||
|
{{range $key, $value := .Table.PKey.Columns -}}
|
||
|
if o.{{titleCase $value}} != args[{{$key}}] {
|
||
|
t.Errorf("Expected args[{{$key}}] to be value of o.{{titleCase $value}}, but got %#v", args[{{$key}}])
|
||
|
}
|
||
|
{{- end}}
|
||
|
}
|
||
|
|
||
|
func Test{{$tableNamePlural}}SliceInPrimaryKeyArgs(t *testing.T) {
|
||
|
var err error
|
||
|
o := make({{$varNameSingular}}Slice, 3)
|
||
|
|
||
|
if err = boil.RandomizeSlice(&o); err != nil {
|
||
|
t.Errorf("Could not randomize slice: %s", err)
|
||
|
}
|
||
|
|
||
|
args := o.inPrimaryKeyArgs()
|
||
|
|
||
|
if len(args) != len({{$varNameSingular}}PrimaryKeyColumns) * 3 {
|
||
|
t.Errorf("Expected args to be len %d, but got %d", len({{$varNameSingular}}PrimaryKeyColumns) * 3, len(args))
|
||
|
}
|
||
|
|
||
|
for i := 0; i < len({{$varNameSingular}}PrimaryKeyColumns) * 3; i++ {
|
||
|
{{range $key, $value := .Table.PKey.Columns -}}
|
||
|
if o[i].{{titleCase $value}} != args[i] {
|
||
|
t.Errorf("Expected args[%d] to be value of o.{{titleCase $value}}, but got %#v", i, args[i])
|
||
|
}
|
||
|
}
|
||
|
{{- end}}
|
||
|
}
|