Create events using helper

This commit is contained in:
Eric Eastwood 2024-06-26 18:52:24 -05:00
parent 2bf39231ed
commit ec2d8dc1e3

View file

@ -37,6 +37,7 @@ from synapse.storage.databases.main.stream import CurrentStateDeltaMembership
from synapse.types import JsonDict, PersistedEventPosition, RoomStreamToken from synapse.types import JsonDict, PersistedEventPosition, RoomStreamToken
from synapse.util import Clock from synapse.util import Clock
from tests.test_utils.event_injection import create_event
from tests.unittest import HomeserverTestCase from tests.unittest import HomeserverTestCase
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -809,56 +810,35 @@ class GetCurrentStateDeltaMembershipChangesForUserTestCase(HomeserverTestCase):
# Persist the user1, user3, and user4 join events in the same batch so they all # Persist the user1, user3, and user4 join events in the same batch so they all
# end up in the `current_state_delta_stream` table with the same # end up in the `current_state_delta_stream` table with the same
# stream_ordering. # stream_ordering.
join_event1 = make_event_from_dict( join_event1, join_event_context1 = self.get_success(
{ create_event(
"sender": user1_id, self.hs,
"type": EventTypes.Member, sender=user1_id,
"state_key": user1_id, type=EventTypes.Member,
"content": {"membership": "join"}, state_key=user1_id,
"room_id": room_id1, content={"membership": "join"},
"depth": 0, room_id=room_id1,
"origin_server_ts": 0, )
"prev_events": [],
"auth_events": [],
},
room_version=RoomVersions.V10,
) )
join_event_context1 = self.get_success( join_event3, join_event_context3 = self.get_success(
self.state_handler.compute_event_context(join_event1) create_event(
self.hs,
sender=user3_id,
type=EventTypes.Member,
state_key=user3_id,
content={"membership": "join"},
room_id=room_id1,
)
) )
join_event3 = make_event_from_dict( join_event4, join_event_context4 = self.get_success(
{ create_event(
"sender": user3_id, self.hs,
"type": EventTypes.Member, sender=user4_id,
"state_key": user3_id, type=EventTypes.Member,
"content": {"membership": "join"}, state_key=user4_id,
"room_id": room_id1, content={"membership": "join"},
"depth": 1, room_id=room_id1,
"origin_server_ts": 1, )
"prev_events": [],
"auth_events": [],
},
room_version=RoomVersions.V10,
)
join_event_context3 = self.get_success(
self.state_handler.compute_event_context(join_event3)
)
join_event4 = make_event_from_dict(
{
"sender": user4_id,
"type": EventTypes.Member,
"state_key": user4_id,
"content": {"membership": "join"},
"room_id": room_id1,
"depth": 2,
"origin_server_ts": 2,
"prev_events": [],
"auth_events": [],
},
room_version=RoomVersions.V10,
)
join_event_context4 = self.get_success(
self.state_handler.compute_event_context(join_event4)
) )
self.get_success( self.get_success(
self.persistence.persist_events( self.persistence.persist_events(