diff --git a/tests/test_forms.py b/tests/test_forms.py index 7d9082b0a..33ba45d3e 100644 --- a/tests/test_forms.py +++ b/tests/test_forms.py @@ -5,7 +5,7 @@ Boolean, Date, DateTime, - Enum, + # Enum, Float, Integer, Numeric, @@ -13,7 +13,6 @@ Text, ) -from flask_appbuilder.fields import EnumField from wtforms import ( BooleanField, DateField, @@ -25,19 +24,19 @@ TextAreaField, ) +from flask import Flask from flask_appbuilder.models.sqla.interface import SQLAInterface +# from flask_appbuilder.fields import EnumField from flask_appbuilder.forms import GeneralModelConverter from flask_appbuilder import Model -from flask import Flask - class FieldsModel(Model): id = Column(Integer, primary_key=True) field_boolean = Column(Boolean()) field_date = Column(Date()) field_datetime = Column(DateTime()) - field_enum = Column(Enum()) + # field_enum = Column(Enum()) field_float = Column(Float()) field_integer = Column(Integer()) field_numeric_scale0 = Column(Numeric()) @@ -56,7 +55,7 @@ def test_model_without_context(self): "field_boolean", "field_date", "field_datetime", - "field_enum", + # "field_enum", "field_float", "field_integer", "field_numeric_scale0", @@ -69,7 +68,7 @@ def test_model_without_context(self): self.assertTrue(form.field_boolean.field_class is BooleanField) self.assertTrue(form.field_date.field_class is DateField) self.assertTrue(form.field_datetime.field_class is DateTimeField) - self.assertTrue(form.field_enum.field_class is EnumField) + # self.assertTrue(form.field_enum.field_class is EnumField) self.assertTrue(form.field_float.field_class is FloatField) self.assertTrue(form.field_integer.field_class is IntegerField) self.assertTrue(form.field_numeric_scale0.field_class is DecimalField and not form.field_numeric_scale0.kwargs["places"]) @@ -85,7 +84,7 @@ def test_model_with_context(self): "field_boolean", "field_date", "field_datetime", - "field_enum", + # "field_enum", "field_float", "field_integer", "field_numeric_scale0", @@ -98,12 +97,11 @@ def test_model_with_context(self): app.config["CSRF_ENABLED"] = True app.config["SECRET_KEY"] = "thisismysecretkey" with app.test_request_context(): - # with app.app_context(): form = conv.create_form(None, columns)() self.assertTrue(isinstance(form.field_boolean, BooleanField)) self.assertTrue(isinstance(form.field_date, DateField)) self.assertTrue(isinstance(form.field_datetime, DateTimeField)) - self.assertTrue(isinstance(form.field_enum, EnumField)) + # self.assertTrue(isinstance(form.field_enum, EnumField)) self.assertTrue(isinstance(form.field_float, FloatField)) self.assertTrue(isinstance(form.field_integer, IntegerField)) self.assertTrue(isinstance(form.field_numeric_scale0, DecimalField) and not form.field_numeric_scale0.places)