mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-11-04 03:32:25 -06:00 
			
		
		
		
	
		
			
	
	
		
			104 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
		
		
			
		
	
	
			104 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| 
								 | 
							
								// 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 <http://www.gnu.org/licenses/>.
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								package admin_test
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								import (
							 | 
						||
| 
								 | 
							
									"context"
							 | 
						||
| 
								 | 
							
									"testing"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									"github.com/stretchr/testify/suite"
							 | 
						||
| 
								 | 
							
									apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
							 | 
						||
| 
								 | 
							
									"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
							 | 
						||
| 
								 | 
							
									"github.com/superseriousbusiness/gotosocial/testrig"
							 | 
						||
| 
								 | 
							
								)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								type AccountTestSuite struct {
							 | 
						||
| 
								 | 
							
									AdminStandardTestSuite
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								func (suite *AccountTestSuite) TestAccountActionSuspend() {
							 | 
						||
| 
								 | 
							
									var (
							 | 
						||
| 
								 | 
							
										ctx       = context.Background()
							 | 
						||
| 
								 | 
							
										adminAcct = suite.testAccounts["admin_account"]
							 | 
						||
| 
								 | 
							
										request   = &apimodel.AdminActionRequest{
							 | 
						||
| 
								 | 
							
											Category: gtsmodel.AdminActionCategoryAccount.String(),
							 | 
						||
| 
								 | 
							
											Type:     gtsmodel.AdminActionSuspend.String(),
							 | 
						||
| 
								 | 
							
											Text:     "stinky",
							 | 
						||
| 
								 | 
							
											TargetID: suite.testAccounts["local_account_1"].ID,
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
									)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									actionID, errWithCode := suite.adminProcessor.AccountAction(
							 | 
						||
| 
								 | 
							
										ctx,
							 | 
						||
| 
								 | 
							
										adminAcct,
							 | 
						||
| 
								 | 
							
										request,
							 | 
						||
| 
								 | 
							
									)
							 | 
						||
| 
								 | 
							
									suite.NoError(errWithCode)
							 | 
						||
| 
								 | 
							
									suite.NotEmpty(actionID)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Wait for action to finish.
							 | 
						||
| 
								 | 
							
									if !testrig.WaitFor(func() bool {
							 | 
						||
| 
								 | 
							
										return suite.adminProcessor.Actions().TotalRunning() == 0
							 | 
						||
| 
								 | 
							
									}) {
							 | 
						||
| 
								 | 
							
										suite.FailNow("timed out waiting for admin action(s) to finish")
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Ensure action marked as
							 | 
						||
| 
								 | 
							
									// completed in the database.
							 | 
						||
| 
								 | 
							
									adminAction, err := suite.db.GetAdminAction(ctx, actionID)
							 | 
						||
| 
								 | 
							
									if err != nil {
							 | 
						||
| 
								 | 
							
										suite.FailNow(err.Error())
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									suite.NotZero(adminAction.CompletedAt)
							 | 
						||
| 
								 | 
							
									suite.Empty(adminAction.Errors)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Ensure target account suspended.
							 | 
						||
| 
								 | 
							
									targetAcct, err := suite.db.GetAccountByID(ctx, request.TargetID)
							 | 
						||
| 
								 | 
							
									if err != nil {
							 | 
						||
| 
								 | 
							
										suite.FailNow(err.Error())
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									suite.NotZero(targetAcct.SuspendedAt)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								func (suite *AccountTestSuite) TestAccountActionUnsupported() {
							 | 
						||
| 
								 | 
							
									var (
							 | 
						||
| 
								 | 
							
										ctx       = context.Background()
							 | 
						||
| 
								 | 
							
										adminAcct = suite.testAccounts["admin_account"]
							 | 
						||
| 
								 | 
							
										request   = &apimodel.AdminActionRequest{
							 | 
						||
| 
								 | 
							
											Category: gtsmodel.AdminActionCategoryAccount.String(),
							 | 
						||
| 
								 | 
							
											Type:     "pee pee poo poo",
							 | 
						||
| 
								 | 
							
											Text:     "stinky",
							 | 
						||
| 
								 | 
							
											TargetID: suite.testAccounts["local_account_1"].ID,
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
									)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									actionID, errWithCode := suite.adminProcessor.AccountAction(
							 | 
						||
| 
								 | 
							
										ctx,
							 | 
						||
| 
								 | 
							
										adminAcct,
							 | 
						||
| 
								 | 
							
										request,
							 | 
						||
| 
								 | 
							
									)
							 | 
						||
| 
								 | 
							
									suite.EqualError(errWithCode, "admin action type pee pee poo poo is not supported for this endpoint, currently supported types are: [\"suspend\"]")
							 | 
						||
| 
								 | 
							
									suite.Empty(actionID)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								func TestAccountTestSuite(t *testing.T) {
							 | 
						||
| 
								 | 
							
									suite.Run(t, new(AccountTestSuite))
							 | 
						||
| 
								 | 
							
								}
							 |