mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-30 19:10:31 +00:00
a483bd9e38
* add delivery worker type that pulls from queue to httpclient package * finish up some code commenting, bodge a vendored activity library change, integrate the deliverypool changes into transportcontroller * hook up queue deletion logic * support deleting queued http requests by target ID * don't index APRequest by hostname in the queue * use gorun * use the original context's values when wrapping msg type as delivery{} * actually log in the AP delivery worker ... * add uncommitted changes * use errors.AsV2() * use errorsv2.AsV2() * finish adding some code comments, add bad host handling to delivery workers * slightly tweak deliveryworkerpool API, use advanced sender multiplier * remove PopCtx() method, let others instead rely on Wait() * shuffle things around to move delivery stuff into transport/ subpkg * remove dead code * formatting * validate request before queueing for delivery * finish adding code comments, fix up backoff code * finish adding more code comments * clamp minimum no. senders to 1 * add start/stop logging to delivery worker, some slight changes * remove double logging * use worker ptrs * expose the embedded log fields in httpclient.Request{} * ensure request context values are preserved when updating ctx * add delivery worker tests * fix linter issues * ensure delivery worker gets inited in testrig * fix tests to delivering messages to check worker delivery queue * update error type to use ptr instead of value receiver * fix test calling Workers{}.Start() instead of testrig.StartWorkers() * update docs for advanced-sender-multiplier * update to the latest activity library version * add comment about not using httptest.Server{}
106 lines
3.3 KiB
Go
106 lines
3.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 processing_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/superseriousbusiness/activity/pub"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
"github.com/superseriousbusiness/gotosocial/testrig"
|
|
)
|
|
|
|
type AccountTestSuite struct {
|
|
ProcessingStandardTestSuite
|
|
}
|
|
|
|
func (suite *AccountTestSuite) TestAccountDeleteLocal() {
|
|
ctx := context.Background()
|
|
deletingAccount := suite.testAccounts["local_account_1"]
|
|
followingAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
// make the following account follow the deleting account so that a delete message will be sent to it via the federating API
|
|
follow := >smodel.Follow{
|
|
ID: "01FJ1S8DX3STJJ6CEYPMZ1M0R3",
|
|
CreatedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
URI: fmt.Sprintf("%s/follow/01FJ1S8DX3STJJ6CEYPMZ1M0R3", followingAccount.URI),
|
|
AccountID: followingAccount.ID,
|
|
TargetAccountID: deletingAccount.ID,
|
|
}
|
|
err := suite.db.Put(ctx, follow)
|
|
suite.NoError(err)
|
|
|
|
errWithCode := suite.processor.Account().DeleteSelf(ctx, suite.testAccounts["local_account_1"])
|
|
suite.NoError(errWithCode)
|
|
|
|
// the delete should be federated outwards to the following account's inbox
|
|
var sent []byte
|
|
delete := new(struct {
|
|
Actor string `json:"actor"`
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
To string `json:"to"`
|
|
CC string `json:"cc"`
|
|
Type string `json:"type"`
|
|
})
|
|
|
|
if !testrig.WaitFor(func() bool {
|
|
delivery, ok := suite.state.Workers.Delivery.Queue.Pop()
|
|
if !ok {
|
|
return false
|
|
}
|
|
if !testrig.EqualRequestURIs(delivery.Request.URL, *followingAccount.SharedInboxURI) {
|
|
panic("differing request uris")
|
|
}
|
|
sent, err = io.ReadAll(delivery.Request.Body)
|
|
if err != nil {
|
|
panic("error reading body: " + err.Error())
|
|
}
|
|
err = json.Unmarshal(sent, delete)
|
|
if err != nil {
|
|
panic("error unmarshaling json: " + err.Error())
|
|
}
|
|
return true
|
|
}) {
|
|
suite.FailNow("timed out waiting for message")
|
|
}
|
|
|
|
suite.Equal(deletingAccount.URI, delete.Actor)
|
|
suite.Equal(deletingAccount.URI, delete.Object)
|
|
suite.Equal(deletingAccount.FollowersURI, delete.To)
|
|
suite.Equal(pub.PublicActivityPubIRI, delete.CC)
|
|
suite.Equal("Delete", delete.Type)
|
|
|
|
if !testrig.WaitFor(func() bool {
|
|
dbAccount, _ := suite.db.GetAccountByID(ctx, deletingAccount.ID)
|
|
return !dbAccount.SuspendedAt.IsZero()
|
|
}) {
|
|
suite.FailNow("timed out waiting for account to be deleted")
|
|
}
|
|
}
|
|
|
|
func TestAccountTestSuite(t *testing.T) {
|
|
suite.Run(t, &AccountTestSuite{})
|
|
}
|