From baf933cb9f3e1053bdb61b90d7027efe9fad1bc2 Mon Sep 17 00:00:00 2001 From: kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com> Date: Wed, 1 Mar 2023 18:26:53 +0000 Subject: [chore] move client/federator workerpools to Workers{} (#1575) * replace concurrency worker pools with base models in State.Workers, update code and tests accordingly * improve code comment * change back testrig default log level * un-comment-out TestAnnounceTwice() and fix --------- Signed-off-by: kim Reviewed-by: tobi --- internal/processing/stream/stream_test.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'internal/processing/stream/stream_test.go') diff --git a/internal/processing/stream/stream_test.go b/internal/processing/stream/stream_test.go index 907c7e1d0..9e1eb57f2 100644 --- a/internal/processing/stream/stream_test.go +++ b/internal/processing/stream/stream_test.go @@ -24,6 +24,7 @@ import ( "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" "github.com/superseriousbusiness/gotosocial/internal/oauth" "github.com/superseriousbusiness/gotosocial/internal/processing/stream" + "github.com/superseriousbusiness/gotosocial/internal/state" "github.com/superseriousbusiness/gotosocial/testrig" ) @@ -33,19 +34,23 @@ type StreamTestSuite struct { testTokens map[string]*gtsmodel.Token db db.DB oauthServer oauth.Server + state state.State streamProcessor stream.Processor } func (suite *StreamTestSuite) SetupTest() { + suite.state.Caches.Init() + testrig.InitTestLog() testrig.InitTestConfig() suite.testAccounts = testrig.NewTestAccounts() suite.testTokens = testrig.NewTestTokens() - suite.db = testrig.NewTestDB() + suite.db = testrig.NewTestDB(&suite.state) + suite.state.DB = suite.db suite.oauthServer = testrig.NewTestOauthServer(suite.db) - suite.streamProcessor = stream.New(suite.db, suite.oauthServer) + suite.streamProcessor = stream.New(&suite.state, suite.oauthServer) testrig.StandardDBSetup(suite.db, suite.testAccounts) } -- cgit v1.2.3