mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 11:52:24 -05:00 
			
		
		
		
	[bugfix] Fix unpinning statuses not working (#1582)
And also fix unpinning/pinning potentially leaking the ID of followers-only statuses through returning 422 instead of 404. Also tests!
This commit is contained in:
		
					parent
					
						
							
								fe6c8b8152
							
						
					
				
			
			
				commit
				
					
						29f8c51ab8
					
				
			
		
					 2 changed files with 149 additions and 1 deletions
				
			
		
							
								
								
									
										132
									
								
								internal/api/client/statuses/statusunpin_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										132
									
								
								internal/api/client/statuses/statusunpin_test.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,132 @@ | ||||||
|  | /* | ||||||
|  |    GoToSocial | ||||||
|  |    Copyright (C) 2021-2023 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 statuses_test | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"net/http" | ||||||
|  | 	"net/http/httptest" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"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/config" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtserror" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/oauth" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type StatusUnpinTestSuite struct { | ||||||
|  | 	StatusStandardTestSuite | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *StatusUnpinTestSuite) createUnpin( | ||||||
|  | 	expectedHTTPStatus int, | ||||||
|  | 	expectedBody string, | ||||||
|  | 	targetStatusID string, | ||||||
|  | ) (*apimodel.Status, error) { | ||||||
|  | 	// instantiate recorder + test context | ||||||
|  | 	recorder := httptest.NewRecorder() | ||||||
|  | 	ctx, _ := testrig.CreateGinTestContext(recorder, nil) | ||||||
|  | 	ctx.Set(oauth.SessionAuthorizedAccount, suite.testAccounts["admin_account"]) | ||||||
|  | 	ctx.Set(oauth.SessionAuthorizedToken, oauth.DBTokenToToken(suite.testTokens["admin_account"])) | ||||||
|  | 	ctx.Set(oauth.SessionAuthorizedApplication, suite.testApplications["application_1"]) | ||||||
|  | 	ctx.Set(oauth.SessionAuthorizedUser, suite.testUsers["admin_account"]) | ||||||
|  | 
 | ||||||
|  | 	// create the request | ||||||
|  | 	ctx.Request = httptest.NewRequest(http.MethodPost, config.GetProtocol()+"://"+config.GetHost()+"/api/"+statuses.BasePath+"/"+targetStatusID+"/unpin", nil) | ||||||
|  | 	ctx.Request.Header.Set("accept", "application/json") | ||||||
|  | 	ctx.AddParam(statuses.IDKey, targetStatusID) | ||||||
|  | 
 | ||||||
|  | 	// trigger the handler | ||||||
|  | 	suite.statusModule.StatusUnpinPOSTHandler(ctx) | ||||||
|  | 
 | ||||||
|  | 	// read the response | ||||||
|  | 	result := recorder.Result() | ||||||
|  | 	defer result.Body.Close() | ||||||
|  | 
 | ||||||
|  | 	b, err := ioutil.ReadAll(result.Body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	errs := gtserror.MultiError{} | ||||||
|  | 
 | ||||||
|  | 	// check code + body | ||||||
|  | 	if resultCode := recorder.Code; expectedHTTPStatus != resultCode { | ||||||
|  | 		errs = append(errs, fmt.Sprintf("expected %d got %d", expectedHTTPStatus, resultCode)) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// if we got an expected body, return early | ||||||
|  | 	if expectedBody != "" && string(b) != expectedBody { | ||||||
|  | 		errs = append(errs, fmt.Sprintf("expected %s got %s", expectedBody, string(b))) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if len(errs) > 0 { | ||||||
|  | 		return nil, errs.Combine() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	resp := &apimodel.Status{} | ||||||
|  | 	if err := json.Unmarshal(b, resp); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return resp, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *StatusUnpinTestSuite) TestUnpinStatusOK() { | ||||||
|  | 	// Unpin a pinned public status that this account owns. | ||||||
|  | 	targetStatus := suite.testStatuses["admin_account_status_1"] | ||||||
|  | 
 | ||||||
|  | 	resp, err := suite.createUnpin(http.StatusOK, "", targetStatus.ID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		suite.FailNow(err.Error()) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	suite.False(resp.Pinned) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *StatusUnpinTestSuite) TestUnpinStatusNotFound() { | ||||||
|  | 	// Unpin a pinned followers-only status owned by another account. | ||||||
|  | 	targetStatus := suite.testStatuses["local_account_2_status_7"] | ||||||
|  | 
 | ||||||
|  | 	if _, err := suite.createUnpin(http.StatusNotFound, `{"error":"Not Found"}`, targetStatus.ID); err != nil { | ||||||
|  | 		suite.FailNow(err.Error()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *StatusUnpinTestSuite) TestUnpinStatusUnprocessable() { | ||||||
|  | 	// Unpin a not-pinned status owned by another account. | ||||||
|  | 	targetStatus := suite.testStatuses["local_account_1_status_1"] | ||||||
|  | 
 | ||||||
|  | 	if _, err := suite.createUnpin( | ||||||
|  | 		http.StatusUnprocessableEntity, | ||||||
|  | 		`{"error":"Unprocessable Entity: status 01F8MHAMCHF6Y650WCRSCP4WMY does not belong to account 01F8MH17FWEB39HZJ76B6VXSKF"}`, | ||||||
|  | 		targetStatus.ID, | ||||||
|  | 	); err != nil { | ||||||
|  | 		suite.FailNow(err.Error()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestStatusUnpinTestSuite(t *testing.T) { | ||||||
|  | 	suite.Run(t, new(StatusUnpinTestSuite)) | ||||||
|  | } | ||||||
|  | @ -35,6 +35,7 @@ const allowedPinnedCount = 10 | ||||||
| // can pin or unpin it. | // can pin or unpin it. | ||||||
| // | // | ||||||
| // It checks: | // It checks: | ||||||
|  | //   - Status is visible to requesting account. | ||||||
| //   - Status belongs to requesting account. | //   - Status belongs to requesting account. | ||||||
| //   - Status is public, unlisted, or followers-only. | //   - Status is public, unlisted, or followers-only. | ||||||
| //   - Status is not a boost. | //   - Status is not a boost. | ||||||
|  | @ -45,6 +46,21 @@ func (p *Processor) getPinnableStatus(ctx context.Context, targetStatusID string | ||||||
| 		return nil, gtserror.NewErrorNotFound(err) | 		return nil, gtserror.NewErrorNotFound(err) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 	requestingAccount, err := p.state.DB.GetAccountByID(ctx, requestingAccountID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, gtserror.NewErrorInternalError(err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	visible, err := p.filter.StatusVisible(ctx, targetStatus, requestingAccount) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, gtserror.NewErrorInternalError(err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if !visible { | ||||||
|  | 		err = fmt.Errorf("status %s not visible to account %s", targetStatusID, requestingAccountID) | ||||||
|  | 		return nil, gtserror.NewErrorNotFound(err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	if targetStatus.AccountID != requestingAccountID { | 	if targetStatus.AccountID != requestingAccountID { | ||||||
| 		err = fmt.Errorf("status %s does not belong to account %s", targetStatusID, requestingAccountID) | 		err = fmt.Errorf("status %s does not belong to account %s", targetStatusID, requestingAccountID) | ||||||
| 		return nil, gtserror.NewErrorUnprocessableEntity(err, err.Error()) | 		return nil, gtserror.NewErrorUnprocessableEntity(err, err.Error()) | ||||||
|  | @ -124,7 +140,7 @@ func (p *Processor) PinRemove(ctx context.Context, requestingAccount *gtsmodel.A | ||||||
| 		return nil, errWithCode | 		return nil, errWithCode | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	if targetStatus.PinnedAt.IsZero() { | 	if !targetStatus.PinnedAt.IsZero() { | ||||||
| 		targetStatus.PinnedAt = time.Time{} | 		targetStatus.PinnedAt = time.Time{} | ||||||
| 		if err := p.state.DB.UpdateStatus(ctx, targetStatus, "pinned_at"); err != nil { | 		if err := p.state.DB.UpdateStatus(ctx, targetStatus, "pinned_at"); err != nil { | ||||||
| 			return nil, gtserror.NewErrorInternalError(fmt.Errorf("db error unpinning status: %w", err)) | 			return nil, gtserror.NewErrorInternalError(fmt.Errorf("db error unpinning status: %w", err)) | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue