Fix create tests

jamiefdhurst · Aug 3, 2018 · 5c5983e · 5c5983e
1 parent baa92bf
commit 5c5983e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/app/controller/apiv1/create_test.go b/internal/app/controller/apiv1/create_test.go
@@ -13,6 +13,8 @@ import (
 
 func TestCreate_Run(t *testing.T) {
 	db := &database.MockSqlite{}
+	db.Result = &database.MockResult{}
+	db.Rows = &database.MockRowsEmpty{}
 	container := &app.Container{Db: db}
 	response := controller.NewMockResponse()
 	response.Reset()

diff --git a/internal/app/controller/web/new_test.go b/internal/app/controller/web/new_test.go
@@ -13,6 +13,8 @@ import (
 
 func TestNew_Run(t *testing.T) {
 	db := &database.MockSqlite{}
+	db.Result = &database.MockResult{}
+	db.Rows = &database.MockRowsEmpty{}
 	container := &app.Container{Db: db}
 	response := controller.NewMockResponse()
 	controller := &New{}