Test for MSC4115 support

Author: richvdh

tests/msc4115/main_test.go

@@ -0,0 +1,11 @@
+package tests
+
+import (
+	"testing"
+
+	"github.com/matrix-org/complement"
+)
+
+func TestMain(m *testing.M) {
+	complement.TestMain(m, "msc4115")
+}

tests/msc4115/msc4115_test.go

@@ -0,0 +1,61 @@
+package tests
+
+import (
+	"github.com/matrix-org/complement/b"
+	"github.com/matrix-org/complement/client"
+	"github.com/matrix-org/complement/runtime"
+	"testing"
+
+	"github.com/matrix-org/complement"
+	"github.com/matrix-org/complement/helpers"
+)
+
+// MSC4115: membership information on events
+//
+// Alice sends one message before bob joins, then one after. Bob reads both messages, and checks the membership state
+// on each.
+func TestMSC4115(t *testing.T) {
+	runtime.SkipIf(t, runtime.Dendrite) // not yet implemented
+
+	deployment := complement.Deploy(t, 1)
+	defer deployment.Destroy(t)
+
+	alice := deployment.Register(t, "hs1", helpers.RegistrationOpts{})
+	bob := deployment.Register(t, "hs1", helpers.RegistrationOpts{})
+
+	roomID := alice.MustCreateRoom(t, map[string]interface{}{"preset": "public_chat"})
+	preJoinEventId := alice.SendEventSynced(t, roomID, b.Event{Type: "m.room.message",
+		Content: map[string]interface{}{
+			"msgtype": "m.text",
+			"body":    "prejoin",
+		}})
+	bob.MustJoinRoom(t, roomID, []string{"hs1"})
+	postJoinEventId := alice.SendEventSynced(t, roomID, b.Event{Type: "m.room.message",
+		Content: map[string]interface{}{
+			"msgtype": "m.text",
+			"body":    "postjoin",
+		}})
+
+	// Now Bob syncs, to get the messages
+	syncResult, _ := bob.MustSync(t, client.SyncReq{})
+	roomSyncResult := syncResult.Get("rooms.join." + client.GjsonEscape(roomID))
+
+	// ... and we check the membership value for each event
+	checkEventMembership := func(eventID string, expectedMembership string) {
+		for _, ev := range roomSyncResult.Get("timeline.events").Array() {
+			if ev.Get("event_id").Str == eventID {
+				// found the right event; check the MSC4115 value
+				t.Logf("Event %s from timeline: %s", eventID, ev)
+				membership := ev.Get("unsigned." + client.GjsonEscape("io.element.msc4115.membership")).Str
+				if membership != expectedMembership {
+					t.Errorf("Incorrect membership for event; got %s, want %s", membership, expectedMembership)
+				}
+				return
+			}
+		}
+		t.Errorf("Did not find event %s in timeline", eventID)
+	}
+
+	checkEventMembership(preJoinEventId, "leave")
+	checkEventMembership(postJoinEventId, "join")
+}