diff options
author | 2021-08-12 21:03:24 +0200 | |
---|---|---|
committer | 2021-08-12 21:03:24 +0200 | |
commit | 98263a7de64269898a2f81207e38943b5c8e8653 (patch) | |
tree | 743c90f109a6c5d27832d1dcef2388d939f0f77a /internal/oauth/clientstore_test.go | |
parent | Text duplication fix (#137) (diff) | |
download | gotosocial-98263a7de64269898a2f81207e38943b5c8e8653.tar.xz |
Grand test fixup (#138)
* start fixing up tests
* fix up tests + automate with drone
* fiddle with linting
* messing about with drone.yml
* some more fiddling
* hmmm
* add cache
* add vendor directory
* verbose
* ci updates
* update some little things
* update sig
Diffstat (limited to 'internal/oauth/clientstore_test.go')
-rw-r--r-- | internal/oauth/clientstore_test.go | 49 |
1 files changed, 5 insertions, 44 deletions
diff --git a/internal/oauth/clientstore_test.go b/internal/oauth/clientstore_test.go index 58c5148b2..c515ff513 100644 --- a/internal/oauth/clientstore_test.go +++ b/internal/oauth/clientstore_test.go @@ -21,12 +21,10 @@ import ( "context" "testing" - "github.com/sirupsen/logrus" "github.com/stretchr/testify/suite" - "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" - "github.com/superseriousbusiness/gotosocial/internal/db/pg" "github.com/superseriousbusiness/gotosocial/internal/oauth" + "github.com/superseriousbusiness/gotosocial/testrig" "github.com/superseriousbusiness/oauth2/v4/models" ) @@ -43,7 +41,7 @@ const () // SetupSuite sets some variables on the suite that we can use as consts (more or less) throughout func (suite *PgClientStoreTestSuite) SetupSuite() { - suite.testClientID = "test-client-id" + suite.testClientID = "01FCVB74EW6YBYAEY7QG9CQQF6" suite.testClientSecret = "test-client-secret" suite.testClientDomain = "https://example.org" suite.testClientUserID = "test-client-user-id" @@ -51,50 +49,13 @@ func (suite *PgClientStoreTestSuite) SetupSuite() { // SetupTest creates a postgres connection and creates the oauth_clients table before each test func (suite *PgClientStoreTestSuite) SetupTest() { - log := logrus.New() - log.SetLevel(logrus.TraceLevel) - c := config.Empty() - c.DBConfig = &config.DBConfig{ - Type: "postgres", - Address: "localhost", - Port: 5432, - User: "postgres", - Password: "postgres", - Database: "postgres", - ApplicationName: "gotosocial", - } - db, err := pg.NewPostgresService(context.Background(), c, log) - if err != nil { - logrus.Panicf("error creating database connection: %s", err) - } - - suite.db = db - - models := []interface{}{ - &oauth.Client{}, - } - - for _, m := range models { - if err := suite.db.CreateTable(m); err != nil { - logrus.Panicf("db connection error: %s", err) - } - } + suite.db = testrig.NewTestDB() + testrig.StandardDBSetup(suite.db, nil) } // TearDownTest drops the oauth_clients table and closes the pg connection after each test func (suite *PgClientStoreTestSuite) TearDownTest() { - models := []interface{}{ - &oauth.Client{}, - } - for _, m := range models { - if err := suite.db.DropTable(m); err != nil { - logrus.Panicf("error dropping table: %s", err) - } - } - if err := suite.db.Stop(context.Background()); err != nil { - logrus.Panicf("error closing db connection: %s", err) - } - suite.db = nil + testrig.StandardDBTeardown(suite.db) } func (suite *PgClientStoreTestSuite) TestClientStoreSetAndGet() { |