mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-11-02 20:02:25 -06:00 
			
		
		
		
	* rewrite cache library as codeberg.org/gruf/go-structr, implement in gotosocial
* use actual go-structr release version (not just commit hash)
* revert go toolchain changes (damn you go for auto changing this)
* fix go mod woes
* ensure %w is used in calls to errs.Appendf()
* fix error checking
* fix possible panic
* remove unnecessary start/stop functions, move to main Cache{} struct, add note regarding which caches require start/stop
* fix copy-paste artifact... 😇
* fix all comment copy-paste artifacts
* remove dropID() function, now we can just use slices.DeleteFunc()
* use util.Deduplicate() instead of collate(), move collate to util
* move orderByIDs() to util package and "generify"
* add a util.DeleteIf() function, use this to delete entries on failed population
* use slices.DeleteFunc() instead of util.DeleteIf() (i had the logic mixed up in my head somehow lol)
* add note about how collate differs from deduplicate
		
	
			
		
			
				
	
	
		
			102 lines
		
	
	
	
		
			3.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			102 lines
		
	
	
	
		
			3.8 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 federatingdb_test
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/suite"
 | 
						|
	"github.com/superseriousbusiness/activity/streams/vocab"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/ap"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/id"
 | 
						|
)
 | 
						|
 | 
						|
type AnnounceTestSuite struct {
 | 
						|
	FederatingDBTestSuite
 | 
						|
}
 | 
						|
 | 
						|
func (suite *AnnounceTestSuite) TestNewAnnounce() {
 | 
						|
	receivingAccount1 := suite.testAccounts["local_account_1"]
 | 
						|
	announcingAccount := suite.testAccounts["remote_account_1"]
 | 
						|
 | 
						|
	ctx := createTestContext(receivingAccount1, announcingAccount)
 | 
						|
	announce1 := suite.testActivities["announce_forwarded_1_zork"]
 | 
						|
 | 
						|
	err := suite.federatingDB.Announce(ctx, announce1.Activity.(vocab.ActivityStreamsAnnounce))
 | 
						|
	suite.NoError(err)
 | 
						|
 | 
						|
	// should be a message heading to the processor now, which we can intercept here
 | 
						|
	msg := <-suite.fromFederator
 | 
						|
	suite.Equal(ap.ActivityAnnounce, msg.APObjectType)
 | 
						|
	suite.Equal(ap.ActivityCreate, msg.APActivityType)
 | 
						|
 | 
						|
	boost, ok := msg.GTSModel.(*gtsmodel.Status)
 | 
						|
	suite.True(ok)
 | 
						|
	suite.Equal(announcingAccount.ID, boost.AccountID)
 | 
						|
 | 
						|
	// only the URI will be set for the boosted status
 | 
						|
	// because it still needs to be dereferenced
 | 
						|
	suite.Nil(boost.BoostOf)
 | 
						|
	suite.Equal("http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1", boost.BoostOfURI)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *AnnounceTestSuite) TestAnnounceTwice() {
 | 
						|
	receivingAccount1 := suite.testAccounts["local_account_1"]
 | 
						|
	receivingAccount2 := suite.testAccounts["local_account_2"]
 | 
						|
 | 
						|
	announcingAccount := suite.testAccounts["remote_account_1"]
 | 
						|
 | 
						|
	ctx1 := createTestContext(receivingAccount1, announcingAccount)
 | 
						|
	announce1 := suite.testActivities["announce_forwarded_1_zork"]
 | 
						|
 | 
						|
	err := suite.federatingDB.Announce(ctx1, announce1.Activity.(vocab.ActivityStreamsAnnounce))
 | 
						|
	suite.NoError(err)
 | 
						|
 | 
						|
	// should be a message heading to the processor now, which we can intercept here
 | 
						|
	msg := <-suite.fromFederator
 | 
						|
	suite.Equal(ap.ActivityAnnounce, msg.APObjectType)
 | 
						|
	suite.Equal(ap.ActivityCreate, msg.APActivityType)
 | 
						|
	boost, ok := msg.GTSModel.(*gtsmodel.Status)
 | 
						|
	suite.True(ok)
 | 
						|
	suite.Equal(announcingAccount.ID, boost.AccountID)
 | 
						|
 | 
						|
	// Insert the boost-of status into the
 | 
						|
	// DB cache to emulate processor handling
 | 
						|
	boost.ID, _ = id.NewULIDFromTime(boost.CreatedAt)
 | 
						|
	suite.state.Caches.GTS.Status.Put(boost)
 | 
						|
 | 
						|
	// only the URI will be set for the boosted status
 | 
						|
	// because it still needs to be dereferenced
 | 
						|
	suite.Nil(boost.BoostOf)
 | 
						|
	suite.Equal("http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1", boost.BoostOfURI)
 | 
						|
 | 
						|
	ctx2 := createTestContext(receivingAccount2, announcingAccount)
 | 
						|
	announce2 := suite.testActivities["announce_forwarded_1_turtle"]
 | 
						|
 | 
						|
	err = suite.federatingDB.Announce(ctx2, announce2.Activity.(vocab.ActivityStreamsAnnounce))
 | 
						|
	suite.NoError(err)
 | 
						|
 | 
						|
	// since this is a repeat announce with the same URI, just delivered to a different inbox,
 | 
						|
	// we should have nothing in the messages channel...
 | 
						|
	suite.Empty(suite.fromFederator)
 | 
						|
}
 | 
						|
 | 
						|
func TestAnnounceTestSuite(t *testing.T) {
 | 
						|
	suite.Run(t, &AnnounceTestSuite{})
 | 
						|
}
 |