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

Commit

Permalink
Fix mocked functions assert_called_with
Browse files Browse the repository at this point in the history
  • Loading branch information
erikjohnston committed Oct 30, 2020
1 parent 7f13aa7 commit 9b9a7bd
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 6 deletions.
6 changes: 5 additions & 1 deletion tests/api/test_auth.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,11 @@ def test_cannot_use_regular_token_as_guest(self):
)
)
self.store.add_access_token_to_user.assert_called_with(
USER_ID, token, "DEVICE", None
user_id=USER_ID,
token=token,
device_id="DEVICE",
valid_until_ms=None,
puppets_user_id=None,
)

def get_user(tok):
Expand Down
11 changes: 6 additions & 5 deletions tests/module_api/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,12 +94,13 @@ def test_sending_events_into_room(self):
self.assertFalse(hasattr(event, "state_key"))
self.assertDictEqual(event.content, content)

expected_requester = create_requester(
user_id, authenticated_entity=self.hs.hostname
)

# Check that the event was sent
self.event_creation_handler.create_and_send_nonmember_event.assert_called_with(
create_requester(user_id),
event_dict,
ratelimit=False,
ignore_shadow_ban=True,
expected_requester, event_dict, ratelimit=False, ignore_shadow_ban=True,
)

# Create and send a state event
Expand Down Expand Up @@ -128,7 +129,7 @@ def test_sending_events_into_room(self):

# Check that the event was sent
self.event_creation_handler.create_and_send_nonmember_event.assert_called_with(
create_requester(user_id),
expected_requester,
{
"type": "m.room.power_levels",
"content": content,
Expand Down

0 comments on commit 9b9a7bd

Please sign in to comment.