def afterSetUp(self): self.reg = TMRegistry()
class TestTMRegistry(ZopeTestCase.TestCase): def afterSetUp(self): self.reg = TMRegistry() def testRegister(self): self.assertEqual(len(self.reg), 0) self.reg.register(Target()) self.assertEqual(len(self.reg), 1) self.reg.register(Target()) self.assertEqual(len(self.reg), 2) def testCount(self): self.assertEqual(self.reg.count(), 0) self.reg.register(Target()) self.assertEqual(self.reg.count(), 1) self.reg.register(Target()) self.assertEqual(self.reg.count(), 2) def testContains(self): target1, target2 = Target(), Target() self.reg.register(target1) self.reg.register(target2) self.failUnless(self.reg.contains(target1)) self.failUnless(self.reg.contains(target2)) def testRemove(self): target1, target2 = Target(), Target() self.reg.register(target1) self.reg.register(target2) self.reg.remove(target1) self.reg.remove(target2) self.failIf(self.reg.contains(target1)) self.failIf(self.reg.contains(target2)) def testGet(self): target1 = Target() self.reg.register(target1) tm = self.reg.get(target1) self.failUnless(isinstance(tm, ProxyTM)) self.assertEqual(target1, tm._target)