You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.8 KiB
114 lines
3.8 KiB
// GoToSocial |
|
// Copyright (C) GoToSocial Authors admin@gotosocial.org |
|
// SPDX-License-Identifier: AGPL-3.0-or-later |
|
// |
|
// This program is free software: you can redistribute it and/or modify |
|
// it under the terms of the GNU Affero General Public License as published by |
|
// the Free Software Foundation, either version 3 of the License, or |
|
// (at your option) any later version. |
|
// |
|
// This program is distributed in the hope that it will be useful, |
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
// GNU Affero General Public License for more details. |
|
// |
|
// You should have received a copy of the GNU Affero General Public License |
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
|
package workers_test |
|
|
|
import ( |
|
"context" |
|
|
|
"github.com/stretchr/testify/suite" |
|
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util" |
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" |
|
"github.com/superseriousbusiness/gotosocial/internal/processing" |
|
"github.com/superseriousbusiness/gotosocial/internal/stream" |
|
"github.com/superseriousbusiness/gotosocial/testrig" |
|
) |
|
|
|
const ( |
|
rMediaPath = "../../../testrig/media" |
|
rTemplatePath = "../../../web/template" |
|
) |
|
|
|
type WorkersTestSuite struct { |
|
// standard suite interfaces |
|
suite.Suite |
|
|
|
// standard suite models |
|
testTokens map[string]*gtsmodel.Token |
|
testApplications map[string]*gtsmodel.Application |
|
testUsers map[string]*gtsmodel.User |
|
testAccounts map[string]*gtsmodel.Account |
|
testFollows map[string]*gtsmodel.Follow |
|
testAttachments map[string]*gtsmodel.MediaAttachment |
|
testStatuses map[string]*gtsmodel.Status |
|
testTags map[string]*gtsmodel.Tag |
|
testMentions map[string]*gtsmodel.Mention |
|
testAutheds map[string]*apiutil.Auth |
|
testBlocks map[string]*gtsmodel.Block |
|
testActivities map[string]testrig.ActivityWithSignature |
|
testLists map[string]*gtsmodel.List |
|
testListEntries map[string]*gtsmodel.ListEntry |
|
} |
|
|
|
func (suite *WorkersTestSuite) SetupSuite() { |
|
suite.testTokens = testrig.NewTestTokens() |
|
suite.testApplications = testrig.NewTestApplications() |
|
suite.testUsers = testrig.NewTestUsers() |
|
suite.testAccounts = testrig.NewTestAccounts() |
|
suite.testFollows = testrig.NewTestFollows() |
|
suite.testAttachments = testrig.NewTestAttachments() |
|
suite.testStatuses = testrig.NewTestStatuses() |
|
suite.testTags = testrig.NewTestTags() |
|
suite.testMentions = testrig.NewTestMentions() |
|
suite.testAutheds = map[string]*apiutil.Auth{ |
|
"local_account_1": { |
|
Application: suite.testApplications["local_account_1"], |
|
User: suite.testUsers["local_account_1"], |
|
Account: suite.testAccounts["local_account_1"], |
|
}, |
|
} |
|
suite.testBlocks = testrig.NewTestBlocks() |
|
suite.testLists = testrig.NewTestLists() |
|
suite.testListEntries = testrig.NewTestListEntries() |
|
} |
|
|
|
func (suite *WorkersTestSuite) SetupTest() { |
|
testrig.InitTestConfig() |
|
testrig.InitTestLog() |
|
suite.testActivities = testrig.NewTestActivities(suite.testAccounts) |
|
} |
|
|
|
func (suite *WorkersTestSuite) openStreams(ctx context.Context, processor *processing.Processor, account *gtsmodel.Account, listIDs []string) map[string]*stream.Stream { |
|
streams := make(map[string]*stream.Stream) |
|
|
|
for _, streamType := range []string{ |
|
stream.TimelineHome, |
|
stream.TimelinePublic, |
|
stream.TimelineNotifications, |
|
stream.TimelineDirect, |
|
} { |
|
stream, err := processor.Stream().Open(ctx, account, streamType) |
|
if err != nil { |
|
suite.FailNow(err.Error()) |
|
} |
|
|
|
streams[streamType] = stream |
|
} |
|
|
|
for _, listID := range listIDs { |
|
streamType := stream.TimelineList + ":" + listID |
|
|
|
stream, err := processor.Stream().Open(ctx, account, streamType) |
|
if err != nil { |
|
suite.FailNow(err.Error()) |
|
} |
|
|
|
streams[streamType] = stream |
|
} |
|
|
|
return streams |
|
}
|
|
|