mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 04:12:25 -05:00 
			
		
		
		
	[chore] Inboxes for iri test (#508)
* tidy up some federation tests * add missing license to following.go * give zork some followers, as a treat * test InboxesForIRI * Go fmt * update timeline tests
This commit is contained in:
		
					parent
					
						
							
								f9b351438b
							
						
					
				
			
			
				commit
				
					
						63f84da3e4
					
				
			
		
					 11 changed files with 247 additions and 110 deletions
				
			
		|  | @ -80,7 +80,7 @@ func (suite *AccountVerifyTestSuite) TestAccountVerifyGet() { | ||||||
| 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/avatar/small/01F8MH58A357CV5K7R7TJMSH6S.jpeg", apimodelAccount.AvatarStatic) | 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/avatar/small/01F8MH58A357CV5K7R7TJMSH6S.jpeg", apimodelAccount.AvatarStatic) | ||||||
| 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/header/original/01PFPMWK2FF0D9WMHEJHR07C3Q.jpeg", apimodelAccount.Header) | 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/header/original/01PFPMWK2FF0D9WMHEJHR07C3Q.jpeg", apimodelAccount.Header) | ||||||
| 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/header/small/01PFPMWK2FF0D9WMHEJHR07C3Q.jpeg", apimodelAccount.HeaderStatic) | 	suite.Equal("http://localhost:8080/fileserver/01F8MH1H7YV1Z7D2C8K2730QBF/header/small/01PFPMWK2FF0D9WMHEJHR07C3Q.jpeg", apimodelAccount.HeaderStatic) | ||||||
| 	suite.Zero(apimodelAccount.FollowersCount) | 	suite.Equal(2, apimodelAccount.FollowersCount) | ||||||
| 	suite.Equal(2, apimodelAccount.FollowingCount) | 	suite.Equal(2, apimodelAccount.FollowingCount) | ||||||
| 	suite.Equal(5, apimodelAccount.StatusesCount) | 	suite.Equal(5, apimodelAccount.StatusesCount) | ||||||
| 	suite.WithinDuration(time.Now(), lastStatusAt, 5*time.Minute) | 	suite.WithinDuration(time.Now(), lastStatusAt, 5*time.Minute) | ||||||
|  |  | ||||||
|  | @ -63,6 +63,7 @@ type StatusStandardTestSuite struct { | ||||||
| 	testAccounts     map[string]*gtsmodel.Account | 	testAccounts     map[string]*gtsmodel.Account | ||||||
| 	testAttachments  map[string]*gtsmodel.MediaAttachment | 	testAttachments  map[string]*gtsmodel.MediaAttachment | ||||||
| 	testStatuses     map[string]*gtsmodel.Status | 	testStatuses     map[string]*gtsmodel.Status | ||||||
|  | 	testFollows      map[string]*gtsmodel.Follow | ||||||
| 
 | 
 | ||||||
| 	// module being tested | 	// module being tested | ||||||
| 	statusModule *status.Module | 	statusModule *status.Module | ||||||
|  | @ -76,25 +77,27 @@ func (suite *StatusStandardTestSuite) SetupSuite() { | ||||||
| 	suite.testAccounts = testrig.NewTestAccounts() | 	suite.testAccounts = testrig.NewTestAccounts() | ||||||
| 	suite.testAttachments = testrig.NewTestAttachments() | 	suite.testAttachments = testrig.NewTestAttachments() | ||||||
| 	suite.testStatuses = testrig.NewTestStatuses() | 	suite.testStatuses = testrig.NewTestStatuses() | ||||||
|  | 	suite.testFollows = testrig.NewTestFollows() | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *StatusStandardTestSuite) SetupTest() { | func (suite *StatusStandardTestSuite) SetupTest() { | ||||||
| 	testrig.InitTestConfig() | 	testrig.InitTestConfig() | ||||||
| 	testrig.InitTestLog() | 	testrig.InitTestLog() | ||||||
| 
 | 
 | ||||||
| 	fedWorker := worker.New[messages.FromFederator](-1, -1) |  | ||||||
| 	clientWorker := worker.New[messages.FromClientAPI](-1, -1) |  | ||||||
| 
 |  | ||||||
| 	suite.db = testrig.NewTestDB() | 	suite.db = testrig.NewTestDB() | ||||||
| 	suite.tc = testrig.NewTestTypeConverter(suite.db) | 	suite.tc = testrig.NewTestTypeConverter(suite.db) | ||||||
| 	suite.storage = testrig.NewTestStorage() | 	suite.storage = testrig.NewTestStorage() | ||||||
|  | 	testrig.StandardDBSetup(suite.db, nil) | ||||||
|  | 	testrig.StandardStorageSetup(suite.storage, "../../../../testrig/media") | ||||||
|  | 
 | ||||||
|  | 	fedWorker := worker.New[messages.FromFederator](-1, -1) | ||||||
|  | 	clientWorker := worker.New[messages.FromClientAPI](-1, -1) | ||||||
|  | 
 | ||||||
| 	suite.mediaManager = testrig.NewTestMediaManager(suite.db, suite.storage) | 	suite.mediaManager = testrig.NewTestMediaManager(suite.db, suite.storage) | ||||||
| 	suite.federator = testrig.NewTestFederator(suite.db, testrig.NewTestTransportController(suite.testHttpClient(), suite.db, fedWorker), suite.storage, suite.mediaManager, fedWorker) | 	suite.federator = testrig.NewTestFederator(suite.db, testrig.NewTestTransportController(suite.testHttpClient(), suite.db, fedWorker), suite.storage, suite.mediaManager, fedWorker) | ||||||
| 	suite.emailSender = testrig.NewEmailSender("../../../../web/template/", nil) | 	suite.emailSender = testrig.NewEmailSender("../../../../web/template/", nil) | ||||||
| 	suite.processor = testrig.NewTestProcessor(suite.db, suite.storage, suite.federator, suite.emailSender, suite.mediaManager, clientWorker, fedWorker) | 	suite.processor = testrig.NewTestProcessor(suite.db, suite.storage, suite.federator, suite.emailSender, suite.mediaManager, clientWorker, fedWorker) | ||||||
| 	suite.statusModule = status.New(suite.processor).(*status.Module) | 	suite.statusModule = status.New(suite.processor).(*status.Module) | ||||||
| 	testrig.StandardDBSetup(suite.db, nil) |  | ||||||
| 	testrig.StandardStorageSetup(suite.storage, "../../../../testrig/media") |  | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *StatusStandardTestSuite) TearDownTest() { | func (suite *StatusStandardTestSuite) TearDownTest() { | ||||||
|  |  | ||||||
|  | @ -16,6 +16,7 @@ | ||||||
| package status_test | package status_test | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
|  | 	"context" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
|  | @ -29,6 +30,7 @@ import ( | ||||||
| 	"github.com/stretchr/testify/suite" | 	"github.com/stretchr/testify/suite" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/api/client/status" | 	"github.com/superseriousbusiness/gotosocial/internal/api/client/status" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/api/model" | 	"github.com/superseriousbusiness/gotosocial/internal/api/model" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/oauth" | 	"github.com/superseriousbusiness/gotosocial/internal/oauth" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
|  | @ -144,6 +146,10 @@ func (suite *StatusBoostTestSuite) TestPostUnboostable() { | ||||||
| // try to boost a status that's not visible to the user | // try to boost a status that's not visible to the user | ||||||
| func (suite *StatusBoostTestSuite) TestPostNotVisible() { | func (suite *StatusBoostTestSuite) TestPostNotVisible() { | ||||||
| 
 | 
 | ||||||
|  | 	// stop local_account_2 following zork | ||||||
|  | 	err := suite.db.DeleteByID(context.Background(), suite.testFollows["local_account_2_local_account_1"].ID, >smodel.Follow{}) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 
 | ||||||
| 	t := suite.testTokens["local_account_2"] | 	t := suite.testTokens["local_account_2"] | ||||||
| 	oauthToken := oauth.DBTokenToToken(t) | 	oauthToken := oauth.DBTokenToToken(t) | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
|  | @ -1,3 +1,21 @@ | ||||||
|  | /* | ||||||
|  |    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 federatingdb | package federatingdb | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
|  |  | ||||||
							
								
								
									
										56
									
								
								internal/federation/federatingdb/inbox_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										56
									
								
								internal/federation/federatingdb/inbox_test.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,56 @@ | ||||||
|  | /* | ||||||
|  |    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 federatingdb_test | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"github.com/stretchr/testify/suite" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type InboxTestSuite struct { | ||||||
|  | 	FederatingDBTestSuite | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *InboxTestSuite) TestInboxesForFollowersIRI() { | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	testAccount := suite.testAccounts["local_account_1"] | ||||||
|  | 
 | ||||||
|  | 	inboxIRIs, err := suite.federatingDB.InboxesForIRI(ctx, testrig.URLMustParse(testAccount.FollowersURI)) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 	suite.Len(inboxIRIs, 2) | ||||||
|  | 	suite.Equal(inboxIRIs[0].String(), suite.testAccounts["local_account_2"].InboxURI) | ||||||
|  | 	suite.Equal(inboxIRIs[1].String(), suite.testAccounts["admin_account"].InboxURI) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *InboxTestSuite) TestInboxesForAccountIRI() { | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	testAccount := suite.testAccounts["local_account_1"] | ||||||
|  | 
 | ||||||
|  | 	inboxIRIs, err := suite.federatingDB.InboxesForIRI(ctx, testrig.URLMustParse(testAccount.URI)) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 	suite.Len(inboxIRIs, 1) | ||||||
|  | 	suite.Equal(inboxIRIs[0].String(), testAccount.InboxURI) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestInboxTestSuite(t *testing.T) { | ||||||
|  | 	suite.Run(t, &InboxTestSuite{}) | ||||||
|  | } | ||||||
							
								
								
									
										122
									
								
								internal/federation/federatingprotocol_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								internal/federation/federatingprotocol_test.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,122 @@ | ||||||
|  | /* | ||||||
|  |    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 federation_test | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"net/http" | ||||||
|  | 	"net/http/httptest" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"github.com/go-fed/httpsig" | ||||||
|  | 	"github.com/stretchr/testify/suite" | ||||||
|  | 	"github.com/superseriousbusiness/activity/pub" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/ap" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/federation" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/messages" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/worker" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type FederatingProtocolTestSuite struct { | ||||||
|  | 	FederatorStandardTestSuite | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // make sure PostInboxRequestBodyHook properly sets the inbox username and activity on the context | ||||||
|  | func (suite *FederatingProtocolTestSuite) TestPostInboxRequestBodyHook() { | ||||||
|  | 	// the activity we're gonna use | ||||||
|  | 	activity := suite.activities["dm_for_zork"] | ||||||
|  | 
 | ||||||
|  | 	fedWorker := worker.New[messages.FromFederator](-1, -1) | ||||||
|  | 
 | ||||||
|  | 	// setup transport controller with a no-op client so we don't make external calls | ||||||
|  | 	tc := testrig.NewTestTransportController(testrig.NewMockHTTPClient(func(req *http.Request) (*http.Response, error) { | ||||||
|  | 		return nil, nil | ||||||
|  | 	}), suite.db, fedWorker) | ||||||
|  | 	// setup module being tested | ||||||
|  | 	federator := federation.NewFederator(suite.db, testrig.NewTestFederatingDB(suite.db, fedWorker), tc, suite.typeConverter, testrig.NewTestMediaManager(suite.db, suite.storage)) | ||||||
|  | 
 | ||||||
|  | 	// setup request | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	request := httptest.NewRequest(http.MethodPost, "http://localhost:8080/users/the_mighty_zork/inbox", nil) // the endpoint we're hitting | ||||||
|  | 	request.Header.Set("Signature", activity.SignatureHeader) | ||||||
|  | 
 | ||||||
|  | 	// trigger the function being tested, and return the new context it creates | ||||||
|  | 	newContext, err := federator.PostInboxRequestBodyHook(ctx, request, activity.Activity) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 	suite.NotNil(newContext) | ||||||
|  | 
 | ||||||
|  | 	// activity should be set on context now | ||||||
|  | 	activityI := newContext.Value(ap.ContextActivity) | ||||||
|  | 	suite.NotNil(activityI) | ||||||
|  | 	returnedActivity, ok := activityI.(pub.Activity) | ||||||
|  | 	suite.True(ok) | ||||||
|  | 	suite.NotNil(returnedActivity) | ||||||
|  | 	suite.EqualValues(activity.Activity, returnedActivity) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (suite *FederatingProtocolTestSuite) TestAuthenticatePostInbox() { | ||||||
|  | 	// the activity we're gonna use | ||||||
|  | 	activity := suite.activities["dm_for_zork"] | ||||||
|  | 	sendingAccount := suite.accounts["remote_account_1"] | ||||||
|  | 	inboxAccount := suite.accounts["local_account_1"] | ||||||
|  | 
 | ||||||
|  | 	fedWorker := worker.New[messages.FromFederator](-1, -1) | ||||||
|  | 
 | ||||||
|  | 	tc := testrig.NewTestTransportController(testrig.NewMockHTTPClient(nil), suite.db, fedWorker) | ||||||
|  | 	// now setup module being tested, with the mock transport controller | ||||||
|  | 	federator := federation.NewFederator(suite.db, testrig.NewTestFederatingDB(suite.db, fedWorker), tc, suite.typeConverter, testrig.NewTestMediaManager(suite.db, suite.storage)) | ||||||
|  | 
 | ||||||
|  | 	request := httptest.NewRequest(http.MethodPost, "http://localhost:8080/users/the_mighty_zork/inbox", nil) | ||||||
|  | 	// we need these headers for the request to be validated | ||||||
|  | 	request.Header.Set("Signature", activity.SignatureHeader) | ||||||
|  | 	request.Header.Set("Date", activity.DateHeader) | ||||||
|  | 	request.Header.Set("Digest", activity.DigestHeader) | ||||||
|  | 
 | ||||||
|  | 	verifier, err := httpsig.NewVerifier(request) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 
 | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	// by the time AuthenticatePostInbox is called, PostInboxRequestBodyHook should have already been called, | ||||||
|  | 	// which should have set the account and username onto the request. We can replicate that behavior here: | ||||||
|  | 	ctxWithAccount := context.WithValue(ctx, ap.ContextReceivingAccount, inboxAccount) | ||||||
|  | 	ctxWithActivity := context.WithValue(ctxWithAccount, ap.ContextActivity, activity) | ||||||
|  | 	ctxWithVerifier := context.WithValue(ctxWithActivity, ap.ContextRequestingPublicKeyVerifier, verifier) | ||||||
|  | 	ctxWithSignature := context.WithValue(ctxWithVerifier, ap.ContextRequestingPublicKeySignature, activity.SignatureHeader) | ||||||
|  | 
 | ||||||
|  | 	// we can pass this recorder as a writer and read it back after | ||||||
|  | 	recorder := httptest.NewRecorder() | ||||||
|  | 
 | ||||||
|  | 	// trigger the function being tested, and return the new context it creates | ||||||
|  | 	newContext, authed, err := federator.AuthenticatePostInbox(ctxWithSignature, recorder, request) | ||||||
|  | 	suite.NoError(err) | ||||||
|  | 	suite.True(authed) | ||||||
|  | 
 | ||||||
|  | 	// since we know this account already it should be set on the context | ||||||
|  | 	requestingAccountI := newContext.Value(ap.ContextRequestingAccount) | ||||||
|  | 	suite.NotNil(requestingAccountI) | ||||||
|  | 	requestingAccount, ok := requestingAccountI.(*gtsmodel.Account) | ||||||
|  | 	suite.True(ok) | ||||||
|  | 	suite.Equal(sendingAccount.Username, requestingAccount.Username) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestFederatingProtocolTestSuite(t *testing.T) { | ||||||
|  | 	suite.Run(t, new(FederatingProtocolTestSuite)) | ||||||
|  | } | ||||||
|  | @ -19,28 +19,16 @@ | ||||||
| package federation_test | package federation_test | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
| 	"context" |  | ||||||
| 	"net/http" |  | ||||||
| 	"net/http/httptest" |  | ||||||
| 	"testing" |  | ||||||
| 
 |  | ||||||
| 	"codeberg.org/gruf/go-store/kv" | 	"codeberg.org/gruf/go-store/kv" | ||||||
| 	"github.com/go-fed/httpsig" |  | ||||||
| 	"github.com/stretchr/testify/assert" |  | ||||||
| 	"github.com/stretchr/testify/suite" | 	"github.com/stretchr/testify/suite" | ||||||
| 	"github.com/superseriousbusiness/activity/pub" |  | ||||||
| 
 | 
 | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/ap" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/db" | 	"github.com/superseriousbusiness/gotosocial/internal/db" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/federation" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/messages" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/typeutils" | 	"github.com/superseriousbusiness/gotosocial/internal/typeutils" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/worker" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/testrig" | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| type ProtocolTestSuite struct { | type FederatorStandardTestSuite struct { | ||||||
| 	suite.Suite | 	suite.Suite | ||||||
| 	db            db.DB | 	db            db.DB | ||||||
| 	storage       *kv.KVStore | 	storage       *kv.KVStore | ||||||
|  | @ -50,104 +38,22 @@ type ProtocolTestSuite struct { | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // SetupSuite sets some variables on the suite that we can use as consts (more or less) throughout | // SetupSuite sets some variables on the suite that we can use as consts (more or less) throughout | ||||||
| func (suite *ProtocolTestSuite) SetupSuite() { | func (suite *FederatorStandardTestSuite) SetupSuite() { | ||||||
| 	// setup standard items | 	// setup standard items | ||||||
| 	suite.storage = testrig.NewTestStorage() | 	suite.storage = testrig.NewTestStorage() | ||||||
| 	suite.typeConverter = testrig.NewTestTypeConverter(suite.db) | 	suite.typeConverter = testrig.NewTestTypeConverter(suite.db) | ||||||
| 	suite.accounts = testrig.NewTestAccounts() | 	suite.accounts = testrig.NewTestAccounts() | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *ProtocolTestSuite) SetupTest() { | func (suite *FederatorStandardTestSuite) SetupTest() { | ||||||
| 	testrig.InitTestLog() |  | ||||||
| 	testrig.InitTestConfig() | 	testrig.InitTestConfig() | ||||||
|  | 	testrig.InitTestLog() | ||||||
| 	suite.db = testrig.NewTestDB() | 	suite.db = testrig.NewTestDB() | ||||||
| 	suite.activities = testrig.NewTestActivities(suite.accounts) | 	suite.activities = testrig.NewTestActivities(suite.accounts) | ||||||
| 	testrig.StandardDBSetup(suite.db, suite.accounts) | 	testrig.StandardDBSetup(suite.db, suite.accounts) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // TearDownTest drops tables to make sure there's no data in the db | // TearDownTest drops tables to make sure there's no data in the db | ||||||
| func (suite *ProtocolTestSuite) TearDownTest() { | func (suite *FederatorStandardTestSuite) TearDownTest() { | ||||||
| 	testrig.StandardDBTeardown(suite.db) | 	testrig.StandardDBTeardown(suite.db) | ||||||
| } | } | ||||||
| 
 |  | ||||||
| // make sure PostInboxRequestBodyHook properly sets the inbox username and activity on the context |  | ||||||
| func (suite *ProtocolTestSuite) TestPostInboxRequestBodyHook() { |  | ||||||
| 	// the activity we're gonna use |  | ||||||
| 	activity := suite.activities["dm_for_zork"] |  | ||||||
| 
 |  | ||||||
| 	fedWorker := worker.New[messages.FromFederator](-1, -1) |  | ||||||
| 
 |  | ||||||
| 	// setup transport controller with a no-op client so we don't make external calls |  | ||||||
| 	tc := testrig.NewTestTransportController(testrig.NewMockHTTPClient(func(req *http.Request) (*http.Response, error) { |  | ||||||
| 		return nil, nil |  | ||||||
| 	}), suite.db, fedWorker) |  | ||||||
| 	// setup module being tested |  | ||||||
| 	federator := federation.NewFederator(suite.db, testrig.NewTestFederatingDB(suite.db, fedWorker), tc, suite.typeConverter, testrig.NewTestMediaManager(suite.db, suite.storage)) |  | ||||||
| 
 |  | ||||||
| 	// setup request |  | ||||||
| 	ctx := context.Background() |  | ||||||
| 	request := httptest.NewRequest(http.MethodPost, "http://localhost:8080/users/the_mighty_zork/inbox", nil) // the endpoint we're hitting |  | ||||||
| 	request.Header.Set("Signature", activity.SignatureHeader) |  | ||||||
| 
 |  | ||||||
| 	// trigger the function being tested, and return the new context it creates |  | ||||||
| 	newContext, err := federator.PostInboxRequestBodyHook(ctx, request, activity.Activity) |  | ||||||
| 	assert.NoError(suite.T(), err) |  | ||||||
| 	assert.NotNil(suite.T(), newContext) |  | ||||||
| 
 |  | ||||||
| 	// activity should be set on context now |  | ||||||
| 	activityI := newContext.Value(ap.ContextActivity) |  | ||||||
| 	assert.NotNil(suite.T(), activityI) |  | ||||||
| 	returnedActivity, ok := activityI.(pub.Activity) |  | ||||||
| 	assert.True(suite.T(), ok) |  | ||||||
| 	assert.NotNil(suite.T(), returnedActivity) |  | ||||||
| 	assert.EqualValues(suite.T(), activity.Activity, returnedActivity) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func (suite *ProtocolTestSuite) TestAuthenticatePostInbox() { |  | ||||||
| 	// the activity we're gonna use |  | ||||||
| 	activity := suite.activities["dm_for_zork"] |  | ||||||
| 	sendingAccount := suite.accounts["remote_account_1"] |  | ||||||
| 	inboxAccount := suite.accounts["local_account_1"] |  | ||||||
| 
 |  | ||||||
| 	fedWorker := worker.New[messages.FromFederator](-1, -1) |  | ||||||
| 
 |  | ||||||
| 	tc := testrig.NewTestTransportController(testrig.NewMockHTTPClient(nil), suite.db, fedWorker) |  | ||||||
| 	// now setup module being tested, with the mock transport controller |  | ||||||
| 	federator := federation.NewFederator(suite.db, testrig.NewTestFederatingDB(suite.db, fedWorker), tc, suite.typeConverter, testrig.NewTestMediaManager(suite.db, suite.storage)) |  | ||||||
| 
 |  | ||||||
| 	request := httptest.NewRequest(http.MethodPost, "http://localhost:8080/users/the_mighty_zork/inbox", nil) |  | ||||||
| 	// we need these headers for the request to be validated |  | ||||||
| 	request.Header.Set("Signature", activity.SignatureHeader) |  | ||||||
| 	request.Header.Set("Date", activity.DateHeader) |  | ||||||
| 	request.Header.Set("Digest", activity.DigestHeader) |  | ||||||
| 
 |  | ||||||
| 	verifier, err := httpsig.NewVerifier(request) |  | ||||||
| 	assert.NoError(suite.T(), err) |  | ||||||
| 
 |  | ||||||
| 	ctx := context.Background() |  | ||||||
| 	// by the time AuthenticatePostInbox is called, PostInboxRequestBodyHook should have already been called, |  | ||||||
| 	// which should have set the account and username onto the request. We can replicate that behavior here: |  | ||||||
| 	ctxWithAccount := context.WithValue(ctx, ap.ContextReceivingAccount, inboxAccount) |  | ||||||
| 	ctxWithActivity := context.WithValue(ctxWithAccount, ap.ContextActivity, activity) |  | ||||||
| 	ctxWithVerifier := context.WithValue(ctxWithActivity, ap.ContextRequestingPublicKeyVerifier, verifier) |  | ||||||
| 	ctxWithSignature := context.WithValue(ctxWithVerifier, ap.ContextRequestingPublicKeySignature, activity.SignatureHeader) |  | ||||||
| 
 |  | ||||||
| 	// we can pass this recorder as a writer and read it back after |  | ||||||
| 	recorder := httptest.NewRecorder() |  | ||||||
| 
 |  | ||||||
| 	// trigger the function being tested, and return the new context it creates |  | ||||||
| 	newContext, authed, err := federator.AuthenticatePostInbox(ctxWithSignature, recorder, request) |  | ||||||
| 	assert.NoError(suite.T(), err) |  | ||||||
| 	assert.True(suite.T(), authed) |  | ||||||
| 
 |  | ||||||
| 	// since we know this account already it should be set on the context |  | ||||||
| 	requestingAccountI := newContext.Value(ap.ContextRequestingAccount) |  | ||||||
| 	assert.NotNil(suite.T(), requestingAccountI) |  | ||||||
| 	requestingAccount, ok := requestingAccountI.(*gtsmodel.Account) |  | ||||||
| 	assert.True(suite.T(), ok) |  | ||||||
| 	assert.Equal(suite.T(), sendingAccount.Username, requestingAccount.Username) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestProtocolTestSuite(t *testing.T) { |  | ||||||
| 	suite.Run(t, new(ProtocolTestSuite)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
|  | @ -74,13 +74,12 @@ func (suite *IndexTestSuite) TestIndexBeforeLowID() { | ||||||
| 	err := suite.timeline.IndexBefore(context.Background(), "00000000000000000000000000", 10) | 	err := suite.timeline.IndexBefore(context.Background(), "00000000000000000000000000", 10) | ||||||
| 	suite.NoError(err) | 	suite.NoError(err) | ||||||
| 
 | 
 | ||||||
| 	// the oldest indexed post should be the lowest one we have in our testrig |  | ||||||
| 	postID, err := suite.timeline.OldestIndexedItemID(context.Background()) | 	postID, err := suite.timeline.OldestIndexedItemID(context.Background()) | ||||||
| 	suite.NoError(err) | 	suite.NoError(err) | ||||||
| 	suite.Equal("01F8MHAYFKS4KMXF8K5Y1C0KRN", postID) | 	suite.Equal("01F8MHBBN8120SYH7D5S050MGK", postID) | ||||||
| 
 | 
 | ||||||
| 	indexLength := suite.timeline.ItemIndexLength(context.Background()) | 	indexLength := suite.timeline.ItemIndexLength(context.Background()) | ||||||
| 	suite.Equal(10, indexLength) | 	suite.Equal(9, indexLength) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *IndexTestSuite) TestIndexBeforeHighID() { | func (suite *IndexTestSuite) TestIndexBeforeHighID() { | ||||||
|  | @ -108,9 +107,9 @@ func (suite *IndexTestSuite) TestIndexBehindHighID() { | ||||||
| 	suite.NoError(err) | 	suite.NoError(err) | ||||||
| 	suite.Equal("01FN3VJGFH10KR7S2PB0GFJZYG", postID) | 	suite.Equal("01FN3VJGFH10KR7S2PB0GFJZYG", postID) | ||||||
| 
 | 
 | ||||||
| 	// indexLength should be 10 because that's all this user has hometimelineable | 	// indexLength should be 9 because that's all this user has hometimelineable | ||||||
| 	indexLength := suite.timeline.ItemIndexLength(context.Background()) | 	indexLength := suite.timeline.ItemIndexLength(context.Background()) | ||||||
| 	suite.Equal(10, indexLength) | 	suite.Equal(9, indexLength) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *IndexTestSuite) TestIndexBehindLowID() { | func (suite *IndexTestSuite) TestIndexBehindLowID() { | ||||||
|  |  | ||||||
|  | @ -41,6 +41,7 @@ type FilterStandardTestSuite struct { | ||||||
| 	testStatuses     map[string]*gtsmodel.Status | 	testStatuses     map[string]*gtsmodel.Status | ||||||
| 	testTags         map[string]*gtsmodel.Tag | 	testTags         map[string]*gtsmodel.Tag | ||||||
| 	testMentions     map[string]*gtsmodel.Mention | 	testMentions     map[string]*gtsmodel.Mention | ||||||
|  | 	testFollows      map[string]*gtsmodel.Follow | ||||||
| 
 | 
 | ||||||
| 	filter visibility.Filter | 	filter visibility.Filter | ||||||
| } | } | ||||||
|  | @ -55,6 +56,7 @@ func (suite *FilterStandardTestSuite) SetupSuite() { | ||||||
| 	suite.testStatuses = testrig.NewTestStatuses() | 	suite.testStatuses = testrig.NewTestStatuses() | ||||||
| 	suite.testTags = testrig.NewTestTags() | 	suite.testTags = testrig.NewTestTags() | ||||||
| 	suite.testMentions = testrig.NewTestMentions() | 	suite.testMentions = testrig.NewTestMentions() | ||||||
|  | 	suite.testFollows = testrig.NewTestFollows() | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func (suite *FilterStandardTestSuite) SetupTest() { | func (suite *FilterStandardTestSuite) SetupTest() { | ||||||
|  |  | ||||||
|  | @ -23,6 +23,7 @@ import ( | ||||||
| 	"testing" | 	"testing" | ||||||
| 
 | 
 | ||||||
| 	"github.com/stretchr/testify/suite" | 	"github.com/stretchr/testify/suite" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| type StatusVisibleTestSuite struct { | type StatusVisibleTestSuite struct { | ||||||
|  | @ -85,6 +86,8 @@ func (suite *StatusVisibleTestSuite) TestDMNotVisibleIfNotMentioned() { | ||||||
| func (suite *StatusVisibleTestSuite) TestStatusNotVisibleIfNotMutuals() { | func (suite *StatusVisibleTestSuite) TestStatusNotVisibleIfNotMutuals() { | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 
 | 
 | ||||||
|  | 	suite.db.DeleteByID(ctx, suite.testFollows["local_account_2_local_account_1"].ID, >smodel.Follow{}) | ||||||
|  | 
 | ||||||
| 	testStatusID := suite.testStatuses["local_account_1_status_4"].ID | 	testStatusID := suite.testStatuses["local_account_1_status_4"].ID | ||||||
| 	testStatus, err := suite.db.GetStatusByID(ctx, testStatusID) | 	testStatus, err := suite.db.GetStatusByID(ctx, testStatusID) | ||||||
| 	suite.NoError(err) | 	suite.NoError(err) | ||||||
|  | @ -99,6 +102,8 @@ func (suite *StatusVisibleTestSuite) TestStatusNotVisibleIfNotMutuals() { | ||||||
| func (suite *StatusVisibleTestSuite) TestStatusNotVisibleIfNotFollowing() { | func (suite *StatusVisibleTestSuite) TestStatusNotVisibleIfNotFollowing() { | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 
 | 
 | ||||||
|  | 	suite.db.DeleteByID(ctx, suite.testFollows["admin_account_local_account_1"].ID, >smodel.Follow{}) | ||||||
|  | 
 | ||||||
| 	testStatusID := suite.testStatuses["local_account_1_status_5"].ID | 	testStatusID := suite.testStatuses["local_account_1_status_5"].ID | ||||||
| 	testStatus, err := suite.db.GetStatusByID(ctx, testStatusID) | 	testStatus, err := suite.db.GetStatusByID(ctx, testStatusID) | ||||||
| 	suite.NoError(err) | 	suite.NoError(err) | ||||||
|  |  | ||||||
|  | @ -1435,6 +1435,26 @@ func NewTestFollows() map[string]*gtsmodel.Follow { | ||||||
| 			URI:             "http://localhost:8080/users/the_mighty_zork/follow/01F8PYDCE8XE23GRE5DPZJDZDP", | 			URI:             "http://localhost:8080/users/the_mighty_zork/follow/01F8PYDCE8XE23GRE5DPZJDZDP", | ||||||
| 			Notify:          false, | 			Notify:          false, | ||||||
| 		}, | 		}, | ||||||
|  | 		"local_account_2_local_account_1": { | ||||||
|  | 			ID:              "01G1TK1RS4K3E0MSFTXBFWAH9Q", | ||||||
|  | 			CreatedAt:       time.Now().Add(-1 * time.Hour), | ||||||
|  | 			UpdatedAt:       time.Now().Add(-1 * time.Hour), | ||||||
|  | 			AccountID:       "01F8MH5NBDF2MV7CTC4Q5128HF", | ||||||
|  | 			TargetAccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", | ||||||
|  | 			ShowReblogs:     true, | ||||||
|  | 			URI:             "http://localhost:8080/users/1happyturtle/follow/01F8PYDCE8XE23GRE5DPZJDZDP", | ||||||
|  | 			Notify:          false, | ||||||
|  | 		}, | ||||||
|  | 		"admin_account_local_account_1": { | ||||||
|  | 			ID:              "01G1TK3PQKFW1BQZ9WVYRTFECK", | ||||||
|  | 			CreatedAt:       time.Now().Add(-46 * time.Hour), | ||||||
|  | 			UpdatedAt:       time.Now().Add(-46 * time.Hour), | ||||||
|  | 			AccountID:       "01F8MH17FWEB39HZJ76B6VXSKF", | ||||||
|  | 			TargetAccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", | ||||||
|  | 			ShowReblogs:     true, | ||||||
|  | 			URI:             "http://localhost:8080/users/admin/follow/01G1TK3PQKFW1BQZ9WVYRTFECK", | ||||||
|  | 			Notify:          false, | ||||||
|  | 		}, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue