diff options
| author | Ilya Zhuravlev <whatever@xyz.is> | 2017-11-02 23:11:28 +0300 |
|---|---|---|
| committer | Ilya Zhuravlev <whatever@xyz.is> | 2017-11-02 23:19:57 +0300 |
| commit | 8a4a0ddea60260014ff09eb0a72b9e30fe43c9e8 (patch) | |
| tree | f451e9e869ea99f037ec5a2742c82e2c43d29895 | |
| parent | Fix appservices being backlogged and not receiving new events due to a bug in... (diff) | |
| download | synapse-8a4a0ddea60260014ff09eb0a72b9e30fe43c9e8.tar.xz | |
Fix appservice tests to account for new behavior of notify_interested_services
| -rw-r--r-- | tests/handlers/test_appservice.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/handlers/test_appservice.py b/tests/handlers/test_appservice.py index 7fe88172c0..a667fb6f0e 100644 --- a/tests/handlers/test_appservice.py +++ b/tests/handlers/test_appservice.py @@ -53,7 +53,10 @@ class AppServiceHandlerTestCase(unittest.TestCase): type="m.room.message", 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() yield self.handler.notify_interested_services(0) 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.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) self.mock_as_api.query_user.assert_called_once_with( services[0], user_id @@ -98,7 +104,10 @@ class AppServiceHandlerTestCase(unittest.TestCase): ) self.mock_as_api.push = 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) self.assertFalse( self.mock_as_api.query_user.called, |