def test_get_field_type_pg_array_column_raises_when_item_type_not_found(
self):
class DummyType:
python_type = None
column = sqa.Column(postgresql.ARRAY(item_type=DummyType))
with self.assertRaises(KeyError):
field_mapping.get_field_type(column)
def test_get_field_type_can_map_pg_array_column(self):
column = sqa.Column(postgresql.ARRAY(item_type=sqa.Integer()))
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, fields.ListField))
self.assertIsInstance(field().child, fields.IntegerField)
def test_get_field_type_can_map_nullable_bool_column(self):
column = sqa.Column(sqa.Boolean(), nullable=True)
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, fields.NullBooleanField))
def test_get_field_type_can_map_date_column(self):
column = sqa.Column(sqa.Date())
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, fields.DateField))
def test_get_field_type_can_map_interval_column(self):
column = sqa.Column(sqa.Interval())
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, fields.DurationField))
def test_get_field_type_can_map_decimal_column(self):
column = sqa.Column(sqa.Numeric(asdecimal=True))
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, fields.DecimalField))
def test_get_field_type_can_map_float_column(self):
field = field_mapping.get_field_type(sqa.Column(sqa.Float()))
self.assertTrue(issubclass(field, fields.FloatField))
def test_get_field_type_can_map_int_column(self):
field = field_mapping.get_field_type(sqa.Column(sqa.BigInteger()))
self.assertTrue(issubclass(field, fields.IntegerField))
def test_get_field_type_can_map_string_column(self):
field = field_mapping.get_field_type(sqa.Column(sqa.String()))
self.assertTrue(issubclass(field, fields.CharField))
def test_get_field_type_can_map_pg_hstore_column(self):
column = sqa.Column(postgresql.HSTORE())
field = field_mapping.get_field_type(column)
self.assertTrue(issubclass(field, CharMappingField))
