summaryrefslogtreecommitdiff
path: root/testrig/federatingdb.go
diff options
context:
space:
mode:
Diffstat (limited to 'testrig/federatingdb.go')
-rw-r--r--testrig/federatingdb.go6
1 files changed, 4 insertions, 2 deletions
diff --git a/testrig/federatingdb.go b/testrig/federatingdb.go
index f01a4e9fe..3f71274ca 100644
--- a/testrig/federatingdb.go
+++ b/testrig/federatingdb.go
@@ -3,9 +3,11 @@ package testrig
import (
"github.com/superseriousbusiness/gotosocial/internal/db"
"github.com/superseriousbusiness/gotosocial/internal/federation/federatingdb"
+ "github.com/superseriousbusiness/gotosocial/internal/messages"
+ "github.com/superseriousbusiness/gotosocial/internal/worker"
)
// NewTestFederatingDB returns a federating DB with the underlying db
-func NewTestFederatingDB(db db.DB) federatingdb.DB {
- return federatingdb.New(db)
+func NewTestFederatingDB(db db.DB, fedWorker *worker.Worker[messages.FromFederator]) federatingdb.DB {
+ return federatingdb.New(db, fedWorker)
}