Don't mock out presence_handler's internal start/stop methods in presencelike unit test; it's rude
parent
fc778e2bce
commit
eef58a299e
|
@ -63,6 +63,7 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase):
|
||||||
db_pool=None,
|
db_pool=None,
|
||||||
datastore=Mock(spec=[
|
datastore=Mock(spec=[
|
||||||
"set_presence_state",
|
"set_presence_state",
|
||||||
|
"is_presence_visible",
|
||||||
|
|
||||||
"set_profile_displayname",
|
"set_profile_displayname",
|
||||||
]),
|
]),
|
||||||
|
@ -83,6 +84,10 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase):
|
||||||
return defer.succeed("Frank")
|
return defer.succeed("Frank")
|
||||||
self.datastore.get_profile_displayname = get_profile_displayname
|
self.datastore.get_profile_displayname = get_profile_displayname
|
||||||
|
|
||||||
|
def is_presence_visible(*args, **kwargs):
|
||||||
|
return defer.succeed(False)
|
||||||
|
self.datastore.is_presence_visible = is_presence_visible
|
||||||
|
|
||||||
def get_profile_avatar_url(user_localpart):
|
def get_profile_avatar_url(user_localpart):
|
||||||
return defer.succeed("http://foo")
|
return defer.succeed("http://foo")
|
||||||
self.datastore.get_profile_avatar_url = get_profile_avatar_url
|
self.datastore.get_profile_avatar_url = get_profile_avatar_url
|
||||||
|
@ -96,14 +101,9 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase):
|
||||||
|
|
||||||
self.handlers = hs.get_handlers()
|
self.handlers = hs.get_handlers()
|
||||||
|
|
||||||
self.mock_start = Mock()
|
|
||||||
self.mock_stop = Mock()
|
|
||||||
|
|
||||||
self.mock_update_client = Mock()
|
self.mock_update_client = Mock()
|
||||||
self.mock_update_client.return_value = defer.succeed(None)
|
self.mock_update_client.return_value = defer.succeed(None)
|
||||||
|
|
||||||
self.handlers.presence_handler.start_polling_presence = self.mock_start
|
|
||||||
self.handlers.presence_handler.stop_polling_presence = self.mock_stop
|
|
||||||
self.handlers.presence_handler.push_update_to_clients = (
|
self.handlers.presence_handler.push_update_to_clients = (
|
||||||
self.mock_update_client)
|
self.mock_update_client)
|
||||||
|
|
||||||
|
@ -132,10 +132,6 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase):
|
||||||
|
|
||||||
mocked_set.assert_called_with("apple",
|
mocked_set.assert_called_with("apple",
|
||||||
{"state": UNAVAILABLE, "status_msg": "Away"})
|
{"state": UNAVAILABLE, "status_msg": "Away"})
|
||||||
self.mock_start.assert_called_with(self.u_apple,
|
|
||||||
state={"state": UNAVAILABLE, "status_msg": "Away",
|
|
||||||
"displayname": "Frank",
|
|
||||||
"avatar_url": "http://foo"})
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_push_local(self):
|
def test_push_local(self):
|
||||||
|
|
Loading…
Reference in New Issue