From 5e2d4fdb19eb4fcd4c0bbfb3e2f29067a58c88c8 Mon Sep 17 00:00:00 2001
From: Vyr Cossont <VyrCossont@users.noreply.github.com>
Date: Thu, 6 Jun 2024 09:38:02 -0700
Subject: [feature] User muting (#2960)

* User muting

* Address review feedback

* Rename uniqueness constraint on user_mutes to match convention

* Remove unused account_id from where clause

* Add UserMute to NewTestDB

* Update test/envparsing.sh with new and fixed cache stuff

* Address tobi's review comments

* Make compiledUserMuteListEntry.expired consistent with UserMute.Expired

* Make sure mute_expires_at is serialized as an explicit null for indefinite mutes

---------

Co-authored-by: tobi <tobi.smethurst@protonmail.com>
---
 testrig/db.go | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'testrig/db.go')

diff --git a/testrig/db.go b/testrig/db.go
index faa7a910d..67a7e2439 100644
--- a/testrig/db.go
+++ b/testrig/db.go
@@ -56,6 +56,7 @@ var testModels = []interface{}{
 	&gtsmodel.ThreadMute{},
 	&gtsmodel.ThreadToStatus{},
 	&gtsmodel.User{},
+	&gtsmodel.UserMute{},
 	&gtsmodel.Emoji{},
 	&gtsmodel.Instance{},
 	&gtsmodel.Notification{},
@@ -338,6 +339,12 @@ func StandardDBSetup(db db.DB, accounts map[string]*gtsmodel.Account) {
 		}
 	}
 
+	for _, v := range NewTestUserMutes() {
+		if err := db.Put(ctx, v); err != nil {
+			log.Panic(nil, err)
+		}
+	}
+
 	if err := db.CreateInstanceAccount(ctx); err != nil {
 		log.Panic(nil, err)
 	}
-- 
cgit v1.2.3