Matrix/synapse
Matrix/synapse
1
0
Fork 0
mirror of https://github.com/element-hq/synapse synced 2024-07-04 08:43:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Actually test get_current_state_delta_membership_changes_for_user(...) in remote join test

Browse source
This commit is contained in:
Eric Eastwood 2024-06-26 19:52:12 -05:00
parent 0b9a903ca1
commit 48d0acfbcd
1 changed files with 36 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
tests/storage/test_stream.py
View file

@ -1001,10 +1001,13 @@ class GetCurrentStateDeltaMembershipChangesForUserFederationTestCase(
def test_remote_join(self) -> None: def test_remote_join(self) -> None:
""" """
Test remote join where the first rows will just be the state when you joined Test remote join where the first rows in `current_state_delta_stream` will just
be the state when you joined the remote room.
""" """
user1_id = self.register_user("user1", "pass") user1_id = self.register_user("user1", "pass")
user1_tok = self.login(user1_id, "pass") _user1_tok = self.login(user1_id, "pass")
before_join_token = self.event_sources.get_current_token()
intially_unjoined_room_id = f"!example:{self.OTHER_SERVER_NAME}" intially_unjoined_room_id = f"!example:{self.OTHER_SERVER_NAME}"
@ -1128,30 +1131,40 @@ class GetCurrentStateDeltaMembershipChangesForUserFederationTestCase(
) )
) )
events_db_dump = self.get_success( after_join_token = self.event_sources.get_current_token()
self.store.db_pool.simple_select_list(
table="events", # Get the membership changes for the user at this point, the
keyvalues={}, # `current_state_delta_stream` table should look like:
retcols=[ #
"*", # | stream_id | room_id | type | state_key | event_id | prev_event_id | instance_name |
], # |-----------|------------------------------|-----------------|------------------------------|----------|---------------|----------------|
desc="debug dump events", # | 2 | '!example:other.example.com' | 'm.room.member' | '@user1:test' | '$xxx' | None | 'master' |
# | 2 | '!example:other.example.com' | 'm.room.create' | '' | '$xxx' | None | 'master' |
# | 2 | '!example:other.example.com' | 'm.room.member' | '@creator:other.example.com' | '$xxx' | None | 'master' |
membership_changes = self.get_success(
self.store.get_current_state_delta_membership_changes_for_user(
user1_id,
from_key=before_join_token.room_key,
to_key=after_join_token.room_key,
) )
) )
logger.info("events_db_dump: %s", events_db_dump) join_pos = self.get_success(
self.store.get_position_for_event(join_event.event_id)
current_state_delta_stream_db_dump = self.get_success(
self.store.db_pool.simple_select_list(
table="current_state_delta_stream",
keyvalues={},
retcols=[
"*",
],
desc="debug dump current_state_delta_stream",
)
) )
logger.info( # Let the whole diff show on failure
"current_state_delta_stream_db_dump: %s", current_state_delta_stream_db_dump self.maxDiff = None
self.assertEqual(
membership_changes,
[
CurrentStateDeltaMembership(
event_id=join_event.event_id,
event_pos=join_pos,
prev_event_id=None,
room_id=intially_unjoined_room_id,
membership="join",
sender=user1_id,
),
],
) )