diff options
Diffstat (limited to 'testrig')
| -rw-r--r-- | testrig/db.go | 6 | ||||
| -rw-r--r-- | testrig/testmodels.go | 20 |
2 files changed, 26 insertions, 0 deletions
diff --git a/testrig/db.go b/testrig/db.go index 508f10b76..3e34344bc 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -261,6 +261,12 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) { } } + for _, v := range NewTestBookmarks() { + if err := db.Put(ctx, v); err != nil { + log.Panic(err) + } + } + if err := db.CreateInstanceAccount(ctx); err != nil { log.Panic(err) } diff --git a/testrig/testmodels.go b/testrig/testmodels.go index b34c1a1d9..cbaa855e8 100644 --- a/testrig/testmodels.go +++ b/testrig/testmodels.go @@ -2251,6 +2251,26 @@ func NewTestFediStatuses() map[string]vocab.ActivityStreamsNote { } } +// NewTestBookmarks returns a map of gts model bookmarks, keyed in the format [bookmarking_account]_[target_status] +func NewTestBookmarks() map[string]*gtsmodel.StatusBookmark { + return map[string]*gtsmodel.StatusBookmark{ + "local_account_1_admin_account_status_1": { + ID: "01F8MHD2QCZSZ6WQS2ATVPEYJ9", + CreatedAt: TimeMustParse("2022-05-14T13:21:09+02:00"), + AccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", // local account 1 + TargetAccountID: "01F8MH17FWEB39HZJ76B6VXSKF", // admin account + StatusID: "01F8MH75CBF9JFX4ZAD54N0W0R", // admin account status 1 + }, + "admin_account_local_account_1_status_1": { + ID: "01F8Q0486ANTDWKG02A7DS1Q24", + CreatedAt: TimeMustParse("2022-05-14T13:21:09+02:00"), + AccountID: "01F8MH17FWEB39HZJ76B6VXSKF", // admin account + TargetAccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", // local account 1 + StatusID: "01F8MHAMCHF6Y650WCRSCP4WMY", // local account status 1 + }, + } +} + // NewTestDereferenceRequests returns a map of incoming dereference requests, with their signatures. func NewTestDereferenceRequests(accounts map[string]*gtsmodel.Account) map[string]ActivityWithSignature { var sig, digest, date string |