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)
