diff options
Diffstat (limited to 'internal/api/client/polls/polls_test.go')
| -rw-r--r-- | internal/api/client/polls/polls_test.go | 25 |
1 files changed, 16 insertions, 9 deletions
diff --git a/internal/api/client/polls/polls_test.go b/internal/api/client/polls/polls_test.go index b5552f39e..8c2bc8ba1 100644 --- a/internal/api/client/polls/polls_test.go +++ b/internal/api/client/polls/polls_test.go @@ -36,14 +36,15 @@ import ( type PollsStandardTestSuite struct { suite.Suite - db db.DB - storage *storage.Driver - mediaManager *media.Manager - federator *federation.Federator - processor *processing.Processor - emailSender email.Sender - sentEmails map[string]string - state state.State + db db.DB + storage *storage.Driver + mediaManager *media.Manager + federator *federation.Federator + processor *processing.Processor + emailSender email.Sender + sentEmails map[string]string + webPushSender *testrig.WebPushMockSender + state state.State // standard suite models testTokens map[string]*gtsmodel.Token @@ -91,7 +92,13 @@ func (suite *PollsStandardTestSuite) SetupTest() { suite.federator = testrig.NewTestFederator(&suite.state, testrig.NewTestTransportController(&suite.state, testrig.NewMockHTTPClient(nil, "../../../../testrig/media")), suite.mediaManager) suite.sentEmails = make(map[string]string) suite.emailSender = testrig.NewEmailSender("../../../../web/template/", suite.sentEmails) - suite.processor = testrig.NewTestProcessor(&suite.state, suite.federator, suite.emailSender, suite.mediaManager) + suite.processor = testrig.NewTestProcessor( + &suite.state, + suite.federator, + suite.emailSender, + testrig.NewNoopWebPushSender(), + suite.mediaManager, + ) suite.pollsModule = polls.New(suite.processor) testrig.StandardDBSetup(suite.db, nil) testrig.StandardStorageSetup(suite.storage, "../../../../testrig/media") |