projects
/
osm
/
POL.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Reformat POL to standardized format
[osm/POL.git]
/
osm_policy_module
/
tests
/
unit
/
common
/
test_message_bus_client.py
diff --git
a/osm_policy_module/tests/unit/common/test_message_bus_client.py
b/osm_policy_module/tests/unit/common/test_message_bus_client.py
index
0b97de7
..
81ca832
100644
(file)
--- a/
osm_policy_module/tests/unit/common/test_message_bus_client.py
+++ b/
osm_policy_module/tests/unit/common/test_message_bus_client.py
@@
-31,43
+31,42
@@
from osm_policy_module.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)