98 lines
3.6 KiB
Smarty
98 lines
3.6 KiB
Smarty
{{- $tableNameSingular := .Table.Name | singular | titleCase -}}
|
|
{{- $tableNamePlural := .Table.Name | plural | titleCase -}}
|
|
{{- $varNamePlural := .Table.Name | plural | camelCase -}}
|
|
{{- $varNameSingular := .Table.Name | singular | camelCase -}}
|
|
func Test{{$tableNamePlural}}Upsert(t *testing.T) {
|
|
var err error
|
|
|
|
o := {{$tableNameSingular}}{}
|
|
|
|
columns := o.generateUpsertColumns([]string{"one", "two"}, []string{"three", "four"}, []string{"five", "six"})
|
|
if columns.conflict[0] != "one" || columns.conflict[1] != "two" {
|
|
t.Errorf("Expected conflict to be %v, got %v", []string{"one", "two"}, columns.conflict)
|
|
}
|
|
|
|
if columns.update[0] != "three" || columns.update[1] != "four" {
|
|
t.Errorf("Expected update to be %v, got %v", []string{"three", "four"}, columns.update)
|
|
}
|
|
|
|
if columns.whitelist[0] != "five" || columns.whitelist[1] != "six" {
|
|
t.Errorf("Expected whitelist to be %v, got %v", []string{"five", "six"}, columns.whitelist)
|
|
}
|
|
|
|
columns = o.generateUpsertColumns(nil, nil, nil)
|
|
if len(columns.whitelist) == 0 {
|
|
t.Errorf("Expected whitelist to contain columns, but got len 0")
|
|
}
|
|
|
|
if len(columns.conflict) == 0 {
|
|
t.Errorf("Expected conflict to contain columns, but got len 0")
|
|
}
|
|
|
|
if len(columns.update) == 0 {
|
|
t.Errorf("expected update to contain columns, but got len 0")
|
|
}
|
|
|
|
upsertCols := upsertData{
|
|
conflict: []string{"key1", `"key2"`},
|
|
update: []string{"aaa", `"bbb"`},
|
|
whitelist: []string{"thing", `"stuff"`},
|
|
returning: []string{},
|
|
}
|
|
|
|
query := o.generateUpsertQuery(false, upsertCols)
|
|
expectedQuery := `INSERT INTO {{.Table.Name}} ("thing", "stuff") VALUES ($1,$2) ON CONFLICT DO NOTHING`
|
|
|
|
if query != expectedQuery {
|
|
t.Errorf("Expected query mismatch:\n\n%s\n%s\n", query, expectedQuery)
|
|
}
|
|
|
|
query = o.generateUpsertQuery(true, upsertCols)
|
|
expectedQuery = `INSERT INTO {{.Table.Name}} ("thing", "stuff") VALUES ($1,$2) ON CONFLICT ("key1", "key2") DO UPDATE SET "aaa" = EXCLUDED."aaa", "bbb" = EXCLUDED."bbb"`
|
|
|
|
if query != expectedQuery {
|
|
t.Errorf("Expected query mismatch:\n\n%s\n%s\n", query, expectedQuery)
|
|
}
|
|
|
|
upsertCols.returning = []string{"stuff"}
|
|
query = o.generateUpsertQuery(true, upsertCols)
|
|
expectedQuery = expectedQuery + ` RETURNING stuff`
|
|
|
|
if query != expectedQuery {
|
|
t.Errorf("Expected query mismatch:\n\n%s\n%s\n", query, expectedQuery)
|
|
}
|
|
|
|
// Attempt the INSERT side of an UPSERT
|
|
if err = boil.RandomizeStruct(&o, {{$varNameSingular}}DBTypes, true); err != nil {
|
|
t.Errorf("Unable to randomize {{$tableNameSingular}} struct: %s", err)
|
|
}
|
|
|
|
if err = o.UpsertG(false, nil, nil); err != nil {
|
|
t.Errorf("Unable to upsert {{$tableNameSingular}}: %s", err)
|
|
}
|
|
|
|
compare, err := {{$tableNameSingular}}FindG({{.Table.PKey.Columns | stringMap .StringFuncs.titleCase | prefixStringSlice "o." | join ", "}})
|
|
if err != nil {
|
|
t.Errorf("Unable to find {{$tableNameSingular}}: %s", err)
|
|
}
|
|
|
|
{{$varNameSingular}}CompareVals(&o, compare, t)
|
|
|
|
// Attempt the UPDATE side of an UPSERT
|
|
if err = boil.RandomizeStruct(&o, {{$varNameSingular}}DBTypes, false, {{$varNameSingular}}PrimaryKeyColumns...); err != nil {
|
|
t.Errorf("Unable to randomize {{$tableNameSingular}} struct: %s", err)
|
|
}
|
|
|
|
if err = o.UpsertG(true, nil, nil); err != nil {
|
|
t.Errorf("Unable to upsert {{$tableNameSingular}}: %s", err)
|
|
}
|
|
|
|
compare, err = {{$tableNameSingular}}FindG({{.Table.PKey.Columns | stringMap .StringFuncs.titleCase | prefixStringSlice "o." | join ", "}})
|
|
if err != nil {
|
|
t.Errorf("Unable to find {{$tableNameSingular}}: %s", err)
|
|
}
|
|
|
|
{{$varNameSingular}}CompareVals(&o, compare, t)
|
|
|
|
{{$varNamePlural}}DeleteAllRows(t)
|
|
}
|