// GoToSocial // Copyright (C) GoToSocial Authors admin@gotosocial.org // SPDX-License-Identifier: AGPL-3.0-or-later // // 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 statuses_test import ( "encoding/json" "fmt" "io/ioutil" "net/http" "net/http/httptest" "strings" "testing" "github.com/gin-gonic/gin" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/api/client/statuses" apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" "github.com/superseriousbusiness/gotosocial/internal/oauth" "github.com/superseriousbusiness/gotosocial/testrig" ) type StatusFaveTestSuite struct { StatusStandardTestSuite } // fave a status func (suite *StatusFaveTestSuite) TestPostFave() { t := suite.testTokens["local_account_1"] oauthToken := oauth.DBTokenToToken(t) targetStatus := suite.testStatuses["admin_account_status_2"] // setup recorder := httptest.NewRecorder() ctx, _ := testrig.CreateGinTestContext(recorder, nil) 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", strings.Replace(statuses.FavouritePath, ":id", targetStatus.ID, 1)), nil) // the endpoint we're hitting ctx.Request.Header.Set("accept", "application/json") // normally the router would populate these params from the path values, // but because we're calling the function directly, we need to set them manually. ctx.Params = gin.Params{ gin.Param{ Key: statuses.IDKey, Value: targetStatus.ID, }, } suite.statusModule.StatusFavePOSTHandler(ctx) // check response suite.EqualValues(http.StatusOK, recorder.Code) result := recorder.Result() defer result.Body.Close() b, err := ioutil.ReadAll(result.Body) assert.NoError(suite.T(), err) statusReply := &apimodel.Status{} err = json.Unmarshal(b, statusReply) assert.NoError(suite.T(), err) assert.Equal(suite.T(), targetStatus.ContentWarning, statusReply.SpoilerText) assert.Equal(suite.T(), targetStatus.Content, statusReply.Content) assert.True(suite.T(), statusReply.Sensitive) assert.Equal(suite.T(), apimodel.VisibilityPublic, statusReply.Visibility) assert.True(suite.T(), statusReply.Favourited) assert.Equal(suite.T(), 1, statusReply.FavouritesCount) } // try to fave a status that's not faveable // TODO: replace this when interaction policies enforced. // func (suite *StatusFaveTestSuite) TestPostUnfaveable() { // t := suite.testTokens["local_account_1"] // oauthToken := oauth.DBTokenToToken(t) // targetStatus := suite.testStatuses["local_account_2_status_3"] // this one is unlikeable and unreplyable // // setup // recorder := httptest.NewRecorder() // ctx, _ := testrig.CreateGinTestContext(recorder, nil) // 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", strings.Replace(statuses.FavouritePath, ":id", targetStatus.ID, 1)), nil) // the endpoint we're hitting // ctx.Request.Header.Set("accept", "application/json") // // normally the router would populate these params from the path values, // // but because we're calling the function directly, we need to set them manually. // ctx.Params = gin.Params{ // gin.Param{ // Key: statuses.IDKey, // Value: targetStatus.ID, // }, // } // suite.statusModule.StatusFavePOSTHandler(ctx) // // check response // suite.EqualValues(http.StatusForbidden, recorder.Code) // result := recorder.Result() // defer result.Body.Close() // b, err := ioutil.ReadAll(result.Body) // assert.NoError(suite.T(), err) // assert.Equal(suite.T(), `{"error":"Forbidden: status is not faveable"}`, string(b)) // } func TestStatusFaveTestSuite(t *testing.T) { suite.Run(t, new(StatusFaveTestSuite)) }