def test_get_message_and_routing_key_none_error(self):
# Test variables:
passed_message = None
LOGGER.info('Message: %s' % passed_message)
# Run code to be checked:
with self.assertRaises(ValueError):
received_key, received_message = rutils.get_routing_key_and_string_message_from_message_if_possible(
passed_message)
def test_get_message_and_routing_key_bogus(self):
# Test variables:
def foo():
print('yeah')
passed_message = foo
LOGGER.info('Message: %s' % passed_message)
# Run code to be checked:
with self.assertRaises(ValueError):
received_key, received_message = rutils.get_routing_key_and_string_message_from_message_if_possible(
passed_message)
def test_get_message_and_routing_key_string_singlequtoes_ok(self):
# Test variables:
passed_message = "{'bla':'foo', 'ROUTING_KEY':'roukey'}"
LOGGER.info('Message: %s' % passed_message)
# Run code to be checked:
received_key, received_message = rutils.get_routing_key_and_string_message_from_message_if_possible(
passed_message)
# Check result:
received_message = received_message.replace("'", '"')
expected_message = str(passed_message).replace("'", '"')
self.assertEqual(received_key, 'roukey',
'Wrong routing key: %s' % received_key)
self.assertEqual(
expected_message, received_message,
'Wrong message.\nExpected: %s\nReceived: %s' %
(expected_message, received_message))
def test_get_message_and_routing_key_json_ok(self):
# Test variables:
passed_message = {"bla": "foo", "ROUTING_KEY": "roukey"}
LOGGER.info('Message: %s' % passed_message)
# Run code to be checked:
received_key, received_message = rutils.get_routing_key_and_string_message_from_message_if_possible(
passed_message)
# Check result:
received_message = received_message.replace("'", '"')
expected_message = str(passed_message).replace("'", '"')
self.assertEquals(received_key, 'roukey',
'Wrong routing key: %s' % received_key)
self.assertEquals(
expected_message, received_message,
'Wrong message.\nExpected: %s\nReceived: %s' %
(expected_message, received_message))
def test_get_message_and_routing_key_string_no_key(self):
# Test variables:
passed_message = '{"bla":"foo", "no_key":"no_key"}'
LOGGER.info('Message: %s' % passed_message)
# Run code to be checked:
received_key, received_message = rutils.get_routing_key_and_string_message_from_message_if_possible(
passed_message)
# Check result:
received_message = received_message.replace("'", '"')
expected_message = str(passed_message).replace("'", '"')
expected_key = esgfpid.utils.RABBIT_DEFAULT_ROUTING_KEY
self.assertEqual(received_key, expected_key,
'Wrong routing key: %s' % received_key)
self.assertEqual(
expected_message, received_message,
'Wrong message.\nExpected: %s\nReceived: %s' %
(expected_message, received_message))
