class TypeRegistryTests(unittest.TestCase): def setUp(self): self.config = pyramid.testing.setUp() self.types = TypeRegistry() def tearDown(self): pyramid.testing.tearDown() def test_register(self): # Register a type self.assertEqual([], self.types.names) self.types.register('foo', None) self.assertEqual(['foo'], self.types.names) def test_unregister_unknown(self): # Unregister unknown self.assertRaises(NotRegisteredError, self.types.unregister, 'foo') def test_unregister(self): self.types.register('bar', None) self.assertEqual(['bar'], self.types.names) self.types.unregister('bar') self.assertEqual([], self.types.names) self.assertRaises(UnknownFieldTypeError, self.types.definition, 'bar') self.assertRaises(UnknownFieldTypeError, self.types.validation, 'bar') def test_register_again(self): self.types.register('foo', None) self.assertRaises(AlreadyRegisteredError, self.types.register, 'foo', None)
def setUp(self): self.config = pyramid.testing.setUp() self.types = TypeRegistry()