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()
