62 lines
2 KiB
Smarty
62 lines
2 KiB
Smarty
{{- $tableNameSingular := .Table.Name | singular | titleCase -}}
|
|
{{- $tableNamePlural := .Table.Name | plural | titleCase -}}
|
|
{{- $varNamePlural := .Table.Name | plural | camelCase -}}
|
|
{{- $varNameSingular := .Table.Name | singular | camelCase -}}
|
|
var {{$varNameSingular}}DBTypes = map[string]string{{"{"}}{{.Table.Columns | columnDBTypes | makeStringMap}}{{"}"}}
|
|
|
|
func test{{$tableNamePlural}}InPrimaryKeyArgs(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var err error
|
|
var o {{$tableNameSingular}}
|
|
o = {{$tableNameSingular}}{}
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &o, {{$varNameSingular}}DBTypes, true); 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) {
|
|
t.Parallel()
|
|
|
|
var err error
|
|
o := make({{$tableNameSingular}}Slice, 3)
|
|
|
|
seed := randomize.NewSeed()
|
|
for i := range o {
|
|
o[i] = &{{$tableNameSingular}}{}
|
|
if err = randomize.Struct(seed, o[i], {{$varNameSingular}}DBTypes, true); err != nil {
|
|
t.Errorf("Could not randomize struct: %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))
|
|
}
|
|
|
|
argC := 0
|
|
for i := 0; i < 3; i++ {
|
|
{{range $key, $value := .Table.PKey.Columns}}
|
|
if o[i].{{titleCase $value}} != args[argC] {
|
|
t.Errorf("Expected args[%d] to be value of o.{{titleCase $value}}, but got %#v", i, args[i])
|
|
}
|
|
argC++
|
|
{{- end}}
|
|
}
|
|
}
|