/* GoToSocial Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org 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 . */ package status import ( "context" "encoding/json" "fmt" "io/ioutil" "net/http" "net/http/httptest" "net/url" "testing" "github.com/gin-gonic/gin" "github.com/sirupsen/logrus" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/config" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/db/model" "github.com/superseriousbusiness/gotosocial/internal/distributor" "github.com/superseriousbusiness/gotosocial/internal/media" "github.com/superseriousbusiness/gotosocial/internal/oauth" "github.com/superseriousbusiness/gotosocial/internal/storage" "github.com/superseriousbusiness/gotosocial/pkg/mastotypes" "github.com/superseriousbusiness/gotosocial/testrig" ) type StatusCreateTestSuite struct { suite.Suite config *config.Config mockOauthServer *oauth.MockServer mockStorage *storage.MockStorage mediaHandler media.MediaHandler distributor *distributor.MockDistributor testTokens map[string]*oauth.Token testClients map[string]*oauth.Client testApplications map[string]*model.Application testUsers map[string]*model.User testAccounts map[string]*model.Account log *logrus.Logger db db.DB statusModule *statusModule } /* TEST INFRASTRUCTURE */ // SetupSuite sets some variables on the suite that we can use as consts (more or less) throughout func (suite *StatusCreateTestSuite) SetupSuite() { // some of our subsequent entities need a log so create this here log := logrus.New() log.SetLevel(logrus.TraceLevel) suite.log = log // Direct config to local postgres instance c := config.Empty() c.Protocol = "http" c.Host = "localhost" c.DBConfig = &config.DBConfig{ Type: "postgres", Address: "localhost", Port: 5432, User: "postgres", Password: "postgres", Database: "postgres", ApplicationName: "gotosocial", } c.MediaConfig = &config.MediaConfig{ MaxImageSize: 2 << 20, } c.StorageConfig = &config.StorageConfig{ Backend: "local", BasePath: "/tmp", ServeProtocol: "http", ServeHost: "localhost", ServeBasePath: "/fileserver/media", } c.StatusesConfig = &config.StatusesConfig{ MaxChars: 500, CWMaxChars: 50, PollMaxOptions: 4, PollOptionMaxChars: 50, MaxMediaFiles: 4, } suite.config = c // use an actual database for this, because it's just easier than mocking one out database, err := db.New(context.Background(), c, log) if err != nil { suite.FailNow(err.Error()) } suite.db = database suite.mockOauthServer = &oauth.MockServer{} suite.mockStorage = &storage.MockStorage{} suite.mediaHandler = media.New(suite.config, suite.db, suite.mockStorage, log) suite.distributor = &distributor.MockDistributor{} suite.distributor.On("FromClientAPI").Return(make(chan distributor.FromClientAPI, 100)) suite.statusModule = New(suite.config, suite.db, suite.mockOauthServer, suite.mediaHandler, suite.distributor, suite.log).(*statusModule) } func (suite *StatusCreateTestSuite) TearDownSuite() { if err := suite.db.Stop(context.Background()); err != nil { logrus.Panicf("error closing db connection: %s", err) } } func (suite *StatusCreateTestSuite) SetupTest() { if err := testrig.StandardDBSetup(suite.db); err != nil { panic(err) } suite.testTokens = testrig.TestTokens() suite.testClients = testrig.TestClients() suite.testApplications = testrig.TestApplications() suite.testUsers = testrig.TestUsers() suite.testAccounts = testrig.TestAccounts() } // TearDownTest drops tables to make sure there's no data in the db func (suite *StatusCreateTestSuite) TearDownTest() { if err := testrig.StandardDBTeardown(suite.db); err != nil { panic(err) } } /* ACTUAL TESTS */ /* TESTING: StatusCreatePOSTHandler */ func (suite *StatusCreateTestSuite) TestStatusCreatePOSTHandlerSuccessful() { t := suite.testTokens["local_account_1"] oauthToken := oauth.PGTokenToOauthToken(t) // setup recorder := httptest.NewRecorder() ctx, _ := gin.CreateTestContext(recorder) ctx.Set(oauth.SessionAuthorizedApplication, suite.testApplications["application_1"]) ctx.Set(oauth.SessionAuthorizedToken, oauthToken) ctx.Set(oauth.SessionAuthorizedUser, suite.testUsers["local_account_1"]) ctx.Set(oauth.SessionAuthorizedAccount, suite.testAccounts["local_account_1"]) ctx.Request = httptest.NewRequest(http.MethodPost, fmt.Sprintf("http://localhost:8080/%s", basePath), nil) // the endpoint we're hitting ctx.Request.Form = url.Values{ "status": {"this is a brand new status!"}, "spoiler_text": {"hello hello"}, "sensitive": {"true"}, "visibility_advanced": {"mutuals_only"}, "likeable": {"false"}, "replyable": {"false"}, "federated": {"false"}, } suite.statusModule.statusCreatePOSTHandler(ctx) // check response // 1. we should have OK from our call to the function suite.EqualValues(http.StatusOK, recorder.Code) result := recorder.Result() defer result.Body.Close() b, err := ioutil.ReadAll(result.Body) assert.NoError(suite.T(), err) fmt.Println(string(b)) statusReply := &mastotypes.Status{} err = json.Unmarshal(b, statusReply) assert.NoError(suite.T(), err) assert.Equal(suite.T(), "hello hello", statusReply.SpoilerText) assert.Equal(suite.T(), "this is a brand new status!", statusReply.Content) assert.True(suite.T(), statusReply.Sensitive) assert.Equal(suite.T(), mastotypes.VisibilityPrivate, statusReply.Visibility) } func (suite *StatusCreateTestSuite) TestStatusCreatePOSTHandlerReplyToFail() { t := suite.testTokens["local_account_1"] oauthToken := oauth.PGTokenToOauthToken(t) // setup recorder := httptest.NewRecorder() ctx, _ := gin.CreateTestContext(recorder) ctx.Set(oauth.SessionAuthorizedApplication, suite.testApplications["application_1"]) ctx.Set(oauth.SessionAuthorizedToken, oauthToken) ctx.Set(oauth.SessionAuthorizedUser, suite.testUsers["local_account_1"]) ctx.Set(oauth.SessionAuthorizedAccount, suite.testAccounts["local_account_1"]) ctx.Request = httptest.NewRequest(http.MethodPost, fmt.Sprintf("http://localhost:8080/%s", basePath), nil) // the endpoint we're hitting ctx.Request.Form = url.Values{ "status": {"this is a reply to a status that doesn't exist"}, "spoiler_text": {"don't open cuz it won't work"}, "in_reply_to_id": {"3759e7ef-8ee1-4c0c-86f6-8b70b9ad3d50"}, } suite.statusModule.statusCreatePOSTHandler(ctx) // check response // 1. we should have OK from our call to the function suite.EqualValues(http.StatusOK, recorder.Code) result := recorder.Result() defer result.Body.Close() b, err := ioutil.ReadAll(result.Body) assert.NoError(suite.T(), err) fmt.Println(string(b)) statusReply := &mastotypes.Status{} err = json.Unmarshal(b, statusReply) assert.NoError(suite.T(), err) assert.Equal(suite.T(), "hello hello", statusReply.SpoilerText) assert.Equal(suite.T(), "this is a brand new status!", statusReply.Content) assert.True(suite.T(), statusReply.Sensitive) assert.Equal(suite.T(), mastotypes.VisibilityPrivate, statusReply.Visibility) } func TestStatusCreateTestSuite(t *testing.T) { suite.Run(t, new(StatusCreateTestSuite)) }