[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

1 files changed, 5 insertions, 99 deletions

diff --git a/internal/federation/federator_test.go b/internal/federation/federator_test.go
index 220c3a193..68ad606f8 100644
--- a/internal/federation/federator_test.go
+++ b/internal/federation/federator_test.go
@@ -19,28 +19,16 @@
 package federation_test
 
 import (
-	"context"
-	"net/http"
-	"net/http/httptest"
-	"testing"
-
 	"codeberg.org/gruf/go-store/kv"
-	"github.com/go-fed/httpsig"
-	"github.com/stretchr/testify/assert"
 	"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/federation"
 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
-	"github.com/superseriousbusiness/gotosocial/internal/messages"
 	"github.com/superseriousbusiness/gotosocial/internal/typeutils"
-	"github.com/superseriousbusiness/gotosocial/internal/worker"
 	"github.com/superseriousbusiness/gotosocial/testrig"
 )
 
-type ProtocolTestSuite struct {
+type FederatorStandardTestSuite struct {
 	suite.Suite
 	db            db.DB
 	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
-func (suite *ProtocolTestSuite) SetupSuite() {
+func (suite *FederatorStandardTestSuite) SetupSuite() {
 	// setup standard items
 	suite.storage = testrig.NewTestStorage()
 	suite.typeConverter = testrig.NewTestTypeConverter(suite.db)
 	suite.accounts = testrig.NewTestAccounts()
 }
 
-func (suite *ProtocolTestSuite) SetupTest() {
-	testrig.InitTestLog()
+func (suite *FederatorStandardTestSuite) SetupTest() {
 	testrig.InitTestConfig()
+	testrig.InitTestLog()
 	suite.db = testrig.NewTestDB()
 	suite.activities = testrig.NewTestActivities(suite.accounts)
 	testrig.StandardDBSetup(suite.db, suite.accounts)
 }
 
 // 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)
 }
-
-// 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))
-}