diff options
Diffstat (limited to 'testrig')
| -rw-r--r-- | testrig/db.go | 1 | ||||
| -rw-r--r-- | testrig/router.go | 9 |
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{}{ >smodel.Emoji{}, >smodel.Instance{}, >smodel.Notification{}, + >smodel.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) } |