def test_get_error_class_with_none_code(self):
pg_code = None
cls = registry.get_error_class(pg_code)
self.assertEqual(cls, PgClientError)
def test_get_error_class_with_none_code(self):
pg_code = None
cls = registry.get_error_class(pg_code)
self.assertEqual(cls, PgClientError)
def test_get_error_class(self):
pg_code = '42P01'
cls = registry.get_error_class(pg_code)
self.assertTrue(issubclass(cls, PgClientError))
self.assertEqual(cls.CLASS_CODE, '42')
def test_get_error_class(self):
pg_code = '42P01'
cls = registry.get_error_class(pg_code)
self.assertTrue(issubclass(cls, PgClientError))
self.assertEqual(cls.CLASS_CODE, '42')
