diff options
Diffstat (limited to 'testrig/db.go')
-rw-r--r-- | testrig/db.go | 7 |
1 files changed, 7 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) } |