def test_integer_field_should_allow_filters():
# given
from tests.app.models import User, UserStatus
# when
instance = User()
instance.status = UserStatus.ACTIVE
instance.save()
User.objects.create(status=UserStatus.INACTIVE)
# then
assert instance.status.value == UserStatus.ACTIVE
instance2 = User.objects.filter(status=UserStatus.ACTIVE).first()
assert instance.pk == instance2.pk
def test_creator_should_call_field_to_python_on_assigment(self):
from tests.app.models import User, UserStatus
with mock.patch(
'choicesenum.django.fields.EnumFieldMixin.to_python') as m:
m.return_value = UserStatus.DELETED
# given
instance = User()
# when
instance.status = UserStatus.ACTIVE
# then
m.assert_called_with(UserStatus.ACTIVE)
assert instance.status == UserStatus.DELETED
def test_field_sqlite():
User(email='*****@*****.**').save()
user = User.objects.filter(email='*****@*****.**').first()
assert user
assert user.email == '*****@*****.**'
def test_field_postgresql():
User(email='*****@*****.**').save(using='psql')
user = User.objects.using('psql').filter(email='*****@*****.**').first()
assert user
assert user.email == '*****@*****.**'