diff options
author | 2021-09-11 20:12:47 +0100 | |
---|---|---|
committer | 2021-09-11 20:12:47 +0100 | |
commit | e43a46e9822a05dd0345a7676e03285ddf83205e (patch) | |
tree | 72db13159e47d7815d001285fadb8225b1f7a571 /internal/api | |
parent | documentation updates (#211) (diff) | |
download | gotosocial-e43a46e9822a05dd0345a7676e03285ddf83205e.tar.xz |
add git.iim.gay/grufwub/go-store for storage backend, replacing blob.Storage
Signed-off-by: kim (grufwub) <grufwub@gmail.com>
Diffstat (limited to 'internal/api')
-rw-r--r-- | internal/api/client/account/account_test.go | 4 | ||||
-rw-r--r-- | internal/api/client/fileserver/servefile_test.go | 6 | ||||
-rw-r--r-- | internal/api/client/media/mediacreate_test.go | 19 | ||||
-rw-r--r-- | internal/api/client/status/status_test.go | 4 | ||||
-rw-r--r-- | internal/api/s2s/user/user_test.go | 4 |
5 files changed, 23 insertions, 14 deletions
diff --git a/internal/api/client/account/account_test.go b/internal/api/client/account/account_test.go index 56f981ab1..0acc611df 100644 --- a/internal/api/client/account/account_test.go +++ b/internal/api/client/account/account_test.go @@ -6,11 +6,11 @@ import ( "net/http" "net/http/httptest" + "git.iim.gay/grufwub/go-store/kv" "github.com/gin-gonic/gin" "github.com/sirupsen/logrus" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/api/client/account" - "github.com/superseriousbusiness/gotosocial/internal/blob" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/federation" @@ -28,7 +28,7 @@ type AccountStandardTestSuite struct { db db.DB log *logrus.Logger tc typeutils.TypeConverter - storage blob.Storage + storage *kv.KVStore federator federation.Federator processor processing.Processor diff --git a/internal/api/client/fileserver/servefile_test.go b/internal/api/client/fileserver/servefile_test.go index 579bb9606..6d5f2d39e 100644 --- a/internal/api/client/fileserver/servefile_test.go +++ b/internal/api/client/fileserver/servefile_test.go @@ -26,12 +26,12 @@ import ( "net/http/httptest" "testing" + "git.iim.gay/grufwub/go-store/kv" "github.com/gin-gonic/gin" "github.com/sirupsen/logrus" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/api/client/fileserver" - "github.com/superseriousbusiness/gotosocial/internal/blob" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/federation" @@ -49,7 +49,7 @@ type ServeFileTestSuite struct { config *config.Config db db.DB log *logrus.Logger - storage blob.Storage + storage *kv.KVStore federator federation.Federator tc typeutils.TypeConverter processor processing.Processor @@ -152,7 +152,7 @@ func (suite *ServeFileTestSuite) TestServeOriginalFileSuccessful() { assert.NoError(suite.T(), err) assert.NotNil(suite.T(), b) - fileInStorage, err := suite.storage.RetrieveFileFrom(targetAttachment.File.Path) + fileInStorage, err := suite.storage.Get(targetAttachment.File.Path) assert.NoError(suite.T(), err) assert.NotNil(suite.T(), fileInStorage) assert.Equal(suite.T(), b, fileInStorage) diff --git a/internal/api/client/media/mediacreate_test.go b/internal/api/client/media/mediacreate_test.go index 1b2c84cf9..53c1ce996 100644 --- a/internal/api/client/media/mediacreate_test.go +++ b/internal/api/client/media/mediacreate_test.go @@ -28,13 +28,13 @@ import ( "net/http/httptest" "testing" + "git.iim.gay/grufwub/go-store/kv" "github.com/gin-gonic/gin" "github.com/sirupsen/logrus" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/suite" mediamodule "github.com/superseriousbusiness/gotosocial/internal/api/client/media" "github.com/superseriousbusiness/gotosocial/internal/api/model" - "github.com/superseriousbusiness/gotosocial/internal/blob" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/federation" @@ -52,7 +52,7 @@ type MediaCreateTestSuite struct { config *config.Config db db.DB log *logrus.Logger - storage blob.Storage + storage *kv.KVStore federator federation.Federator tc typeutils.TypeConverter mediaHandler media.Handler @@ -118,7 +118,6 @@ func (suite *MediaCreateTestSuite) TearDownTest() { */ func (suite *MediaCreateTestSuite) TestStatusCreatePOSTImageHandlerSuccessful() { - // set up the context for the request t := suite.testTokens["local_account_1"] oauthToken := oauth.DBTokenToToken(t) @@ -130,10 +129,15 @@ func (suite *MediaCreateTestSuite) TestStatusCreatePOSTImageHandlerSuccessful() ctx.Set(oauth.SessionAuthorizedAccount, suite.testAccounts["local_account_1"]) // see what's in storage *before* the request - storageKeysBeforeRequest, err := suite.storage.ListKeys() + storageKeysBeforeRequest := []string{} + iter, err := suite.storage.Iterator(nil) if err != nil { panic(err) } + for iter.Next() { + storageKeysBeforeRequest = append(storageKeysBeforeRequest, iter.Key()) + } + iter.Release() // create the request buf, w, err := testrig.CreateMultipartFormData("file", "../../../../testrig/media/test-jpeg.jpg", map[string]string{ @@ -150,10 +154,15 @@ func (suite *MediaCreateTestSuite) TestStatusCreatePOSTImageHandlerSuccessful() suite.mediaModule.MediaCreatePOSTHandler(ctx) // check what's in storage *after* the request - storageKeysAfterRequest, err := suite.storage.ListKeys() + storageKeysAfterRequest := []string{} + iter, err = suite.storage.Iterator(nil) if err != nil { panic(err) } + for iter.Next() { + storageKeysAfterRequest = append(storageKeysAfterRequest, iter.Key()) + } + iter.Release() // check response suite.EqualValues(http.StatusOK, recorder.Code) diff --git a/internal/api/client/status/status_test.go b/internal/api/client/status/status_test.go index f9b4e3671..214a00790 100644 --- a/internal/api/client/status/status_test.go +++ b/internal/api/client/status/status_test.go @@ -19,10 +19,10 @@ package status_test import ( + "git.iim.gay/grufwub/go-store/kv" "github.com/sirupsen/logrus" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/api/client/status" - "github.com/superseriousbusiness/gotosocial/internal/blob" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/federation" @@ -41,7 +41,7 @@ type StatusStandardTestSuite struct { tc typeutils.TypeConverter federator federation.Federator processor processing.Processor - storage blob.Storage + storage *kv.KVStore // standard suite models testTokens map[string]*gtsmodel.Token diff --git a/internal/api/s2s/user/user_test.go b/internal/api/s2s/user/user_test.go index ecd5fadc8..269ca4bab 100644 --- a/internal/api/s2s/user/user_test.go +++ b/internal/api/s2s/user/user_test.go @@ -1,11 +1,11 @@ package user_test import ( + "git.iim.gay/grufwub/go-store/kv" "github.com/sirupsen/logrus" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/api/s2s/user" "github.com/superseriousbusiness/gotosocial/internal/api/security" - "github.com/superseriousbusiness/gotosocial/internal/blob" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/federation" @@ -24,7 +24,7 @@ type UserStandardTestSuite struct { tc typeutils.TypeConverter federator federation.Federator processor processing.Processor - storage blob.Storage + storage *kv.KVStore securityModule *security.Module // standard suite models |