diff options
Diffstat (limited to 'testrig/db.go')
-rw-r--r-- | testrig/db.go | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/testrig/db.go b/testrig/db.go index d95c8f941..c169669d7 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -39,6 +39,8 @@ var testModels = []interface{}{ >smodel.EmailDomainBlock{}, >smodel.Follow{}, >smodel.FollowRequest{}, + >smodel.List{}, + >smodel.ListEntry{}, >smodel.MediaAttachment{}, >smodel.Mention{}, >smodel.Status{}, @@ -248,6 +250,18 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) { } } + for _, v := range NewTestLists() { + if err := db.Put(ctx, v); err != nil { + log.Panic(nil, err) + } + } + + for _, v := range NewTestListEntries() { + if err := db.Put(ctx, v); err != nil { + log.Panic(nil, err) + } + } + for _, v := range NewTestNotifications() { if err := db.Put(ctx, v); err != nil { log.Panic(nil, err) |