def test_reset(self): registry = Registry() expected = DEFAULT_GENERATOR_NAMES registry.register('abstract', AbstractKeyGenerator) now_expected = DEFAULT_GENERATOR_NAMES + ['abstract'] self.assertEqual(set(registry.available()), set(now_expected)) registry.reset() self.assertEqual(set(registry.available()), set(expected))
def test_available(self): registry = Registry() available = registry.available() expected = DEFAULT_GENERATOR_NAMES self.assertEqual(set(available), set(expected))