def test_with_threads(self):
sp = SubPub()
q = sp.subscribe(r'\d+')
for i in range(10):
t = threading.Thread(target=sp.publish, args=(str(i), i))
t.start()
self.assertEqual([q.get()[1] for _ in range(10)], list(range(10)))
def test_subscribe_simple(self):
sp = SubPub()
q = sp.subscribe("Aerith/Theme")
recv = sp.publish("Aerith/Theme", 81611)
self.assertTrue(recv)
match, data = q.get()
self.assertTrue(match)
self.assertEqual(data, 81611)
def test_publish_remove_retained(self):
sp = SubPub()
topic = 'JudgmentDay'
sp.publish(topic, 'anything but None', retain=True)
sp.subscribe(topic).get_nowait()
sp.publish(topic, None, retain=True)
with self.assertRaises(queue.Empty):
sp.subscribe(topic).get_nowait()
def test_publish_retained(self):
sp = SubPub()
topic = 'Sober'
data_in = 'Nope'
recv = sp.publish(topic, data_in, retain=True)
self.assertFalse(recv)
q = sp.subscribe(topic)
match, data_out = q.get_nowait()
self.assertEqual(match.re.pattern, topic)
self.assertIs(data_out, data_in)
def test_unsubscribe_gc(self):
sp = SubPub()
def f():
q = sp.subscribe('Trädgårdsgatan/23')
recv = sp.publish('Trädgårdsgatan/23', 404)
self.assertTrue(recv)
f()
recv = sp.publish('Trädgårdsgatan/23', 404)
# q garbage collected => automatic unsubscribe
self.assertFalse(recv)
def test_publish_block_and_timeout(self):
sp = SubPub()
topic = 'Saturn'
q = sp.subscribe(topic, queue=queue.Queue(1))
sp.publish(topic, 'Pollux')
with self.assertRaises(queue.Full):
sp.publish(topic, 'Castor', timeout=0.001)
def test_mqtt_style(self):
sp = SubPub()
topic = MqttTopic('Daniel/+/+/+/#')
q = sp.subscribe(topic)
sp.publish('Daniel/2005/12/18/02:45:00', 'So help me God')
match, data = q.get(timeout=1)
self.assertEqual(match.groups(), ('2005', '12', '18', '02:45:00'))
self.assertEqual(data, 'So help me God')
self.assertTrue(sp.unsubscribe(topic))
def test_unsubscribe(self):
sp = SubPub()
topic = "Olofsbäcks Gård"
q = sp.subscribe(topic)
sp.publish(topic, 81611)
self.assertTrue(q.get_nowait())
ok = sp.unsubscribe(topic)
self.assertTrue(ok)
with self.assertRaises(queue.Empty):
q.get_nowait()
def test_subscribe_custom_queue(self):
sp = SubPub()
q = queue.PriorityQueue()
q_out = sp.subscribe('', queue=q)
self.assertIs(q_out, q)
q_out = sp.subscribe('', queue=q)
self.assertIs(q_out, q)
q_out = sp.subscribe('')
self.assertIsInstance(q_out, queue.SimpleQueue)
q_out = sp.subscribe('', queue=queue.LifoQueue())
self.assertIsInstance(q_out, queue.LifoQueue)
def test_unsubscribe_all(self):
sp = SubPub()
self.assertEqual(sp.unsubscribe_all(), 0)
sp.subscribe("Chalmers")
sp.subscribe("GU")
self.assertEqual(sp.unsubscribe_all(), 0)
# Keep references
q1 = sp.subscribe("Chalmers")
q2 = sp.subscribe("GU")
self.assertEqual(sp.unsubscribe_all(), 2)
def test_unsubscribe_not_subscribed(self):
sp = SubPub()
self.assertFalse(sp.unsubscribe("Happiness"))
def test_subscribe_twice(self):
sp = SubPub()
q1 = sp.subscribe('EyesOnMe')
q2 = sp.subscribe('EyesOnMe')
self.assertIsNot(q1, q2)
def test_queue_type(self):
sp = SubPub(ExceptionAwareQueue)
q = sp.subscribe('MelodiesOfLife')
self.assertIsInstance(q, ExceptionAwareQueue)
def test_custom_queue_factory(self):
self.assertIs(
SubPub(ExceptionAwareQueue).queue_factory, ExceptionAwareQueue)
def test_default_queue_factory(self):
self.assertIs(SubPub().queue_factory, queue.SimpleQueue)
def test_publish_no_one_cares(self):
sp = SubPub()
recv = sp.publish('Hans Andersson', 'Water')
self.assertFalse(recv)