gtsocial-umbx

Unnamed repository; edit this file 'description' to name the repository.
Log | Files | Refs | README | LICENSE

media_test.go (2815B)

      1 // GoToSocial
      2 // Copyright (C) GoToSocial Authors admin@gotosocial.org
      3 // SPDX-License-Identifier: AGPL-3.0-or-later
      4 //
      5 // This program is free software: you can redistribute it and/or modify
      6 // it under the terms of the GNU Affero General Public License as published by
      7 // the Free Software Foundation, either version 3 of the License, or
      8 // (at your option) any later version.
      9 //
     10 // This program is distributed in the hope that it will be useful,
     11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
     12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     13 // GNU Affero General Public License for more details.
     14 //
     15 // You should have received a copy of the GNU Affero General Public License
     16 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
     17 
     18 package media_test
     19 
     20 import (
     21 	"github.com/stretchr/testify/suite"
     22 	"github.com/superseriousbusiness/gotosocial/internal/db"
     23 	gtsmodel "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
     24 	"github.com/superseriousbusiness/gotosocial/internal/media"
     25 	"github.com/superseriousbusiness/gotosocial/internal/state"
     26 	"github.com/superseriousbusiness/gotosocial/internal/storage"
     27 	"github.com/superseriousbusiness/gotosocial/internal/transport"
     28 	"github.com/superseriousbusiness/gotosocial/internal/visibility"
     29 	"github.com/superseriousbusiness/gotosocial/testrig"
     30 )
     31 
     32 type MediaStandardTestSuite struct {
     33 	suite.Suite
     34 
     35 	db                  db.DB
     36 	storage             *storage.Driver
     37 	state               state.State
     38 	manager             *media.Manager
     39 	transportController transport.Controller
     40 	testAttachments     map[string]*gtsmodel.MediaAttachment
     41 	testAccounts        map[string]*gtsmodel.Account
     42 	testEmojis          map[string]*gtsmodel.Emoji
     43 }
     44 
     45 func (suite *MediaStandardTestSuite) SetupTest() {
     46 	testrig.InitTestConfig()
     47 	testrig.InitTestLog()
     48 
     49 	suite.state.Caches.Init()
     50 	testrig.StartWorkers(&suite.state)
     51 
     52 	suite.db = testrig.NewTestDB(&suite.state)
     53 	suite.storage = testrig.NewInMemoryStorage()
     54 	suite.state.DB = suite.db
     55 	suite.state.Storage = suite.storage
     56 
     57 	testrig.StandardStorageSetup(suite.storage, "../../testrig/media")
     58 	testrig.StandardDBSetup(suite.db, nil)
     59 
     60 	testrig.StartTimelines(
     61 		&suite.state,
     62 		visibility.NewFilter(&suite.state),
     63 		testrig.NewTestTypeConverter(suite.db),
     64 	)
     65 
     66 	suite.testAttachments = testrig.NewTestAttachments()
     67 	suite.testAccounts = testrig.NewTestAccounts()
     68 	suite.testEmojis = testrig.NewTestEmojis()
     69 	suite.manager = testrig.NewTestMediaManager(&suite.state)
     70 	suite.transportController = testrig.NewTestTransportController(&suite.state, testrig.NewMockHTTPClient(nil, "../../testrig/media"))
     71 }
     72 
     73 func (suite *MediaStandardTestSuite) TearDownTest() {
     74 	testrig.StandardDBTeardown(suite.db)
     75 	testrig.StandardStorageTeardown(suite.storage)
     76 	testrig.StopWorkers(&suite.state)
     77 }