diff options
Diffstat (limited to 'testrig')
| -rw-r--r-- | testrig/db.go | 7 | ||||
| -rw-r--r-- | testrig/testmodels.go | 12 |
2 files changed, 19 insertions, 0 deletions
diff --git a/testrig/db.go b/testrig/db.go index c169669d7..eb8a23f42 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -60,6 +60,7 @@ var testModels = []interface{}{ >smodel.EmojiCategory{}, >smodel.Tombstone{}, >smodel.Report{}, + >smodel.AccountNote{}, } // NewTestDB returns a new initialized, empty database for testing. @@ -280,6 +281,12 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) { } } + for _, v := range NewTestAccountNotes() { + if err := db.Put(ctx, v); err != nil { + log.Panic(nil, err) + } + } + if err := db.CreateInstanceAccount(ctx); err != nil { log.Panic(nil, err) } diff --git a/testrig/testmodels.go b/testrig/testmodels.go index 837869c83..b22b6089c 100644 --- a/testrig/testmodels.go +++ b/testrig/testmodels.go @@ -1893,6 +1893,18 @@ func NewTestFaves() map[string]*gtsmodel.StatusFave { } } +// NewTestAccountNotes returns some account notes for use in testing. +func NewTestAccountNotes() map[string]*gtsmodel.AccountNote { + return map[string]*gtsmodel.AccountNote{ + "local_account_2_note_on_1": { + ID: "01H53TM628GNC4ZDNRGQGPK8S0", + AccountID: "01F8MH5NBDF2MV7CTC4Q5128HF", + TargetAccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", + Comment: "extremely average poster", + }, + } +} + // NewTestNotifications returns some notifications for use in testing. func NewTestNotifications() map[string]*gtsmodel.Notification { return map[string]*gtsmodel.Notification{ |