Fix appservice tests to account for new behavior of notify_interested_services
This commit is contained in:
parent
45fbe4ff67
commit
8a4a0ddea6
|
@ -53,7 +53,10 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
||||||
type="m.room.message",
|
type="m.room.message",
|
||||||
room_id="!foo:bar"
|
room_id="!foo:bar"
|
||||||
)
|
)
|
||||||
self.mock_store.get_new_events_for_appservice.return_value = (0, [event])
|
self.mock_store.get_new_events_for_appservice.side_effect = [
|
||||||
|
(0, [event]),
|
||||||
|
(0, [])
|
||||||
|
]
|
||||||
self.mock_as_api.push = Mock()
|
self.mock_as_api.push = Mock()
|
||||||
yield self.handler.notify_interested_services(0)
|
yield self.handler.notify_interested_services(0)
|
||||||
self.mock_scheduler.submit_event_for_as.assert_called_once_with(
|
self.mock_scheduler.submit_event_for_as.assert_called_once_with(
|
||||||
|
@ -75,7 +78,10 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
||||||
)
|
)
|
||||||
self.mock_as_api.push = Mock()
|
self.mock_as_api.push = Mock()
|
||||||
self.mock_as_api.query_user = Mock()
|
self.mock_as_api.query_user = Mock()
|
||||||
self.mock_store.get_new_events_for_appservice.return_value = (0, [event])
|
self.mock_store.get_new_events_for_appservice.side_effect = [
|
||||||
|
(0, [event]),
|
||||||
|
(0, [])
|
||||||
|
]
|
||||||
yield self.handler.notify_interested_services(0)
|
yield self.handler.notify_interested_services(0)
|
||||||
self.mock_as_api.query_user.assert_called_once_with(
|
self.mock_as_api.query_user.assert_called_once_with(
|
||||||
services[0], user_id
|
services[0], user_id
|
||||||
|
@ -98,7 +104,10 @@ class AppServiceHandlerTestCase(unittest.TestCase):
|
||||||
)
|
)
|
||||||
self.mock_as_api.push = Mock()
|
self.mock_as_api.push = Mock()
|
||||||
self.mock_as_api.query_user = Mock()
|
self.mock_as_api.query_user = Mock()
|
||||||
self.mock_store.get_new_events_for_appservice.return_value = (0, [event])
|
self.mock_store.get_new_events_for_appservice.side_effect = [
|
||||||
|
(0, [event]),
|
||||||
|
(0, [])
|
||||||
|
]
|
||||||
yield self.handler.notify_interested_services(0)
|
yield self.handler.notify_interested_services(0)
|
||||||
self.assertFalse(
|
self.assertFalse(
|
||||||
self.mock_as_api.query_user.called,
|
self.mock_as_api.query_user.called,
|
||||||
|
|
Loading…
Reference in New Issue