def test_unregister(self): gen = AbstractKeyGenerator() gen.register() registry = Registry() self.assertEqual(registry.get('abstract'), gen) gen.unregister() self.assertRaises(GeneratorError, registry.get, 'abstract')
def test_register(self): gen = AbstractKeyGenerator() gen.register() registry = Registry() self.assertEqual(registry.get('abstract'), gen)
def test_unregister(self): registry = Registry() registry.register('abstract', AbstractKeyGenerator) registry.unregister('abstract') self.assertRaises(GeneratorError, registry.get, 'abstract') self.assertFalse(registry.get('abstract', False))
def test_register(self): registry = Registry() registry.register('abstract', AbstractKeyGenerator) self.assertEqual(registry.get('abstract'), AbstractKeyGenerator)
def test_get(self): registry = Registry() sms = registry.get('sms') self.assertEqual(sms, SMSKeyGenerator)