def test_get_topics(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(12, "whiplash")
connection.add_publishing_channel(21, "reload")
topics = connection.get_topics()
expected_topics = {'reload': 21, 'whiplash': 12}
self.assertEqual(topics, expected_topics)
def test_get_publisher_topics(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(12, "again.do.nothing")
connection.add_subscription_channel(32, "start.wars")
expected_topics = ["again.do.nothing", "start.wars"]
topics = connection.get_subscriber_topics()
self.assertEqual(sorted(topics), expected_topics)
def test_get_publisher_topics(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(12, "again.do.nothing")
connection.add_subscription_channel(32, "start.wars")
expected_topics = ["again.do.nothing", "start.wars"]
topics = connection.get_subscriber_topics()
self.assertEqual(sorted(topics), expected_topics)
def test_get_topics(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(12, "whiplash")
connection.add_publishing_channel(21, "reload")
topics = connection.get_topics()
expected_topics = {'reload': 21, 'whiplash': 12}
self.assertEqual(topics, expected_topics)
def test_add_subscription_channel(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(7, "start.wars")
expected_topics = {
"subscriber": {
"start.wars": 7
},
'publisher': {}
}
self.assertEqual(connection.topics, expected_topics)
def test_get_subscription_id(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(5, "a.new.hope")
connection.add_subscription_channel(6, "the.empire.strikes.back")
subscription_id = connection.get_subscription_id("a.new.hope")
self.assertEqual(subscription_id, 5)
def test_get_subscription_id(self):
connection = ClientConnection(websocket=None)
connection.add_subscription_channel(5, "a.new.hope")
connection.add_subscription_channel(6, "the.empire.strikes.back")
subscription_id = connection.get_subscription_id("a.new.hope")
self.assertEqual(subscription_id, 5)
def test_topics_by_subscription_id(self):
connection = ClientConnection(None)
connection.add_subscription_channel(10, "weird")
self.assertEqual(connection.topics_by_subscription_id[10], "weird")
