1 files changed, 104 insertions, 0 deletions
diff --git a/internal/oauth/pgclientstore_test.go b/internal/oauth/pgclientstore_test.go
new file mode 100644
index 000000000..3f8de064d
--- /dev/null
+++ b/internal/oauth/pgclientstore_test.go
@@ -0,0 +1,104 @@
+package oauth
+
+import (
+	"context"
+	"testing"
+
+	"github.com/go-pg/pg/v10"
+	"github.com/go-pg/pg/v10/orm"
+	"github.com/gotosocial/oauth2/v4/models"
+	"github.com/sirupsen/logrus"
+	"github.com/stretchr/testify/suite"
+)
+
+type PgClientStoreTestSuite struct {
+	suite.Suite
+	conn           *pg.DB
+	testClientID      string
+	testClientSecret  string
+	testClientDomain  string
+	testClientUserID string
+}
+
+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.testClientSecret = "test-client-secret"
+	suite.testClientDomain = "https://example.org"
+	suite.testClientUserID = "test-client-user-id"
+}
+
+// SetupTest creates a postgres connection and creates the oauth_clients table before each test
+func (suite *PgClientStoreTestSuite) SetupTest() {
+	suite.conn = pg.Connect(&pg.Options{})
+	if err := suite.conn.Ping(context.Background()); err != nil {
+		logrus.Panicf("db connection error: %s", err)
+	}
+	if err := suite.conn.Model(&oauthClient{}).CreateTable(&orm.CreateTableOptions{
+		IfNotExists: true,
+	}); err != nil {
+		logrus.Panicf("db connection error: %s", err)
+	}
+}
+
+// TearDownTest drops the oauth_clients table and closes the pg connection after each test
+func (suite *PgClientStoreTestSuite) TearDownTest() {
+	if err := suite.conn.Model(&oauthClient{}).DropTable(&orm.DropTableOptions{}); err != nil {
+		logrus.Panicf("drop table error: %s", err)
+	}
+	if err := suite.conn.Close(); err != nil {
+		logrus.Panicf("error closing db connection: %s", err)
+	}
+	suite.conn = nil
+}
+
+func (suite *PgClientStoreTestSuite) TestClientStoreSetAndGet() {
+	// set a new client in the store
+	cs := NewPGClientStore(suite.conn)
+	if err := cs.Set(context.Background(), suite.testClientID, models.New(suite.testClientID, suite.testClientSecret, suite.testClientDomain, suite.testClientUserID)); err != nil {
+		suite.FailNow(err.Error())
+	}
+
+	// fetch that client from the store
+	client, err := cs.GetByID(context.Background(), suite.testClientID)
+	if err != nil {
+		suite.FailNow(err.Error())
+	}
+
+	// check that the values are the same
+	suite.NotNil(client)
+	suite.EqualValues(models.New(suite.testClientID, suite.testClientSecret, suite.testClientDomain, suite.testClientUserID), client)
+}
+
+func (suite *PgClientStoreTestSuite) TestClientSetAndDelete() {
+	// set a new client in the store
+	cs := NewPGClientStore(suite.conn)
+	if err := cs.Set(context.Background(), suite.testClientID, models.New(suite.testClientID, suite.testClientSecret, suite.testClientDomain, suite.testClientUserID)); err != nil {
+		suite.FailNow(err.Error())
+	}
+
+	// fetch the client from the store
+	client, err := cs.GetByID(context.Background(), suite.testClientID)
+	if err != nil {
+		suite.FailNow(err.Error())
+	}
+
+	// check that the values are the same
+	suite.NotNil(client)
+	suite.EqualValues(models.New(suite.testClientID, suite.testClientSecret, suite.testClientDomain, suite.testClientUserID), client)
+	if err := cs.Delete(context.Background(), suite.testClientID); err != nil {
+		suite.FailNow(err.Error())
+	}
+
+	// try to get the deleted client; we should get an error
+	deletedClient, err := cs.GetByID(context.Background(), suite.testClientID)
+	suite.Assert().Nil(deletedClient)
+	suite.Assert().NotNil(err)
+	suite.EqualValues("pg: no rows in result set", err.Error())
+}
+
+func TestPgClientStoreTestSuite(t *testing.T) {
+	suite.Run(t, new(PgClientStoreTestSuite))
+}