2023-03-12 15:00:57 +00:00
|
|
|
// 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/>.
|
2021-09-04 11:29:56 +00:00
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
package workers_test
|
2021-09-04 11:29:56 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2021-10-01 17:08:50 +00:00
|
|
|
"encoding/json"
|
2021-09-30 08:56:02 +00:00
|
|
|
"fmt"
|
2021-09-04 11:29:56 +00:00
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/ap"
|
2023-01-02 12:10:50 +00:00
|
|
|
apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
|
2021-09-04 11:29:56 +00:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/messages"
|
2021-11-22 18:03:21 +00:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/stream"
|
2023-08-07 17:38:11 +00:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/util"
|
2021-10-10 10:39:25 +00:00
|
|
|
"github.com/superseriousbusiness/gotosocial/testrig"
|
2021-09-04 11:29:56 +00:00
|
|
|
)
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
type FromFediAPITestSuite struct {
|
|
|
|
WorkersTestSuite
|
2021-09-04 11:29:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// remote_account_1 boosts the first status of local_account_1
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessFederationAnnounce() {
|
2021-09-04 11:29:56 +00:00
|
|
|
boostedStatus := suite.testStatuses["local_account_1_status_1"]
|
|
|
|
boostingAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
announceStatus := >smodel.Status{}
|
|
|
|
announceStatus.URI = "https://example.org/some-announce-uri"
|
2023-12-01 14:27:15 +00:00
|
|
|
announceStatus.BoostOfURI = boostedStatus.URI
|
2021-09-04 11:29:56 +00:00
|
|
|
announceStatus.CreatedAt = time.Now()
|
|
|
|
announceStatus.UpdatedAt = time.Now()
|
|
|
|
announceStatus.AccountID = boostingAccount.ID
|
|
|
|
announceStatus.AccountURI = boostingAccount.URI
|
|
|
|
announceStatus.Account = boostingAccount
|
|
|
|
announceStatus.Visibility = boostedStatus.Visibility
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err := suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
|
2021-09-04 11:29:56 +00:00
|
|
|
APObjectType: ap.ActivityAnnounce,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: announceStatus,
|
|
|
|
ReceivingAccount: suite.testAccounts["local_account_1"],
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// side effects should be triggered
|
|
|
|
// 1. status should have an ID, and be in the database
|
|
|
|
suite.NotEmpty(announceStatus.ID)
|
|
|
|
_, err = suite.db.GetStatusByID(context.Background(), announceStatus.ID)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// 2. a notification should exist for the announce
|
|
|
|
where := []db.Where{
|
|
|
|
{
|
|
|
|
Key: "status_id",
|
|
|
|
Value: announceStatus.ID,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
notif := >smodel.Notification{}
|
|
|
|
err = suite.db.GetWhere(context.Background(), where, notif)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal(gtsmodel.NotificationReblog, notif.NotificationType)
|
|
|
|
suite.Equal(boostedStatus.AccountID, notif.TargetAccountID)
|
|
|
|
suite.Equal(announceStatus.AccountID, notif.OriginAccountID)
|
|
|
|
suite.Equal(announceStatus.ID, notif.StatusID)
|
2022-08-15 10:35:05 +00:00
|
|
|
suite.False(*notif.Read)
|
2021-09-04 11:29:56 +00:00
|
|
|
}
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessReplyMention() {
|
2021-09-14 10:23:56 +00:00
|
|
|
repliedAccount := suite.testAccounts["local_account_1"]
|
|
|
|
repliedStatus := suite.testStatuses["local_account_1_status_1"]
|
|
|
|
replyingAccount := suite.testAccounts["remote_account_1"]
|
2022-08-15 10:35:05 +00:00
|
|
|
|
2023-11-04 20:21:20 +00:00
|
|
|
// Set the replyingAccount's last fetched_at
|
2024-03-04 12:30:12 +00:00
|
|
|
// date to something recent so no refresh is attempted,
|
|
|
|
// and ensure it isn't a suspended account.
|
2023-11-04 20:21:20 +00:00
|
|
|
replyingAccount.FetchedAt = time.Now()
|
2024-03-04 12:30:12 +00:00
|
|
|
replyingAccount.SuspendedAt = time.Time{}
|
|
|
|
replyingAccount.SuspensionOrigin = ""
|
|
|
|
err := suite.state.DB.UpdateAccount(context.Background(),
|
|
|
|
replyingAccount,
|
|
|
|
"fetched_at",
|
|
|
|
"suspended_at",
|
|
|
|
"suspension_origin",
|
|
|
|
)
|
2023-11-04 20:21:20 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// Get replying statusable to use from remote test statuses.
|
|
|
|
const replyingURI = "http://fossbros-anonymous.io/users/foss_satan/statuses/106221634728637552"
|
|
|
|
replyingStatusable := testrig.NewTestFediStatuses()[replyingURI]
|
|
|
|
ap.AppendInReplyTo(replyingStatusable, testrig.URLMustParse(repliedStatus.URI))
|
2021-09-14 10:23:56 +00:00
|
|
|
|
2023-11-04 20:21:20 +00:00
|
|
|
// Open a websocket stream to later test the streamed status reply.
|
2023-02-22 15:05:26 +00:00
|
|
|
wssStream, errWithCode := suite.processor.Stream().Open(context.Background(), repliedAccount, stream.TimelineHome)
|
2021-11-22 18:03:21 +00:00
|
|
|
suite.NoError(errWithCode)
|
|
|
|
|
2023-11-04 20:21:20 +00:00
|
|
|
// Send the replied status off to the fedi worker to be further processed.
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
|
2021-09-14 10:23:56 +00:00
|
|
|
APObjectType: ap.ObjectNote,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
2023-11-04 20:21:20 +00:00
|
|
|
APObjectModel: replyingStatusable,
|
2021-09-14 10:23:56 +00:00
|
|
|
ReceivingAccount: suite.testAccounts["local_account_1"],
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// side effects should be triggered
|
|
|
|
// 1. status should be in the database
|
2023-11-04 20:21:20 +00:00
|
|
|
replyingStatus, err := suite.state.DB.GetStatusByURI(context.Background(), replyingURI)
|
2021-09-14 10:23:56 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// 2. a notification should exist for the mention
|
2023-05-12 09:15:54 +00:00
|
|
|
var notif gtsmodel.Notification
|
|
|
|
err = suite.db.GetWhere(context.Background(), []db.Where{
|
|
|
|
{Key: "status_id", Value: replyingStatus.ID},
|
|
|
|
}, ¬if)
|
2021-09-14 10:23:56 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal(gtsmodel.NotificationMention, notif.NotificationType)
|
|
|
|
suite.Equal(replyingStatus.InReplyToAccountID, notif.TargetAccountID)
|
|
|
|
suite.Equal(replyingStatus.AccountID, notif.OriginAccountID)
|
|
|
|
suite.Equal(replyingStatus.ID, notif.StatusID)
|
2022-08-15 10:35:05 +00:00
|
|
|
suite.False(*notif.Read)
|
|
|
|
|
2024-02-20 18:07:49 +00:00
|
|
|
ctx, _ := context.WithTimeout(context.Background(), time.Second*5)
|
|
|
|
msg, ok := wssStream.Recv(ctx)
|
|
|
|
suite.True(ok)
|
2021-11-22 18:03:21 +00:00
|
|
|
|
|
|
|
suite.Equal(stream.EventTypeNotification, msg.Event)
|
|
|
|
suite.NotEmpty(msg.Payload)
|
|
|
|
suite.EqualValues([]string{stream.TimelineHome}, msg.Stream)
|
2023-01-02 12:10:50 +00:00
|
|
|
notifStreamed := &apimodel.Notification{}
|
2021-11-22 18:03:21 +00:00
|
|
|
err = json.Unmarshal([]byte(msg.Payload), notifStreamed)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal("mention", notifStreamed.Type)
|
|
|
|
suite.Equal(replyingAccount.ID, notifStreamed.Account.ID)
|
2021-09-14 10:23:56 +00:00
|
|
|
}
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessFave() {
|
2021-09-27 15:42:20 +00:00
|
|
|
favedAccount := suite.testAccounts["local_account_1"]
|
|
|
|
favedStatus := suite.testStatuses["local_account_1_status_1"]
|
|
|
|
favingAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
|
2023-02-22 15:05:26 +00:00
|
|
|
wssStream, errWithCode := suite.processor.Stream().Open(context.Background(), favedAccount, stream.TimelineNotifications)
|
2021-09-27 15:42:20 +00:00
|
|
|
suite.NoError(errWithCode)
|
|
|
|
|
|
|
|
fave := >smodel.StatusFave{
|
|
|
|
ID: "01FGKJPXFTVQPG9YSSZ95ADS7Q",
|
|
|
|
CreatedAt: time.Now(),
|
|
|
|
UpdatedAt: time.Now(),
|
|
|
|
AccountID: favingAccount.ID,
|
|
|
|
Account: favingAccount,
|
|
|
|
TargetAccountID: favedAccount.ID,
|
|
|
|
TargetAccount: favedAccount,
|
|
|
|
StatusID: favedStatus.ID,
|
|
|
|
Status: favedStatus,
|
|
|
|
URI: favingAccount.URI + "/faves/aaaaaaaaaaaa",
|
|
|
|
}
|
|
|
|
|
|
|
|
err := suite.db.Put(context.Background(), fave)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
|
2021-09-27 15:42:20 +00:00
|
|
|
APObjectType: ap.ActivityLike,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: fave,
|
|
|
|
ReceivingAccount: favedAccount,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// side effects should be triggered
|
|
|
|
// 1. a notification should exist for the fave
|
|
|
|
where := []db.Where{
|
|
|
|
{
|
|
|
|
Key: "status_id",
|
|
|
|
Value: favedStatus.ID,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Key: "origin_account_id",
|
|
|
|
Value: favingAccount.ID,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
notif := >smodel.Notification{}
|
|
|
|
err = suite.db.GetWhere(context.Background(), where, notif)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal(gtsmodel.NotificationFave, notif.NotificationType)
|
|
|
|
suite.Equal(fave.TargetAccountID, notif.TargetAccountID)
|
|
|
|
suite.Equal(fave.AccountID, notif.OriginAccountID)
|
|
|
|
suite.Equal(fave.StatusID, notif.StatusID)
|
2022-08-15 10:35:05 +00:00
|
|
|
suite.False(*notif.Read)
|
2021-09-27 15:42:20 +00:00
|
|
|
|
2024-02-20 18:07:49 +00:00
|
|
|
ctx, _ := context.WithTimeout(context.Background(), time.Second*5)
|
|
|
|
msg, ok := wssStream.Recv(ctx)
|
|
|
|
suite.True(ok)
|
|
|
|
|
2021-11-22 18:03:21 +00:00
|
|
|
suite.Equal(stream.EventTypeNotification, msg.Event)
|
2021-09-27 15:42:20 +00:00
|
|
|
suite.NotEmpty(msg.Payload)
|
2021-11-22 18:03:21 +00:00
|
|
|
suite.EqualValues([]string{stream.TimelineNotifications}, msg.Stream)
|
2021-09-27 15:42:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// TestProcessFaveWithDifferentReceivingAccount ensures that when an account receives a fave that's for
|
|
|
|
// another account in their AP inbox, a notification isn't streamed to the receiving account.
|
|
|
|
//
|
|
|
|
// This tests for an issue we were seeing where Misskey sends out faves to inboxes of people that don't own
|
|
|
|
// the fave, but just follow the actor who received the fave.
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessFaveWithDifferentReceivingAccount() {
|
2021-09-27 15:42:20 +00:00
|
|
|
receivingAccount := suite.testAccounts["local_account_2"]
|
|
|
|
favedAccount := suite.testAccounts["local_account_1"]
|
|
|
|
favedStatus := suite.testStatuses["local_account_1_status_1"]
|
|
|
|
favingAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
|
2023-02-22 15:05:26 +00:00
|
|
|
wssStream, errWithCode := suite.processor.Stream().Open(context.Background(), receivingAccount, stream.TimelineHome)
|
2021-09-27 15:42:20 +00:00
|
|
|
suite.NoError(errWithCode)
|
|
|
|
|
|
|
|
fave := >smodel.StatusFave{
|
|
|
|
ID: "01FGKJPXFTVQPG9YSSZ95ADS7Q",
|
|
|
|
CreatedAt: time.Now(),
|
|
|
|
UpdatedAt: time.Now(),
|
|
|
|
AccountID: favingAccount.ID,
|
|
|
|
Account: favingAccount,
|
|
|
|
TargetAccountID: favedAccount.ID,
|
|
|
|
TargetAccount: favedAccount,
|
|
|
|
StatusID: favedStatus.ID,
|
|
|
|
Status: favedStatus,
|
|
|
|
URI: favingAccount.URI + "/faves/aaaaaaaaaaaa",
|
|
|
|
}
|
|
|
|
|
|
|
|
err := suite.db.Put(context.Background(), fave)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
|
2021-09-27 15:42:20 +00:00
|
|
|
APObjectType: ap.ActivityLike,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: fave,
|
|
|
|
ReceivingAccount: receivingAccount,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// side effects should be triggered
|
|
|
|
// 1. a notification should exist for the fave
|
|
|
|
where := []db.Where{
|
|
|
|
{
|
|
|
|
Key: "status_id",
|
|
|
|
Value: favedStatus.ID,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Key: "origin_account_id",
|
|
|
|
Value: favingAccount.ID,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
notif := >smodel.Notification{}
|
|
|
|
err = suite.db.GetWhere(context.Background(), where, notif)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal(gtsmodel.NotificationFave, notif.NotificationType)
|
|
|
|
suite.Equal(fave.TargetAccountID, notif.TargetAccountID)
|
|
|
|
suite.Equal(fave.AccountID, notif.OriginAccountID)
|
|
|
|
suite.Equal(fave.StatusID, notif.StatusID)
|
2022-08-15 10:35:05 +00:00
|
|
|
suite.False(*notif.Read)
|
2021-09-27 15:42:20 +00:00
|
|
|
|
|
|
|
// 2. no notification should be streamed to the account that received the fave message, because they weren't the target
|
2024-02-20 18:07:49 +00:00
|
|
|
ctx, _ := context.WithTimeout(context.Background(), time.Second*5)
|
|
|
|
_, ok := wssStream.Recv(ctx)
|
|
|
|
suite.False(ok)
|
2021-09-27 15:42:20 +00:00
|
|
|
}
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessAccountDelete() {
|
2021-09-30 08:56:02 +00:00
|
|
|
ctx := context.Background()
|
|
|
|
|
|
|
|
deletedAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
receivingAccount := suite.testAccounts["local_account_1"]
|
|
|
|
|
|
|
|
// before doing the delete....
|
|
|
|
// make local_account_1 and remote_account_1 into mufos
|
|
|
|
zorkFollowSatan := >smodel.Follow{
|
|
|
|
ID: "01FGRY72ASHBSET64353DPHK9T",
|
|
|
|
CreatedAt: time.Now().Add(-1 * time.Hour),
|
|
|
|
UpdatedAt: time.Now().Add(-1 * time.Hour),
|
|
|
|
AccountID: deletedAccount.ID,
|
|
|
|
TargetAccountID: receivingAccount.ID,
|
2023-08-07 17:38:11 +00:00
|
|
|
ShowReblogs: util.Ptr(true),
|
2021-09-30 08:56:02 +00:00
|
|
|
URI: fmt.Sprintf("%s/follows/01FGRY72ASHBSET64353DPHK9T", deletedAccount.URI),
|
2023-08-07 17:38:11 +00:00
|
|
|
Notify: util.Ptr(false),
|
2021-09-30 08:56:02 +00:00
|
|
|
}
|
|
|
|
err := suite.db.Put(ctx, zorkFollowSatan)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
satanFollowZork := >smodel.Follow{
|
|
|
|
ID: "01FGRYAVAWWPP926J175QGM0WV",
|
|
|
|
CreatedAt: time.Now().Add(-1 * time.Hour),
|
|
|
|
UpdatedAt: time.Now().Add(-1 * time.Hour),
|
|
|
|
AccountID: receivingAccount.ID,
|
|
|
|
TargetAccountID: deletedAccount.ID,
|
2023-08-07 17:38:11 +00:00
|
|
|
ShowReblogs: util.Ptr(true),
|
2021-09-30 08:56:02 +00:00
|
|
|
URI: fmt.Sprintf("%s/follows/01FGRYAVAWWPP926J175QGM0WV", receivingAccount.URI),
|
2023-08-07 17:38:11 +00:00
|
|
|
Notify: util.Ptr(false),
|
2021-09-30 08:56:02 +00:00
|
|
|
}
|
|
|
|
err = suite.db.Put(ctx, satanFollowZork)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// now they are mufos!
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
|
2021-09-30 08:56:02 +00:00
|
|
|
APObjectType: ap.ObjectProfile,
|
|
|
|
APActivityType: ap.ActivityDelete,
|
|
|
|
GTSModel: deletedAccount,
|
|
|
|
ReceivingAccount: receivingAccount,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// local account 2 blocked foss_satan, that block should be gone now
|
|
|
|
testBlock := suite.testBlocks["local_account_2_block_remote_account_1"]
|
|
|
|
dbBlock := >smodel.Block{}
|
|
|
|
err = suite.db.GetByID(ctx, testBlock.ID, dbBlock)
|
|
|
|
suite.ErrorIs(err, db.ErrNoEntries)
|
|
|
|
|
|
|
|
// the mufos should be gone now too
|
[performance] refactoring + add fave / follow / request / visibility caching (#1607)
* refactor visibility checking, add caching for visibility
* invalidate visibility cache items on account / status deletes
* fix requester ID passed to visibility cache nil ptr
* de-interface caches, fix home / public timeline caching + visibility
* finish adding code comments for visibility filter
* fix angry goconst linter warnings
* actually finish adding filter visibility code comments for timeline functions
* move home timeline status author check to after visibility
* remove now-unused code
* add more code comments
* add TODO code comment, update printed cache start names
* update printed cache names on stop
* start adding separate follow(request) delete db functions, add specific visibility cache tests
* add relationship type caching
* fix getting local account follows / followed-bys, other small codebase improvements
* simplify invalidation using cache hooks, add more GetAccountBy___() functions
* fix boosting to return 404 if not boostable but no error (to not leak status ID)
* remove dead code
* improved placement of cache invalidation
* update license headers
* add example follow, follow-request config entries
* add example visibility cache configuration to config file
* use specific PutFollowRequest() instead of just Put()
* add tests for all GetAccountBy()
* add GetBlockBy() tests
* update block to check primitive fields
* update and finish adding Get{Account,Block,Follow,FollowRequest}By() tests
* fix copy-pasted code
* update envparsing test
* whitespace
* fix bun struct tag
* add license header to gtscontext
* fix old license header
* improved error creation to not use fmt.Errorf() when not needed
* fix various rebase conflicts, fix account test
* remove commented-out code, fix-up mention caching
* fix mention select bun statement
* ensure mention target account populated, pass in context to customrenderer logging
* remove more uncommented code, fix typeutil test
* add statusfave database model caching
* add status fave cache configuration
* add status fave cache example config
* woops, catch missed error. nice catch linter!
* add back testrig panic on nil db
* update example configuration to match defaults, slight tweak to cache configuration defaults
* update envparsing test with new defaults
* fetch followingget to use the follow target account
* use accounnt.IsLocal() instead of empty domain check
* use constants for the cache visibility type check
* use bun.In() for notification type restriction in db query
* include replies when fetching PublicTimeline() (to account for single-author threads in Visibility{}.StatusPublicTimelineable())
* use bun query building for nested select statements to ensure working with postgres
* update public timeline future status checks to match visibility filter
* same as previous, for home timeline
* update public timeline tests to dynamically check for appropriate statuses
* migrate accounts to allow unique constraint on public_key
* provide minimal account with publicKey
---------
Signed-off-by: kim <grufwub@gmail.com>
Co-authored-by: tsmethurst <tobi.smethurst@protonmail.com>
2023-03-28 13:03:14 +00:00
|
|
|
satanFollowsZork, err := suite.db.IsFollowing(ctx, deletedAccount.ID, receivingAccount.ID)
|
2021-09-30 08:56:02 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
suite.False(satanFollowsZork)
|
[performance] refactoring + add fave / follow / request / visibility caching (#1607)
* refactor visibility checking, add caching for visibility
* invalidate visibility cache items on account / status deletes
* fix requester ID passed to visibility cache nil ptr
* de-interface caches, fix home / public timeline caching + visibility
* finish adding code comments for visibility filter
* fix angry goconst linter warnings
* actually finish adding filter visibility code comments for timeline functions
* move home timeline status author check to after visibility
* remove now-unused code
* add more code comments
* add TODO code comment, update printed cache start names
* update printed cache names on stop
* start adding separate follow(request) delete db functions, add specific visibility cache tests
* add relationship type caching
* fix getting local account follows / followed-bys, other small codebase improvements
* simplify invalidation using cache hooks, add more GetAccountBy___() functions
* fix boosting to return 404 if not boostable but no error (to not leak status ID)
* remove dead code
* improved placement of cache invalidation
* update license headers
* add example follow, follow-request config entries
* add example visibility cache configuration to config file
* use specific PutFollowRequest() instead of just Put()
* add tests for all GetAccountBy()
* add GetBlockBy() tests
* update block to check primitive fields
* update and finish adding Get{Account,Block,Follow,FollowRequest}By() tests
* fix copy-pasted code
* update envparsing test
* whitespace
* fix bun struct tag
* add license header to gtscontext
* fix old license header
* improved error creation to not use fmt.Errorf() when not needed
* fix various rebase conflicts, fix account test
* remove commented-out code, fix-up mention caching
* fix mention select bun statement
* ensure mention target account populated, pass in context to customrenderer logging
* remove more uncommented code, fix typeutil test
* add statusfave database model caching
* add status fave cache configuration
* add status fave cache example config
* woops, catch missed error. nice catch linter!
* add back testrig panic on nil db
* update example configuration to match defaults, slight tweak to cache configuration defaults
* update envparsing test with new defaults
* fetch followingget to use the follow target account
* use accounnt.IsLocal() instead of empty domain check
* use constants for the cache visibility type check
* use bun.In() for notification type restriction in db query
* include replies when fetching PublicTimeline() (to account for single-author threads in Visibility{}.StatusPublicTimelineable())
* use bun query building for nested select statements to ensure working with postgres
* update public timeline future status checks to match visibility filter
* same as previous, for home timeline
* update public timeline tests to dynamically check for appropriate statuses
* migrate accounts to allow unique constraint on public_key
* provide minimal account with publicKey
---------
Signed-off-by: kim <grufwub@gmail.com>
Co-authored-by: tsmethurst <tobi.smethurst@protonmail.com>
2023-03-28 13:03:14 +00:00
|
|
|
zorkFollowsSatan, err := suite.db.IsFollowing(ctx, receivingAccount.ID, deletedAccount.ID)
|
2021-09-30 08:56:02 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
suite.False(zorkFollowsSatan)
|
|
|
|
|
|
|
|
// no statuses from foss satan should be left in the database
|
2022-09-21 17:55:52 +00:00
|
|
|
if !testrig.WaitFor(func() bool {
|
2023-02-25 12:16:30 +00:00
|
|
|
s, err := suite.db.GetAccountStatuses(ctx, deletedAccount.ID, 0, false, false, "", "", false, false)
|
2022-10-03 08:46:11 +00:00
|
|
|
return s == nil && err == db.ErrNoEntries
|
2022-09-21 17:55:52 +00:00
|
|
|
}) {
|
|
|
|
suite.FailNow("timeout waiting for statuses to be deleted")
|
|
|
|
}
|
2021-09-30 08:56:02 +00:00
|
|
|
|
2023-11-08 14:32:17 +00:00
|
|
|
var dbAccount *gtsmodel.Account
|
|
|
|
|
|
|
|
// account data should be zeroed.
|
|
|
|
if !testrig.WaitFor(func() bool {
|
|
|
|
dbAccount, err = suite.db.GetAccountByID(ctx, deletedAccount.ID)
|
|
|
|
return err == nil && dbAccount.DisplayName == ""
|
|
|
|
}) {
|
|
|
|
suite.FailNow("timeout waiting for statuses to be deleted")
|
|
|
|
}
|
2021-09-30 08:56:02 +00:00
|
|
|
|
|
|
|
suite.Empty(dbAccount.Note)
|
|
|
|
suite.Empty(dbAccount.DisplayName)
|
|
|
|
suite.Empty(dbAccount.AvatarMediaAttachmentID)
|
|
|
|
suite.Empty(dbAccount.AvatarRemoteURL)
|
|
|
|
suite.Empty(dbAccount.HeaderMediaAttachmentID)
|
|
|
|
suite.Empty(dbAccount.HeaderRemoteURL)
|
|
|
|
suite.Empty(dbAccount.Fields)
|
2022-08-15 10:35:05 +00:00
|
|
|
suite.False(*dbAccount.Discoverable)
|
2021-09-30 08:56:02 +00:00
|
|
|
suite.WithinDuration(time.Now(), dbAccount.SuspendedAt, 30*time.Second)
|
|
|
|
suite.Equal(dbAccount.ID, dbAccount.SuspensionOrigin)
|
|
|
|
}
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessFollowRequestLocked() {
|
2021-10-01 17:08:50 +00:00
|
|
|
ctx := context.Background()
|
|
|
|
|
|
|
|
originAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
|
|
|
|
// target is a locked account
|
|
|
|
targetAccount := suite.testAccounts["local_account_2"]
|
|
|
|
|
2023-02-22 15:05:26 +00:00
|
|
|
wssStream, errWithCode := suite.processor.Stream().Open(context.Background(), targetAccount, stream.TimelineHome)
|
2021-10-01 17:08:50 +00:00
|
|
|
suite.NoError(errWithCode)
|
|
|
|
|
|
|
|
// put the follow request in the database as though it had passed through the federating db already
|
|
|
|
satanFollowRequestTurtle := >smodel.FollowRequest{
|
|
|
|
ID: "01FGRYAVAWWPP926J175QGM0WV",
|
|
|
|
CreatedAt: time.Now(),
|
|
|
|
UpdatedAt: time.Now(),
|
|
|
|
AccountID: originAccount.ID,
|
|
|
|
Account: originAccount,
|
|
|
|
TargetAccountID: targetAccount.ID,
|
|
|
|
TargetAccount: targetAccount,
|
2023-08-07 17:38:11 +00:00
|
|
|
ShowReblogs: util.Ptr(true),
|
2021-10-01 17:08:50 +00:00
|
|
|
URI: fmt.Sprintf("%s/follows/01FGRYAVAWWPP926J175QGM0WV", originAccount.URI),
|
2023-08-07 17:38:11 +00:00
|
|
|
Notify: util.Ptr(false),
|
2021-10-01 17:08:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
err := suite.db.Put(ctx, satanFollowRequestTurtle)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
|
2021-10-01 17:08:50 +00:00
|
|
|
APObjectType: ap.ActivityFollow,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: satanFollowRequestTurtle,
|
|
|
|
ReceivingAccount: targetAccount,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2024-02-20 18:07:49 +00:00
|
|
|
ctx, _ = context.WithTimeout(ctx, time.Second*5)
|
|
|
|
msg, ok := wssStream.Recv(context.Background())
|
|
|
|
suite.True(ok)
|
|
|
|
|
2021-11-22 18:03:21 +00:00
|
|
|
suite.Equal(stream.EventTypeNotification, msg.Event)
|
2021-10-01 17:08:50 +00:00
|
|
|
suite.NotEmpty(msg.Payload)
|
2021-11-22 18:03:21 +00:00
|
|
|
suite.EqualValues([]string{stream.TimelineHome}, msg.Stream)
|
2023-01-02 12:10:50 +00:00
|
|
|
notif := &apimodel.Notification{}
|
2021-10-01 17:08:50 +00:00
|
|
|
err = json.Unmarshal([]byte(msg.Payload), notif)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal("follow_request", notif.Type)
|
|
|
|
suite.Equal(originAccount.ID, notif.Account.ID)
|
|
|
|
|
|
|
|
// no messages should have been sent out, since we didn't need to federate an accept
|
2024-02-20 18:07:49 +00:00
|
|
|
suite.Empty(&suite.httpClient.SentMessages)
|
2021-10-01 17:08:50 +00:00
|
|
|
}
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestProcessFollowRequestUnlocked() {
|
2021-10-01 17:08:50 +00:00
|
|
|
ctx := context.Background()
|
|
|
|
|
|
|
|
originAccount := suite.testAccounts["remote_account_1"]
|
|
|
|
|
|
|
|
// target is an unlocked account
|
|
|
|
targetAccount := suite.testAccounts["local_account_1"]
|
|
|
|
|
2023-02-22 15:05:26 +00:00
|
|
|
wssStream, errWithCode := suite.processor.Stream().Open(context.Background(), targetAccount, stream.TimelineHome)
|
2021-10-01 17:08:50 +00:00
|
|
|
suite.NoError(errWithCode)
|
|
|
|
|
|
|
|
// put the follow request in the database as though it had passed through the federating db already
|
|
|
|
satanFollowRequestTurtle := >smodel.FollowRequest{
|
|
|
|
ID: "01FGRYAVAWWPP926J175QGM0WV",
|
|
|
|
CreatedAt: time.Now(),
|
|
|
|
UpdatedAt: time.Now(),
|
|
|
|
AccountID: originAccount.ID,
|
|
|
|
Account: originAccount,
|
|
|
|
TargetAccountID: targetAccount.ID,
|
|
|
|
TargetAccount: targetAccount,
|
2023-08-07 17:38:11 +00:00
|
|
|
ShowReblogs: util.Ptr(true),
|
2021-10-01 17:08:50 +00:00
|
|
|
URI: fmt.Sprintf("%s/follows/01FGRYAVAWWPP926J175QGM0WV", originAccount.URI),
|
2023-08-07 17:38:11 +00:00
|
|
|
Notify: util.Ptr(false),
|
2021-10-01 17:08:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
err := suite.db.Put(ctx, satanFollowRequestTurtle)
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
|
2021-10-01 17:08:50 +00:00
|
|
|
APObjectType: ap.ActivityFollow,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: satanFollowRequestTurtle,
|
|
|
|
ReceivingAccount: targetAccount,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
2022-08-31 15:31:21 +00:00
|
|
|
// an accept message should be sent to satan's inbox
|
|
|
|
var sent [][]byte
|
|
|
|
if !testrig.WaitFor(func() bool {
|
2022-09-23 19:27:35 +00:00
|
|
|
sentI, ok := suite.httpClient.SentMessages.Load(*originAccount.SharedInboxURI)
|
2022-08-31 15:31:21 +00:00
|
|
|
if ok {
|
|
|
|
sent, ok = sentI.([][]byte)
|
|
|
|
if !ok {
|
|
|
|
panic("SentMessages entry was not []byte")
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}) {
|
|
|
|
suite.FailNow("timed out waiting for message")
|
2022-08-15 10:35:05 +00:00
|
|
|
}
|
2021-10-01 17:08:50 +00:00
|
|
|
|
|
|
|
accept := &struct {
|
|
|
|
Actor string `json:"actor"`
|
|
|
|
ID string `json:"id"`
|
|
|
|
Object struct {
|
|
|
|
Actor string `json:"actor"`
|
|
|
|
ID string `json:"id"`
|
|
|
|
Object string `json:"object"`
|
|
|
|
To string `json:"to"`
|
|
|
|
Type string `json:"type"`
|
|
|
|
}
|
|
|
|
To string `json:"to"`
|
|
|
|
Type string `json:"type"`
|
|
|
|
}{}
|
2022-08-31 15:31:21 +00:00
|
|
|
err = json.Unmarshal(sent[0], accept)
|
2021-10-01 17:08:50 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
suite.Equal(targetAccount.URI, accept.Actor)
|
|
|
|
suite.Equal(originAccount.URI, accept.Object.Actor)
|
|
|
|
suite.Equal(satanFollowRequestTurtle.URI, accept.Object.ID)
|
|
|
|
suite.Equal(targetAccount.URI, accept.Object.Object)
|
|
|
|
suite.Equal(targetAccount.URI, accept.Object.To)
|
|
|
|
suite.Equal("Follow", accept.Object.Type)
|
|
|
|
suite.Equal(originAccount.URI, accept.To)
|
|
|
|
suite.Equal("Accept", accept.Type)
|
2022-08-31 15:31:21 +00:00
|
|
|
|
2024-02-20 18:07:49 +00:00
|
|
|
ctx, _ = context.WithTimeout(ctx, time.Second*5)
|
|
|
|
msg, ok := wssStream.Recv(context.Background())
|
|
|
|
suite.True(ok)
|
|
|
|
|
2022-08-31 15:31:21 +00:00
|
|
|
suite.Equal(stream.EventTypeNotification, msg.Event)
|
|
|
|
suite.NotEmpty(msg.Payload)
|
|
|
|
suite.EqualValues([]string{stream.TimelineHome}, msg.Stream)
|
2023-01-02 12:10:50 +00:00
|
|
|
notif := &apimodel.Notification{}
|
2022-08-31 15:31:21 +00:00
|
|
|
err = json.Unmarshal([]byte(msg.Payload), notif)
|
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal("follow", notif.Type)
|
|
|
|
suite.Equal(originAccount.ID, notif.Account.ID)
|
2021-10-01 17:08:50 +00:00
|
|
|
}
|
|
|
|
|
2021-10-10 10:39:25 +00:00
|
|
|
// TestCreateStatusFromIRI checks if a forwarded status can be dereferenced by the processor.
|
2023-08-09 17:14:33 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestCreateStatusFromIRI() {
|
2021-10-10 10:39:25 +00:00
|
|
|
ctx := context.Background()
|
|
|
|
|
|
|
|
receivingAccount := suite.testAccounts["local_account_1"]
|
|
|
|
statusCreator := suite.testAccounts["remote_account_2"]
|
|
|
|
|
2023-08-09 17:14:33 +00:00
|
|
|
err := suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
|
2021-10-10 10:39:25 +00:00
|
|
|
APObjectType: ap.ObjectNote,
|
|
|
|
APActivityType: ap.ActivityCreate,
|
|
|
|
GTSModel: nil, // gtsmodel is nil because this is a forwarded status -- we want to dereference it using the iri
|
|
|
|
ReceivingAccount: receivingAccount,
|
2022-11-30 16:44:02 +00:00
|
|
|
APIri: testrig.URLMustParse("http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1"),
|
2021-10-10 10:39:25 +00:00
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// status should now be in the database, attributed to remote_account_2
|
2022-11-30 16:44:02 +00:00
|
|
|
s, err := suite.db.GetStatusByURI(context.Background(), "http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1")
|
2021-10-10 10:39:25 +00:00
|
|
|
suite.NoError(err)
|
|
|
|
suite.Equal(statusCreator.URI, s.AccountURI)
|
|
|
|
}
|
|
|
|
|
2024-03-12 14:34:08 +00:00
|
|
|
func (suite *FromFediAPITestSuite) TestMoveAccount() {
|
|
|
|
// We're gonna migrate foss_satan to our local admin account.
|
|
|
|
ctx := context.Background()
|
|
|
|
receivingAcct := suite.testAccounts["local_account_1"]
|
|
|
|
|
|
|
|
// Copy requesting and target accounts
|
|
|
|
// since we'll be changing these.
|
|
|
|
requestingAcct := >smodel.Account{}
|
|
|
|
*requestingAcct = *suite.testAccounts["remote_account_1"]
|
|
|
|
targetAcct := >smodel.Account{}
|
|
|
|
*targetAcct = *suite.testAccounts["admin_account"]
|
|
|
|
|
|
|
|
// Set alsoKnownAs on the admin account.
|
|
|
|
targetAcct.AlsoKnownAsURIs = []string{requestingAcct.URI}
|
|
|
|
if err := suite.state.DB.UpdateAccount(ctx, targetAcct, "also_known_as_uris"); err != nil {
|
|
|
|
suite.FailNow(err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove existing follow from zork to admin account.
|
|
|
|
if err := suite.state.DB.DeleteFollowByID(
|
|
|
|
ctx,
|
|
|
|
suite.testFollows["local_account_1_admin_account"].ID,
|
|
|
|
); err != nil {
|
|
|
|
suite.FailNow(err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Have Zork follow foss_satan instead.
|
|
|
|
if err := suite.state.DB.PutFollow(ctx, >smodel.Follow{
|
|
|
|
ID: "01HRA0XZYFZC5MNWTKEBR58SSE",
|
|
|
|
URI: "http://localhost:8080/users/the_mighty_zork/follows/01HRA0XZYFZC5MNWTKEBR58SSE",
|
|
|
|
AccountID: receivingAcct.ID,
|
|
|
|
TargetAccountID: requestingAcct.ID,
|
|
|
|
}); err != nil {
|
|
|
|
suite.FailNow(err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Process the Move.
|
|
|
|
err := suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
|
|
|
|
APObjectType: ap.ObjectProfile,
|
|
|
|
APActivityType: ap.ActivityMove,
|
|
|
|
GTSModel: >smodel.Move{
|
|
|
|
OriginURI: requestingAcct.URI,
|
|
|
|
Origin: testrig.URLMustParse(requestingAcct.URI),
|
|
|
|
TargetURI: targetAcct.URI,
|
|
|
|
Target: testrig.URLMustParse(targetAcct.URI),
|
|
|
|
URI: "https://fossbros-anonymous.io/users/foss_satan/moves/01HRA064871MR8HGVSAFJ333GM",
|
|
|
|
},
|
|
|
|
ReceivingAccount: receivingAcct,
|
|
|
|
RequestingAccount: requestingAcct,
|
|
|
|
})
|
|
|
|
suite.NoError(err)
|
|
|
|
|
|
|
|
// Zork should now be following admin account.
|
|
|
|
follows, err := suite.state.DB.IsFollowing(ctx, receivingAcct.ID, targetAcct.ID)
|
|
|
|
if err != nil {
|
|
|
|
suite.FailNow(err.Error())
|
|
|
|
}
|
|
|
|
suite.True(follows)
|
|
|
|
|
|
|
|
// Move should be in the DB.
|
|
|
|
move, err := suite.state.DB.GetMoveByURI(ctx, "https://fossbros-anonymous.io/users/foss_satan/moves/01HRA064871MR8HGVSAFJ333GM")
|
|
|
|
if err != nil {
|
|
|
|
suite.FailNow(err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Move should be marked as completed.
|
|
|
|
suite.WithinDuration(time.Now(), move.SucceededAt, 1*time.Minute)
|
|
|
|
}
|
|
|
|
|
2021-09-04 11:29:56 +00:00
|
|
|
func TestFromFederatorTestSuite(t *testing.T) {
|
2023-08-09 17:14:33 +00:00
|
|
|
suite.Run(t, &FromFediAPITestSuite{})
|
2021-09-04 11:29:56 +00:00
|
|
|
}
|