paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-29 11:03:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

python: Reformat with Black, except quotes.

Browse Source 
Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg
2021-02-11 23:19:30 -08:00
committed by Tim Abbott
parent 5028c081cb
commit 11741543da
817 changed files with 44952 additions and 24860 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
zerver/tests/test_queue.py
View File

@@ -16,7 +16,12 @@ class TestTornadoQueueClient(ZulipTestCase):
connection = TornadoQueueClient()
connection.connection.channel.side_effect = ConnectionClosed('500', 'test')
connection._on_open(mock.MagicMock())
self.assertEqual(m.output, ['WARNING:zulip.queue:TornadoQueueClient couldn\'t open channel: connection already closed'])
self.assertEqual(
m.output,
[
'WARNING:zulip.queue:TornadoQueueClient couldn\'t open channel: connection already closed'
],
)
class TestQueueImplementation(ZulipTestCase):
@@ -73,19 +78,20 @@ class TestQueueImplementation(ZulipTestCase):
self.counter = 0
def throw_connection_error_once(self_obj: Any, *args: Any,
**kwargs: Any) -> None:
def throw_connection_error_once(self_obj: Any, *args: Any, **kwargs: Any) -> None:
self.counter += 1
if self.counter <= 1:
raise AMQPConnectionError("test")
actual_publish(*args, **kwargs)
with mock.patch("zerver.lib.queue.SimpleQueueClient.publish",
throw_connection_error_once), self.assertLogs('zulip.queue', level='WARN') as warn_logs:
with mock.patch(
"zerver.lib.queue.SimpleQueueClient.publish", throw_connection_error_once
), self.assertLogs('zulip.queue', level='WARN') as warn_logs:
queue_json_publish("test_suite", {"event": "my_event"})
self.assertEqual(warn_logs.output, [
'WARNING:zulip.queue:Failed to send to rabbitmq, trying to reconnect and send again'
])
self.assertEqual(
warn_logs.output,
['WARNING:zulip.queue:Failed to send to rabbitmq, trying to reconnect and send again'],
)
assert queue_client.channel
(_, _, message) = queue_client.channel.basic_get("test_suite")