sqlboiler/templates_test/delete.tpl

121 lines
3.5 KiB
Smarty
Raw Normal View History

{{- $tableNameSingular := .Table.Name | singular | titleCase -}}
{{- $dbName := singular .Table.Name -}}
{{- $tableNamePlural := .Table.Name | plural | titleCase -}}
{{- $varNamePlural := .Table.Name | plural | camelCase -}}
{{- $varNameSingular := .Table.Name | singular | camelCase -}}
func {{$varNamePlural}}DeleteAllRows(t *testing.T) {
// Delete all rows to give a clean slate
2016-08-01 07:10:10 +02:00
err := {{$tableNamePlural}}G().DeleteAll()
if err != nil {
t.Errorf("Unable to delete all from {{$tableNamePlural}}: %s", err)
}
}
2016-06-10 07:38:58 +02:00
func Test{{$tableNamePlural}}QueryDeleteAll(t *testing.T) {
var err error
var c int64
// Start from a clean slate
{{$varNamePlural}}DeleteAllRows(t)
// Check number of rows in table to ensure DeleteAll was successful
2016-08-01 07:10:10 +02:00
c, err = {{$tableNamePlural}}G().Count()
if c != 0 {
t.Errorf("Expected 0 rows after ObjDeleteAllRows() call, but got %d rows", c)
}
o := make({{$tableNameSingular}}Slice, 3)
2016-07-14 07:09:05 +02:00
if err = boil.RandomizeSlice(&o, {{$varNameSingular}}DBTypes, true); err != nil {
t.Errorf("Unable to randomize {{$tableNameSingular}} slice: %s", err)
}
// insert random columns to test DeleteAll
for i := 0; i < len(o); i++ {
2016-08-01 07:10:10 +02:00
err = o[i].InsertG()
if err != nil {
t.Errorf("Unable to insert {{$tableNameSingular}}:\n%#v\nErr: %s", o[i], err)
}
}
2016-06-10 04:09:48 +02:00
// Test DeleteAll() query function
2016-08-01 07:10:10 +02:00
err = {{$tableNamePlural}}G().DeleteAll()
if err != nil {
t.Errorf("Unable to delete all from {{$tableNamePlural}}: %s", err)
}
// Check number of rows in table to ensure DeleteAll was successful
2016-08-01 07:10:10 +02:00
c, err = {{$tableNamePlural}}G().Count()
if c != 0 {
t.Errorf("Expected 0 rows after Obj().DeleteAll() call, but got %d rows", c)
}
2016-06-10 07:38:58 +02:00
}
func Test{{$tableNamePlural}}SliceDeleteAll(t *testing.T) {
var err error
var c int64
2016-06-10 04:09:48 +02:00
// insert random columns to test DeleteAll
o := make({{$tableNameSingular}}Slice, 3)
2016-07-14 07:09:05 +02:00
if err = boil.RandomizeSlice(&o, {{$varNameSingular}}DBTypes, true); err != nil {
t.Errorf("Unable to randomize {{$tableNameSingular}} slice: %s", err)
}
2016-06-10 04:09:48 +02:00
for i := 0; i < len(o); i++ {
2016-08-01 07:10:10 +02:00
err = o[i].InsertG()
2016-06-10 04:09:48 +02:00
if err != nil {
t.Errorf("Unable to insert {{$tableNameSingular}}:\n%#v\nErr: %s", o[i], err)
}
}
// test DeleteAll slice function
2016-08-01 07:10:10 +02:00
if err = o.DeleteAllG(); err != nil {
t.Errorf("Unable to objSlice.DeleteAll(): %s", err)
}
2016-06-10 04:09:48 +02:00
// Check number of rows in table to ensure DeleteAll was successful
2016-08-01 07:10:10 +02:00
c, err = {{$tableNamePlural}}G().Count()
2016-06-10 04:09:48 +02:00
if c != 0 {
t.Errorf("Expected 0 rows after objSlice.DeleteAll() call, but got %d rows", c)
2016-06-10 04:09:48 +02:00
}
2016-06-10 07:38:58 +02:00
}
func Test{{$tableNamePlural}}Delete(t *testing.T) {
var err error
var c int64
2016-06-10 04:09:48 +02:00
// insert random columns to test Delete
o := make({{$tableNameSingular}}Slice, 3)
2016-07-14 07:09:05 +02:00
if err = boil.RandomizeSlice(&o, {{$varNameSingular}}DBTypes, true); err != nil {
t.Errorf("Unable to randomize {{$tableNameSingular}} slice: %s", err)
}
for i := 0; i < len(o); i++ {
2016-08-01 07:10:10 +02:00
err = o[i].InsertG()
if err != nil {
t.Errorf("Unable to insert {{$tableNameSingular}}:\n%#v\nErr: %s", o[i], err)
}
}
2016-08-01 07:10:10 +02:00
o[0].DeleteG()
2016-06-10 04:09:48 +02:00
// Check number of rows in table to ensure DeleteAll was successful
2016-08-01 07:10:10 +02:00
c, err = {{$tableNamePlural}}G().Count()
2016-06-10 04:09:48 +02:00
if c != 2 {
t.Errorf("Expected 2 rows after obj.Delete() call, but got %d rows", c)
2016-06-10 04:09:48 +02:00
}
2016-06-10 07:38:58 +02:00
2016-08-01 07:10:10 +02:00
o[1].DeleteG()
o[2].DeleteG()
2016-06-10 07:38:58 +02:00
// Check number of rows in table to ensure Delete worked for all rows
2016-08-01 07:10:10 +02:00
c, err = {{$tableNamePlural}}G().Count()
2016-06-10 07:38:58 +02:00
if c != 0 {
t.Errorf("Expected 0 rows after all obj.Delete() calls, but got %d rows", c)
}
}