def test_subscribe_new(self): p = PubSubManager() p.subscribe("client", "topic") # check that the topic was created self.assertEqual(len(p.subscriptions.keys()), 1) # check that we are subbed self.assertTrue("client" in p.subscriptions["topic"]) self.assertTrue(len(p.subscriptions["topic"]), 1)
def test_subscribe_new(self): p = PubSubManager() p.subscribe("client", "topic") # check that the topic was created self.assertEqual(len(p.subscriptions.keys()), 1) # check that we are subbed self.assertTrue("client" in p.subscriptions["topic"]) self.assertTrue(len(p.subscriptions["topic"]), 1)
def test_subscribe_existing(self): p = PubSubManager() p.subscriptions["topic"] = {"other client"} p.subscribe("client", "topic") # check that the topic was reused self.assertEqual(len(p.subscriptions.keys()), 1) # check that we are subbed self.assertTrue("client" in p.subscriptions["topic"]) self.assertTrue(len(p.subscriptions["topic"]), 2)
def test_subscribe_existing(self): p = PubSubManager() p.subscriptions["topic"] = set(["other client"]) p.subscribe("client", "topic") # check that the topic was reused self.assertEqual(len(p.subscriptions.keys()), 1) # check that we are subbed self.assertTrue("client" in p.subscriptions["topic"]) self.assertTrue(len(p.subscriptions["topic"]), 2)