projects
/
osm
/
MON.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Bug 1842: LTS Support
[osm/MON.git]
/
osm_mon
/
tests
/
unit
/
core
/
test_message_bus_client.py
diff --git
a/osm_mon/tests/unit/core/test_message_bus_client.py
b/osm_mon/tests/unit/core/test_message_bus_client.py
index
292fbe3
..
126eb9f
100644
(file)
--- a/
osm_mon/tests/unit/core/test_message_bus_client.py
+++ b/
osm_mon/tests/unit/core/test_message_bus_client.py
@@
-31,43
+31,42
@@
from osm_mon.core.config import Config
class TestMessageBusClient(TestCase):
class TestMessageBusClient(TestCase):
-
def setUp(self):
self.config = Config()
def setUp(self):
self.config = Config()
- self.config.set(
'message', 'driver', 'kafka'
)
+ self.config.set(
"message", "driver", "kafka"
)
self.loop = asyncio.new_event_loop()
self.loop = asyncio.new_event_loop()
- @mock.patch.object(MsgKafka,
'aioread'
)
+ @mock.patch.object(MsgKafka,
"aioread"
)
def test_aioread(self, aioread):
async def mock_callback():
pass
future = asyncio.Future(loop=self.loop)
def test_aioread(self, aioread):
async def mock_callback():
pass
future = asyncio.Future(loop=self.loop)
- future.set_result(
'mock'
)
+ future.set_result(
"mock"
)
aioread.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
aioread.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
- topic =
'test_topic'
+ topic =
"test_topic"
self.loop.run_until_complete(msg_bus.aioread([topic], mock_callback))
self.loop.run_until_complete(msg_bus.aioread([topic], mock_callback))
- aioread.assert_called_with([
'test_topic'
], self.loop, aiocallback=mock_callback)
+ aioread.assert_called_with([
"test_topic"
], self.loop, aiocallback=mock_callback)
- @mock.patch.object(MsgKafka,
'aiowrite'
)
+ @mock.patch.object(MsgKafka,
"aiowrite"
)
def test_aiowrite(self, aiowrite):
future = asyncio.Future(loop=self.loop)
def test_aiowrite(self, aiowrite):
future = asyncio.Future(loop=self.loop)
- future.set_result(
'mock'
)
+ future.set_result(
"mock"
)
aiowrite.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
aiowrite.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
- topic =
'test_topic'
- key =
'test_key'
- msg = {
'test': 'test_msg'
}
+ topic =
"test_topic"
+ key =
"test_key"
+ msg = {
"test": "test_msg"
}
self.loop.run_until_complete(msg_bus.aiowrite(topic, key, msg))
aiowrite.assert_called_with(topic, key, msg, self.loop)
self.loop.run_until_complete(msg_bus.aiowrite(topic, key, msg))
aiowrite.assert_called_with(topic, key, msg, self.loop)
- @mock.patch.object(MsgKafka,
'aioread'
)
+ @mock.patch.object(MsgKafka,
"aioread"
)
def test_aioread_once(self, aioread):
future = asyncio.Future(loop=self.loop)
def test_aioread_once(self, aioread):
future = asyncio.Future(loop=self.loop)
- future.set_result(
'mock'
)
+ future.set_result(
"mock"
)
aioread.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
aioread.return_value = future
msg_bus = MessageBusClient(self.config, loop=self.loop)
- topic =
'test_topic'
+ topic =
"test_topic"
self.loop.run_until_complete(msg_bus.aioread_once(topic))
self.loop.run_until_complete(msg_bus.aioread_once(topic))
- aioread.assert_called_with(
'test_topic'
, self.loop)
+ aioread.assert_called_with(
"test_topic"
, self.loop)