def test_subscriber_shall_be_attachable_to_subscriptions(cls):
subscription = "sub msg"
pro = Provider()
cls.assertEqual(len(pro.subscribers), 0)
sub = Subscriber("sub name", pro)
sub.subscribe(subscription)
cls.assertEqual(len(pro.subscribers[subscription]), 1)
def test_provider_shall_update_affected_subscribers_with_published_subscription(
cls, ):
pro = Provider()
pub = Publisher(pro)
sub1 = Subscriber("sub 1 name", pro)
sub1.subscribe("sub 1 msg 1")
sub1.subscribe("sub 1 msg 2")
sub2 = Subscriber("sub 2 name", pro)
sub2.subscribe("sub 2 msg 1")
sub2.subscribe("sub 2 msg 2")
with patch.object(sub1, "run") as mock_subscriber1_run, patch.object(
sub2, "run") as mock_subscriber2_run:
pro.update()
cls.assertEqual(mock_subscriber1_run.call_count, 0)
cls.assertEqual(mock_subscriber2_run.call_count, 0)
pub.publish("sub 1 msg 1")
pub.publish("sub 1 msg 2")
pub.publish("sub 2 msg 1")
pub.publish("sub 2 msg 2")
with patch.object(sub1, "run") as mock_subscriber1_run, patch.object(
sub2, "run") as mock_subscriber2_run:
pro.update()
expected_sub1_calls = [call("sub 1 msg 1"), call("sub 1 msg 2")]
mock_subscriber1_run.assert_has_calls(expected_sub1_calls)
expected_sub2_calls = [call("sub 2 msg 1"), call("sub 2 msg 2")]
mock_subscriber2_run.assert_has_calls(expected_sub2_calls)
def test_subscriber_shall_be_detachable_from_subscriptions(cls):
subscription = 'sub msg'
pro = Provider()
sub = Subscriber('sub name', pro)
sub.subscribe(subscription)
cls.assertEqual(len(pro.subscribers[subscription]), 1)
sub.unsubscribe(subscription)
cls.assertEqual(len(pro.subscribers[subscription]), 0)