1 files changed, 15 insertions, 1 deletions
diff --git a/testrig/db.go b/testrig/db.go
index 57e94a4bf..771345fe1 100644
--- a/testrig/db.go
+++ b/testrig/db.go
@@ -49,8 +49,10 @@ var testModels = []interface{}{
 	&gtsmodel.StatusToTag{},
 	&gtsmodel.StatusFave{},
 	&gtsmodel.StatusBookmark{},
-	&gtsmodel.StatusMute{},
 	&gtsmodel.Tag{},
+	&gtsmodel.Thread{},
+	&gtsmodel.ThreadMute{},
+	&gtsmodel.ThreadToStatus{},
 	&gtsmodel.User{},
 	&gtsmodel.Emoji{},
 	&gtsmodel.Instance{},
@@ -301,6 +303,18 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) {
 		}
 	}
 
+	for _, v := range NewTestThreads() {
+		if err := db.Put(ctx, v); err != nil {
+			log.Panic(nil, err)
+		}
+	}
+
+	for _, v := range NewTestThreadToStatus() {
+		if err := db.Put(ctx, v); err != nil {
+			log.Panic(nil, err)
+		}
+	}
+
 	if err := db.CreateInstanceAccount(ctx); err != nil {
 		log.Panic(nil, err)
 	}