Disable MSC4306 push rules in non-MSC4306 test

Author: reivilibre

client/client.go

@@ -319,6 +319,14 @@ func (c *CSAPI) SetPushRule(t ct.TestLike, scope string, kind string, ruleID str
 	return c.MustDo(t, "PUT", []string{"_matrix", "client", "v3", "pushrules", scope, kind, ruleID}, WithJSONBody(t, body), WithQueries(queryParams))
 }
 
+// MustDisablePushRule disables a push rule on the user.
+// Fails the test if response is non-2xx.
+func (c *CSAPI) MustDisablePushRule(t ct.TestLike, scope string, kind string, ruleID string) {
+	c.MustDo(t, "PUT", []string{"_matrix", "client", "v3", "pushrules", scope, kind, ruleID, "enabled"}, WithJSONBody(t, map[string]interface{}{
+		"enabled": false,
+	}))
+}
+
 // Unsafe_SendEventUnsynced sends `e` into the room. This function is UNSAFE as it does not wait
 // for the event to be fully processed. This can cause flakey tests. Prefer `SendEventSynced`.
 // Returns the event ID of the sent event.

tests/csapi/thread_notifications_test.go

@@ -51,6 +51,21 @@ func syncHasThreadedReadReceipt(roomID, userID, eventID, threadID string) client
 	})
 }
 
+// Disables push rules that are introduced in MSC4306 (if present),
+// because they interfere with the normal semantics of notifications in threads.
+func disableMsc4306PushRules(t *testing.T, user *client.CSAPI) {
+	rules := []string{".io.element.msc4306.rule.subscribed_thread", ".io.element.msc4306.rule.unsubscribed_thread"}
+	for _, rule := range rules {
+		res := user.GetPushRule(t, "global", "postcontent", rule)
+		if res.StatusCode == 404 {
+			// No push rule to disable
+			continue
+		}
+
+		user.MustDisablePushRule(t, "global", "postcontent", rule)
+	}
+}
+
 // Test behavior of threaded receipts and notifications.
 //
 // 1. Send a series of messages, some of which are in threads.
@@ -79,7 +94,9 @@ func TestThreadedReceipts(t *testing.T) {
 
 	// Create a room with alice and bob.
 	alice := deployment.Register(t, "hs1", helpers.RegistrationOpts{})
+	disableMsc4306PushRules(t, alice)
 	bob := deployment.Register(t, "hs1", helpers.RegistrationOpts{})
+	disableMsc4306PushRules(t, bob)
 
 	roomID := alice.MustCreateRoom(t, map[string]interface{}{"preset": "public_chat"})
 	bob.MustJoinRoom(t, roomID, nil)
@@ -312,7 +329,9 @@ func TestThreadReceiptsInSyncMSC4102(t *testing.T) {
 
 	// Create a room with alice and bob.
 	alice := deployment.Register(t, "hs1", helpers.RegistrationOpts{})
+	disableMsc4306PushRules(t, alice)
 	bob := deployment.Register(t, "hs2", helpers.RegistrationOpts{})
+	disableMsc4306PushRules(t, bob)
 	roomID := alice.MustCreateRoom(t, map[string]interface{}{"preset": "public_chat"})
 	bob.MustJoinRoom(t, roomID, []spec.ServerName{
 		deployment.GetFullyQualifiedHomeserverName(t, "hs1"),