def testCanSeeRegistrationOnFirstRegistrant(self):
res = initA.register()
self.assertEqual(res, 0)
res = initB.register()
self.assertEqual(res, 0)
registrantsA = initA.getRegistrants()
self.assertEqual(len(registrantsA), 2)
self.assertIn(initA.key, registrantsA)
self.assertIn(initB.key, registrantsA)
def testReregisterFails(self):
initAcopy = copy(initA)
initAcopy.key = "0x1"
resA = initAcopy.register()
self.assertNotEqual(resA, 0)
registrantsA = initA.getRegistrants()
self.assertEqual(len(registrantsA), 2)
self.assertIn(initA.key, registrantsA)
self.assertIn(initB.key, registrantsA)
def testCanUnregister(self):
res = initA.unregister()
self.assertEqual(res, 0)
registrants = initA.getRegistrants()
self.assertEqual(len(registrants), 1)
res = initB.unregister()
self.assertEqual(res, 0)
registrants = initB.getRegistrants()
self.assertEqual(len(registrants), 0)
def testCanRegisterAndIgnore(self):
# register with key "0x1"
initAcopy = copy(initA)
initAcopy.key = "0x1"
result = initA.registerAndIgnore(initAcopy.key)
self.assertEqual(result, 0)
registrantsA = initAcopy.getRegistrants()
self.assertNotIn(initA.key, registrantsA)
self.assertIn(initAcopy.key, registrantsA)
# re-register with normal key
result = initAcopy.registerAndIgnore(initA.key)
self.assertEqual(result, 0)
registrantsA = initA.getRegistrants()
self.assertNotIn(initAcopy.key, registrantsA)
self.assertIn(initA.key, registrantsA)
def testCanSeeNoRegistrations(self):
registrantsA = initA.getRegistrants()
self.assertEqual(len(registrantsA), 0)
