def test_init_messaging_core(self): messaging = messaging_core.MessagingCore("", "", "") self.assertIsInstance(messaging.serializer, rpc.RequestContextSerializer) self.assertIsInstance(messaging.conductor_topic_handler, messaging_handler.MessagingHandler) self.assertIsInstance(messaging.status_topic_handler, messaging_handler.MessagingHandler)
def test_publish_status(self, m_handler_cls): m_handler = mock.Mock() m_handler_cls.return_value = m_handler payload = { "name": "value", } event = "MyEvent" messaging = messaging_core.MessagingCore("", "", "") messaging.publish_status(event, payload) m_handler.publish_event.assert_called_once_with(event, payload, None)
def test_messaging_build_topic_handler(self): messaging = messaging_core.MessagingCore( "pub_id", "test_topic", "does not matter") topic = messaging.build_topic_handler("test_topic") self.assertIsInstance(topic, messaging_handler.MessagingHandler) self.assertEqual(messaging.publisher_id, "pub_id") self.assertEqual(topic.publisher_id, "pub_id") self.assertEqual( messaging.conductor_topic_handler.topic_name, "test_topic") self.assertEqual(topic.topic_name, "test_topic")
def test_response(self, mock_call): event = "My event" context = {'request_id': 12} message = "My Message" messaging = messaging_core.MessagingCore("", "", "") messaging.response(event, context, message) expected_payload = { 'request_id': context['request_id'], 'msg': message } mock_call.assert_called_once_with(event, expected_payload)
def test_build_topic_handler(self): topic_name = "MyTopic" messaging = messaging_core.MessagingCore("", "", "") handler = messaging.build_topic_handler(topic_name) self.assertIsNotNone(handler)
def test_disconnect(self, m_handler): messaging = messaging_core.MessagingCore("", "", "") messaging.disconnect() self.assertEqual(m_handler.call_count, 2)