Skip to content
This repository was archived by the owner on Apr 26, 2024. It is now read-only.

Commit aaa9679

Browse files
committed
Simplify case
1 parent cfa5e57 commit aaa9679

File tree

1 file changed

+17
-41
lines changed

1 file changed

+17
-41
lines changed

tests/handlers/test_federation_event.py

Lines changed: 17 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -966,7 +966,7 @@ async def get_room_state(
966966
event_injection.inject_member_event(self.hs, room_id, user_alice, "join")
967967
)
968968

969-
event_before0 = self.get_success(
969+
event_before = self.get_success(
970970
inject_event(
971971
self.hs,
972972
room_id=room_id,
@@ -975,19 +975,9 @@ async def get_room_state(
975975
content={"body": "eventBefore0", "msgtype": "m.text"},
976976
)
977977
)
978-
_add_to_known_event_list(event_before0)
979-
event_before1 = self.get_success(
980-
inject_event(
981-
self.hs,
982-
room_id=room_id,
983-
sender=user_alice,
984-
type=EventTypes.Message,
985-
content={"body": "eventBefore1", "msgtype": "m.text"},
986-
)
987-
)
988-
_add_to_known_event_list(event_before1)
978+
_add_to_known_event_list(event_before)
989979

990-
event_after0 = self.get_success(
980+
event_after = self.get_success(
991981
inject_event(
992982
self.hs,
993983
room_id=room_id,
@@ -996,20 +986,10 @@ async def get_room_state(
996986
content={"body": "eventAfter0", "msgtype": "m.text"},
997987
)
998988
)
999-
_add_to_known_event_list(event_after0)
1000-
event_after1 = self.get_success(
1001-
inject_event(
1002-
self.hs,
1003-
room_id=room_id,
1004-
sender=user_alice,
1005-
type=EventTypes.Message,
1006-
content={"body": "eventAfter1", "msgtype": "m.text"},
1007-
)
1008-
)
1009-
_add_to_known_event_list(event_after1)
989+
_add_to_known_event_list(event_after)
1010990

1011991
state_map = self.get_success(
1012-
state_storage_controller.get_state_for_event(event_before1.event_id)
992+
state_storage_controller.get_state_for_event(event_before.event_id)
1013993
)
1014994

1015995
room_create_event = state_map.get((EventTypes.Create, ""))
@@ -1032,7 +1012,7 @@ async def get_room_state(
10321012
state_event.event_id for state_event in historical_state_events
10331013
]
10341014

1035-
inherited_depth = event_after0.depth
1015+
inherited_depth = event_after.depth
10361016
batch_id = random_string(8)
10371017
next_batch_id = random_string(8)
10381018
insertion_event, _ = self.get_success(
@@ -1092,7 +1072,7 @@ async def get_room_state(
10921072
EventContentFields.MSC2716_NEXT_BATCH_ID: batch_id,
10931073
EventContentFields.MSC2716_HISTORICAL: True,
10941074
},
1095-
prev_event_ids=[event_before1.event_id],
1075+
prev_event_ids=[event_before.event_id],
10961076
auth_event_ids=historical_auth_event_ids,
10971077
state_event_ids=historical_state_event_ids,
10981078
depth=inherited_depth,
@@ -1105,25 +1085,25 @@ async def get_room_state(
11051085
# # Beginning of room (oldest messages)
11061086
# # *list(state_map.values()),
11071087
# room_create_event,
1108-
# as_membership_event,
11091088
# pl_event,
1089+
# as_membership_event,
11101090
# state_map.get((EventTypes.JoinRules, "")),
11111091
# state_map.get((EventTypes.RoomHistoryVisibility, "")),
11121092
# alice_membership_event,
1113-
# event_before0,
1114-
# event_before1,
1093+
# event_before,
11151094
# # HISTORICAL MESSAGE END
11161095
# insertion_event,
11171096
# historical_message_event,
11181097
# batch_event,
11191098
# base_insertion_event,
11201099
# # HISTORICAL MESSAGE START
1121-
# event_after0,
1122-
# event_after1,
1100+
# event_after,
11231101
# # Latest in the room (newest messages)
11241102
# ]
11251103

1126-
# The random pattern that may make it be expected
1104+
# The order that we get after passing reverse chronological events in
1105+
# that mostly passes. Only the insertion event is rejected but the
1106+
# historical messages appear /messages scrollback.
11271107
pulled_events: List[EventBase] = [
11281108
# Beginning of room (oldest messages)
11291109
# *list(state_map.values()),
@@ -1133,14 +1113,12 @@ async def get_room_state(
11331113
state_map.get((EventTypes.JoinRules, "")),
11341114
state_map.get((EventTypes.RoomHistoryVisibility, "")),
11351115
alice_membership_event,
1136-
event_before0,
1137-
event_before1,
1138-
event_after0,
1116+
event_before,
1117+
event_after,
11391118
base_insertion_event,
11401119
batch_event,
11411120
historical_message_event,
11421121
insertion_event,
1143-
event_after1,
11441122
# Latest in the room (newest messages)
11451123
]
11461124

@@ -1196,16 +1174,14 @@ async def get_room_state(
11961174
state_map.get((EventTypes.JoinRules, "")),
11971175
state_map.get((EventTypes.RoomHistoryVisibility, "")),
11981176
alice_membership_event,
1199-
event_before0,
1200-
event_before1,
1177+
event_before,
12011178
# HISTORICAL MESSAGE END
12021179
insertion_event,
12031180
historical_message_event,
12041181
batch_event,
12051182
base_insertion_event,
12061183
# HISTORICAL MESSAGE START
1207-
event_after0,
1208-
event_after1,
1184+
event_after,
12091185
# Latest in the room (newest messages)
12101186
]
12111187

0 commit comments

Comments
 (0)