mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-05 14:40:42 +01:00
122 lines
3.6 KiB
Go
122 lines
3.6 KiB
Go
|
// 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 testrig
|
||
|
|
||
|
import (
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/cleaner"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/email"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/filter/interaction"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/filter/visibility"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/processing"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/processing/common"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/state"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/typeutils"
|
||
|
)
|
||
|
|
||
|
// TestStructs encapsulates structs needed to
|
||
|
// run one test independently. Each test should
|
||
|
// call SetupTestStructs to get a new TestStructs,
|
||
|
// and defer TearDownTestStructs to close it when
|
||
|
// the test is complete. The reason for doing things
|
||
|
// this way here is to prevent the tests in a
|
||
|
// package from overwriting one another's processors
|
||
|
// and worker queues, which was causing issues
|
||
|
// when running all tests at once.
|
||
|
type TestStructs struct {
|
||
|
State *state.State
|
||
|
Common *common.Processor
|
||
|
Processor *processing.Processor
|
||
|
HTTPClient *MockHTTPClient
|
||
|
TypeConverter *typeutils.Converter
|
||
|
EmailSender email.Sender
|
||
|
}
|
||
|
|
||
|
func SetupTestStructs(
|
||
|
rMediaPath string,
|
||
|
rTemplatePath string,
|
||
|
) *TestStructs {
|
||
|
state := state.State{}
|
||
|
|
||
|
state.Caches.Init()
|
||
|
|
||
|
db := NewTestDB(&state)
|
||
|
state.DB = db
|
||
|
|
||
|
storage := NewInMemoryStorage()
|
||
|
state.Storage = storage
|
||
|
typeconverter := typeutils.NewConverter(&state)
|
||
|
visFilter := visibility.NewFilter(&state)
|
||
|
intFilter := interaction.NewFilter(&state)
|
||
|
|
||
|
StartTimelines(
|
||
|
&state,
|
||
|
visFilter,
|
||
|
typeconverter,
|
||
|
)
|
||
|
|
||
|
httpClient := NewMockHTTPClient(nil, rMediaPath)
|
||
|
httpClient.TestRemotePeople = NewTestFediPeople()
|
||
|
httpClient.TestRemoteStatuses = NewTestFediStatuses()
|
||
|
|
||
|
transportController := NewTestTransportController(&state, httpClient)
|
||
|
mediaManager := NewTestMediaManager(&state)
|
||
|
federator := NewTestFederator(&state, transportController, mediaManager)
|
||
|
oauthServer := NewTestOauthServer(db)
|
||
|
emailSender := NewEmailSender(rTemplatePath, nil)
|
||
|
|
||
|
common := common.New(
|
||
|
&state,
|
||
|
mediaManager,
|
||
|
typeconverter,
|
||
|
federator,
|
||
|
visFilter,
|
||
|
)
|
||
|
|
||
|
processor := processing.NewProcessor(
|
||
|
cleaner.New(&state),
|
||
|
typeconverter,
|
||
|
federator,
|
||
|
oauthServer,
|
||
|
mediaManager,
|
||
|
&state,
|
||
|
emailSender,
|
||
|
visFilter,
|
||
|
intFilter,
|
||
|
)
|
||
|
|
||
|
StartWorkers(&state, processor.Workers())
|
||
|
|
||
|
StandardDBSetup(db, nil)
|
||
|
StandardStorageSetup(storage, rMediaPath)
|
||
|
|
||
|
return &TestStructs{
|
||
|
State: &state,
|
||
|
Common: &common,
|
||
|
Processor: processor,
|
||
|
HTTPClient: httpClient,
|
||
|
TypeConverter: typeconverter,
|
||
|
EmailSender: emailSender,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TearDownTestStructs(testStructs *TestStructs) {
|
||
|
StandardDBTeardown(testStructs.State.DB)
|
||
|
StandardStorageTeardown(testStructs.State.Storage)
|
||
|
StopWorkers(testStructs.State)
|
||
|
}
|