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 88237d7d0..2e974f100 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -55,6 +55,7 @@ var testModels = []interface{}{ >smodel.RouterSession{}, >smodel.Token{}, >smodel.Client{}, + >smodel.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) } |