diff options
Diffstat (limited to 'testrig')
| -rw-r--r-- | testrig/db.go | 7 | ||||
| -rw-r--r-- | testrig/testmodels.go | 19 |
2 files changed, 26 insertions, 0 deletions
diff --git a/testrig/db.go b/testrig/db.go index eb8a23f42..4d8dfefa5 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -41,6 +41,7 @@ var testModels = []interface{}{ >smodel.FollowRequest{}, >smodel.List{}, >smodel.ListEntry{}, + >smodel.Marker{}, >smodel.MediaAttachment{}, >smodel.Mention{}, >smodel.Status{}, @@ -287,6 +288,12 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) { } } + for _, v := range NewTestMarkers() { + if err := db.Put(ctx, v); err != nil { + log.Panic(nil, err) + } + } + if err := db.CreateInstanceAccount(ctx); err != nil { log.Panic(nil, err) } diff --git a/testrig/testmodels.go b/testrig/testmodels.go index b22b6089c..c7a293455 100644 --- a/testrig/testmodels.go +++ b/testrig/testmodels.go @@ -2007,6 +2007,25 @@ func NewTestListEntries() map[string]*gtsmodel.ListEntry { } } +func NewTestMarkers() map[string]*gtsmodel.Marker { + return map[string]*gtsmodel.Marker{ + "local_account_1_home_marker": { + AccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", + Name: gtsmodel.MarkerNameHome, + UpdatedAt: TimeMustParse("2022-05-14T13:21:09+02:00"), + Version: 0, + LastReadID: "01F8MH82FYRXD2RC6108DAJ5HB", + }, + "local_account_1_notification_marker": { + AccountID: "01F8MH1H7YV1Z7D2C8K2730QBF", + Name: gtsmodel.MarkerNameNotifications, + UpdatedAt: TimeMustParse("2022-05-14T13:21:09+02:00"), + Version: 4, + LastReadID: "01F8Q0ANPTWW10DAKTX7BRPBJP", + }, + } +} + func NewTestBlocks() map[string]*gtsmodel.Block { return map[string]*gtsmodel.Block{ "local_account_2_block_remote_account_1": { |