def testGetSubscriberList(self): pub = Publisher() dum1 = Dummy() dum2 = Dummy() pub.addSubscriber(dum1) pub.addSubscriber(dum2) list1 = pub.getSubscriberList() list2 = [dum1, dum2] self.assertCountEqual(list1, list2)
def testRemoveSubscriber1(self): # tests to see if subscriber in list is removed pub = Publisher() dum1 = Dummy() dum2 = Dummy() pub.addSubscriber(dum1) pub.addSubscriber(dum2) pub.removeSubscriber(dum2) self.assertEqual(len(pub.getSubscriberList()), 1)
def testRemoveSubscriber2(self): # tests to make sure nothing is removed since not in list pub = Publisher() dum1 = Dummy() dum2 = Dummy() notAdded = BigDummy() pub.addSubscriber(dum1) pub.addSubscriber(dum2) pub.removeSubscriber(notAdded) self.assertEqual(len(pub.getSubscriberList()), 2)
def testAddSubscriber2(self): # tests to make sure invalid object not added pub = Publisher() dum = BigDummy() pub.addSubscriber(dum) self.assertEqual(len(pub.getSubscriberList()), 0)
def testAddSubscriber1(self): # tests that a valid object is added pub = Publisher() dum = Dummy() pub.addSubscriber(dum) self.assertEqual(len(pub.getSubscriberList()), 1)