def test_message_init(self):
"""Test %s MessageListener() initialisation."""
# Ensure non-Message() inputs are caught.
with self.assertRaises(TypeError):
MessageListener(dict)
with self.assertRaises(TypeError):
MessageListener(0)
# Ensure errors are propagated.
with self.assertRaises(Exception):
MessageListener(self.BadMessage)
# Create an instance of MessageListener() with defaults.
listener = MessageListener(self.Message)
self.assertEqual(listener.topics, None)
self.assertTrue(listener.is_open)
listener.close()
# Create an instance of MessageListener() with a specific topic.
listener = MessageListener(self.Message, topics=TOPIC)
self.assertEqual(listener.topics, TOPIC)
self.assertTrue(listener.is_open)
listener.close()
# Ensure non-string topics are caught.
with self.assertRaises(TypeError):
MessageListener(self.Message, topics=False)
def test_message_init(self):
"""Test %s MessageListener() initialisation."""
# Ensure non-Message() inputs are caught.
with self.assertRaises(TypeError):
MessageListener(dict)
with self.assertRaises(TypeError):
MessageListener(0)
# Ensure errors are propagated.
with self.assertRaises(Exception):
MessageListener(self.BadMessage)
# Create an instance of MessageListener() with defaults.
listener = MessageListener(self.Message)
self.assertEqual(listener.topics, None)
self.assertTrue(listener.is_open)
listener.close()
# Create an instance of MessageListener() with a specific topic.
listener = MessageListener(self.Message, topics=TOPIC)
self.assertEqual(listener.topics, TOPIC)
self.assertTrue(listener.is_open)
listener.close()
# Ensure non-string topics are caught.
with self.assertRaises(TypeError):
MessageListener(self.Message, topics=False)
def test_message_send_receive(self):
"""Test %s with MessageBroadcaster/Listener() objects."""
# NOTE: this test listens for multiple topics from many. Rather than
# sending raw data (a previous test), Message() objects are
# sent. This tests all the functionality of the
# MessageBroadcaster() and MessageListener() objects.
# Send multiple topics, receive SOME topics.
send_topics = ['topic A', 'topic B', 'topic C', 'topic D', 'topic E']
listen_topics = ['topic A', 'topic C', 'topic E']
# Create broadcaster.
broadcaster = MessageBroadcaster(self.Message)
# Catch messages with a specific topic.
topic_buffer = list()
listener_topic = MessageListener(self.Message, topics=listen_topics)
# Subscribe callback.
def callback(data): topic_buffer.append(data)
self.assertTrue(listener_topic.subscribe(callback))
self.assertTrue(listener_topic.is_subscribed(callback))
self.assertEqual(listener_topic.num_subscriptions(), 1)
# Catch all messages. This ensures the unit-test does not time out
# waiting for messages that are filtered out by topic.
message_buffer = list()
listener_message = MessageListener(self.Message)
# Ensure network objects are open.
self.assertTrue(broadcaster.is_open)
self.assertTrue(listener_topic.is_open)
self.assertTrue(listener_message.is_open)
# Publish messages with different topics.
messages = list()
for (i, topic) in enumerate(send_topics):
messages.append(self.Message())
messages[-1]['text'] = '%s: %1.8f' % (topic, time.time())
# Perform test.
message_buffer = self.publish(broadcaster,
listener_message,
messages[-1],
topic=topic,
received_buffer=message_buffer)
# Close connections.
broadcaster.close()
listener_topic.close()
listener_message.close()
self.assertFalse(broadcaster.is_open)
self.assertFalse(listener_topic.is_open)
self.assertFalse(listener_message.is_open)
# Ensure all topics were received.
self.assertEqual(len(topic_buffer), len(listen_topics))
for i, topic in enumerate(listen_topics):
self.assertEqual(topic_buffer[i]['payload'],
messages[send_topics.index(topic)])
def test_message_send_receive(self):
"""Test %s with MessageBroadcaster/Listener() objects."""
# NOTE: this test listens for multiple topics from many. Rather than
# sending raw data (a previous test), Message() objects are
# sent. This tests all the functionality of the
# MessageBroadcaster() and MessageListener() objects.
# Send multiple topics, receive SOME topics.
send_topics = ['topic A', 'topic B', 'topic C', 'topic D', 'topic E']
listen_topics = ['topic A', 'topic C', 'topic E']
# Create broadcaster.
broadcaster = MessageBroadcaster(self.Message)
# Catch messages with a specific topic.
topic_buffer = list()
listener_topic = MessageListener(self.Message, topics=listen_topics)
# Subscribe callback.
def callback(data):
topic_buffer.append(data)
self.assertTrue(listener_topic.subscribe(callback))
self.assertTrue(listener_topic.is_subscribed(callback))
self.assertEqual(listener_topic.num_subscriptions(), 1)
# Catch all messages. This ensures the unit-test does not time out
# waiting for messages that are filtered out by topic.
message_buffer = list()
listener_message = MessageListener(self.Message)
# Ensure network objects are open.
self.assertTrue(broadcaster.is_open)
self.assertTrue(listener_topic.is_open)
self.assertTrue(listener_message.is_open)
# Publish messages with different topics.
messages = list()
for (i, topic) in enumerate(send_topics):
messages.append(self.Message())
messages[-1]['text'] = '%s: %1.8f' % (topic, time.time())
# Perform test.
message_buffer = self.publish(broadcaster,
listener_message,
messages[-1],
topic=topic,
received_buffer=message_buffer)
# Close connections.
broadcaster.close()
listener_topic.close()
listener_message.close()
self.assertFalse(broadcaster.is_open)
self.assertFalse(listener_topic.is_open)
self.assertFalse(listener_message.is_open)
# Ensure all topics were received.
self.assertEqual(len(topic_buffer), len(listen_topics))
for i, topic in enumerate(listen_topics):
self.assertEqual(topic_buffer[i]['payload'],
messages[send_topics.index(topic)])
