Skip to content

Commit

Permalink
restore this
Browse files Browse the repository at this point in the history
  • Loading branch information
silviogutierrez committed Aug 15, 2024
1 parent d07676f commit 8152655
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 2 deletions.
4 changes: 2 additions & 2 deletions reactivated/fields.py
Original file line number Diff line number Diff line change
Expand Up @@ -408,8 +408,8 @@ def get_enum_name(self) -> str:
def db_type(self, connection: Any) -> Optional[str]:
if connection.settings_dict["ENGINE"] != "django.db.backends.postgresql":
raise DatabaseError("EnumField is only supported on PostgreSQL")
return f"{self.get_enum_name()}[]"
# return super().db_type(connection)
# return f"{self.get_enum_name()}[]"
return super().db_type(connection)

def __init__(self, *args, **kwargs) -> None:
self.enum = kwargs.pop("enum")
Expand Down
14 changes: 14 additions & 0 deletions tests/fields.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,20 @@ class Meta:
assert form.instance.hemisphere == models.Continent.Hemisphere.NORTHERN


@pytest.mark.django_db
def test_enum_array_field():
continent = models.Continent.objects.create(name="Europe")
birth_country = models.Country.objects.create(name="Germany", continent=continent)
other = models.Country.objects.create(name="Switzerland", continent=continent)

composer = models.Composer.objects.create(name="Wagner")
models.ComposerCountry.objects.create(
composer=composer, country=birth_country, was_born=True
)
models.ComposerCountry.objects.create(composer=composer, country=other)
opera = models.Opera.objects.create(name="Götterdämmerung", composer=composer)


def test_enum_form_field():
class EnumForm(forms.Form):
enum_form_field = EnumChoiceField(enum=EnumTest)
Expand Down

0 comments on commit 8152655

Please sign in to comment.