mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 14:12:26 -05:00 
			
		
		
		
	
		
			
	
	
		
			92 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
		
		
			
		
	
	
			92 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
|  | /* | ||
|  |    GoToSocial | ||
|  |    Copyright (C) 2021-2022 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 <http://www.gnu.org/licenses/>. | ||
|  | */ | ||
|  | 
 | ||
|  | package status_test | ||
|  | 
 | ||
|  | import ( | ||
|  | 	"encoding/json" | ||
|  | 	"errors" | ||
|  | 	"fmt" | ||
|  | 	"io/ioutil" | ||
|  | 	"net/http" | ||
|  | 	"net/http/httptest" | ||
|  | 	"strings" | ||
|  | 	"testing" | ||
|  | 
 | ||
|  | 	"github.com/gin-gonic/gin" | ||
|  | 	"github.com/stretchr/testify/suite" | ||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/api/client/status" | ||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/api/model" | ||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/db" | ||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/oauth" | ||
|  | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||
|  | ) | ||
|  | 
 | ||
|  | type StatusDeleteTestSuite struct { | ||
|  | 	StatusStandardTestSuite | ||
|  | } | ||
|  | 
 | ||
|  | func (suite *StatusDeleteTestSuite) TestPostDelete() { | ||
|  | 	t := suite.testTokens["local_account_1"] | ||
|  | 	oauthToken := oauth.DBTokenToToken(t) | ||
|  | 	targetStatus := suite.testStatuses["local_account_1_status_1"] | ||
|  | 
 | ||
|  | 	// 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.MethodDelete, fmt.Sprintf("http://localhost:8080%s", strings.Replace(status.BasePathWithID, ":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:   status.IDKey, | ||
|  | 			Value: targetStatus.ID, | ||
|  | 		}, | ||
|  | 	} | ||
|  | 
 | ||
|  | 	suite.statusModule.StatusDELETEHandler(ctx) | ||
|  | 
 | ||
|  | 	// check response | ||
|  | 	suite.EqualValues(http.StatusOK, recorder.Code) | ||
|  | 
 | ||
|  | 	result := recorder.Result() | ||
|  | 	defer result.Body.Close() | ||
|  | 	b, err := ioutil.ReadAll(result.Body) | ||
|  | 	suite.NoError(err) | ||
|  | 
 | ||
|  | 	statusReply := &model.Status{} | ||
|  | 	err = json.Unmarshal(b, statusReply) | ||
|  | 	suite.NoError(err) | ||
|  | 	suite.NotNil(statusReply) | ||
|  | 
 | ||
|  | 	if !testrig.WaitFor(func() bool { | ||
|  | 		_, err := suite.db.GetStatusByID(ctx, targetStatus.ID) | ||
|  | 		return errors.Is(err, db.ErrNoEntries) | ||
|  | 	}) { | ||
|  | 		suite.FailNow("time out waiting for status to be deleted") | ||
|  | 	} | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | func TestStatusDeleteTestSuite(t *testing.T) { | ||
|  | 	suite.Run(t, new(StatusDeleteTestSuite)) | ||
|  | } |