summaryrefslogtreecommitdiff
path: root/testrig/db.go
diff options
context:
space:
mode:
Diffstat (limited to 'testrig/db.go')
-rw-r--r--testrig/db.go7
1 files changed, 7 insertions, 0 deletions
diff --git a/testrig/db.go b/testrig/db.go
index 88237d7d0..2e974f100 100644
--- a/testrig/db.go
+++ b/testrig/db.go
@@ -55,6 +55,7 @@ var testModels = []interface{}{
&gtsmodel.RouterSession{},
&gtsmodel.Token{},
&gtsmodel.Client{},
+ &gtsmodel.Tombstone{},
}
// NewTestDB returns a new initialized, empty database for testing.
@@ -240,6 +241,12 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) {
}
}
+ for _, v := range NewTestTombstones() {
+ if err := db.Put(ctx, v); err != nil {
+ log.Panic(err)
+ }
+ }
+
if err := db.CreateInstanceAccount(ctx); err != nil {
log.Panic(err)
}