clean up some weirdness in the router (#80)

2 files changed, 7 insertions, 3 deletions

diff --git a/testrig/db.go b/testrig/db.go
index 5fa019adc..01cf93934 100644
--- a/testrig/db.go
+++ b/testrig/db.go
@@ -47,6 +47,7 @@ var testModels []interface{} = []interface{}{
 	&gtsmodel.Emoji{},
 	&gtsmodel.Instance{},
 	&gtsmodel.Notification{},
+	&gtsmodel.RouterSession{},
 	&oauth.Token{},
 	&oauth.Client{},
 }
diff --git a/testrig/router.go b/testrig/router.go
index 83ce7b602..5770191ea 100644
--- a/testrig/router.go
+++ b/testrig/router.go
@@ -18,11 +18,14 @@
 
 package testrig
 
-import "github.com/superseriousbusiness/gotosocial/internal/router"
+import (
+	"github.com/superseriousbusiness/gotosocial/internal/db"
+	"github.com/superseriousbusiness/gotosocial/internal/router"
+)
 
 // NewTestRouter returns a Router suitable for testing
-func NewTestRouter() router.Router {
-	r, err := router.New(NewTestConfig(), NewTestLog())
+func NewTestRouter(db db.DB) router.Router {
+	r, err := router.New(NewTestConfig(), db, NewTestLog())
 	if err != nil {
 		panic(err)
 	}