From a022a0c561f30e5f23e42b015f9e02e0fa6d3983 Mon Sep 17 00:00:00 2001 From: Neil Alexander Date: Fri, 22 Jul 2022 17:03:36 +0100 Subject: [PATCH] Fix unit test --- roomserver/storage/tables/membership_table_test.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/roomserver/storage/tables/membership_table_test.go b/roomserver/storage/tables/membership_table_test.go index 14e8ce50a..f789ef4ac 100644 --- a/roomserver/storage/tables/membership_table_test.go +++ b/roomserver/storage/tables/membership_table_test.go @@ -60,6 +60,9 @@ func TestMembershipTable(t *testing.T) { // This inserts a left user to the room err = tab.InsertMembership(ctx, nil, 1, stateKeyNID, true) assert.NoError(t, err) + // We must update the membership with a non-zero event NID or it will get filtered out in later queries + _, err = tab.UpdateMembership(ctx, nil, 1, stateKeyNID, userNIDs[0], tables.MembershipStateLeaveOrBan, 1, false) + assert.NoError(t, err) } // ... so this should be false