summaryrefslogtreecommitdiff
path: root/testrig/db.go
diff options
context:
space:
mode:
Diffstat (limited to 'testrig/db.go')
-rw-r--r--testrig/db.go14
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{}{
&gtsmodel.EmailDomainBlock{},
&gtsmodel.Follow{},
&gtsmodel.FollowRequest{},
+ &gtsmodel.List{},
+ &gtsmodel.ListEntry{},
&gtsmodel.MediaAttachment{},
&gtsmodel.Mention{},
&gtsmodel.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)