{{- if ne .DriverName "mssql" -}} {{- $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) { t.Parallel() if len({{$varNameSingular}}Columns) == len({{$varNameSingular}}PrimaryKeyColumns) { t.Skip("Skipping table with only primary key columns") } seed := randomize.NewSeed() var err error // Attempt the INSERT side of an UPSERT {{$varNameSingular}} := {{$tableNameSingular}}{} if err = randomize.Struct(seed, &{{$varNameSingular}}, {{$varNameSingular}}DBTypes, true); err != nil { t.Errorf("Unable to randomize {{$tableNameSingular}} struct: %s", err) } tx := MustTx(boil.Begin()) defer tx.Rollback() if err = {{$varNameSingular}}.Upsert(tx, {{if eq .DriverName "postgres"}}false, nil, {{end}}nil); err != nil { t.Errorf("Unable to upsert {{$tableNameSingular}}: %s", err) } count, err := {{$tableNamePlural}}(tx).Count() if err != nil { t.Error(err) } if count != 1 { t.Error("want one record, got:", count) } // Attempt the UPDATE side of an UPSERT if err = randomize.Struct(seed, &{{$varNameSingular}}, {{$varNameSingular}}DBTypes, false, {{$varNameSingular}}PrimaryKeyColumns...); err != nil { t.Errorf("Unable to randomize {{$tableNameSingular}} struct: %s", err) } if err = {{$varNameSingular}}.Upsert(tx, {{if eq .DriverName "postgres"}}true, nil, {{end}}nil); err != nil { t.Errorf("Unable to upsert {{$tableNameSingular}}: %s", err) } count, err = {{$tableNamePlural}}(tx).Count() if err != nil { t.Error(err) } if count != 1 { t.Error("want one record, got:", count) } } {{- end}}