Fixup tests

This commit is contained in:
Erik Johnston 2019-12-05 15:53:10 +00:00
parent 1a0997bbd5
commit 4ca3ef10b9
3 changed files with 15 additions and 2 deletions

View file

@ -52,6 +52,14 @@ class MockHandlerProfileTestCase(unittest.TestCase):
]
)
self.mock_handler.get_displayname.return_value = defer.succeed(Mock())
self.mock_handler.set_displayname.return_value = defer.succeed(Mock())
self.mock_handler.get_avatar_url.return_value = defer.succeed(Mock())
self.mock_handler.set_avatar_url.return_value = defer.succeed(Mock())
self.mock_handler.check_profile_query_allowed.return_value = defer.succeed(
Mock()
)
hs = yield setup_test_homeserver(
self.addCleanup,
"test",
@ -63,7 +71,7 @@ class MockHandlerProfileTestCase(unittest.TestCase):
)
def _get_user_by_req(request=None, allow_guest=False):
return synapse.types.create_requester(myid)
return defer.succeed(synapse.types.create_requester(myid))
hs.get_auth().get_user_by_req = _get_user_by_req