mirror of
https://github.com/Hopiu/django-model-utils.git
synced 2026-03-16 20:00:23 +00:00
33 lines
1 KiB
Python
33 lines
1 KiB
Python
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
from model_utils.fields import StatusField
|
||
|
|
from model_utils.tests.models import (
|
||
|
|
Article, StatusFieldDefaultFilled, StatusFieldDefaultNotFilled,
|
||
|
|
StatusFieldChoicesName,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class StatusFieldTests(TestCase):
|
||
|
|
|
||
|
|
def test_status_with_default_filled(self):
|
||
|
|
instance = StatusFieldDefaultFilled()
|
||
|
|
self.assertEqual(instance.status, instance.STATUS.yes)
|
||
|
|
|
||
|
|
def test_status_with_default_not_filled(self):
|
||
|
|
instance = StatusFieldDefaultNotFilled()
|
||
|
|
self.assertEqual(instance.status, instance.STATUS.no)
|
||
|
|
|
||
|
|
def test_no_check_for_status(self):
|
||
|
|
field = StatusField(no_check_for_status=True)
|
||
|
|
# this model has no STATUS attribute, so checking for it would error
|
||
|
|
field.prepare_class(Article)
|
||
|
|
|
||
|
|
def test_get_status_display(self):
|
||
|
|
instance = StatusFieldDefaultFilled()
|
||
|
|
self.assertEqual(instance.get_status_display(), "Yes")
|
||
|
|
|
||
|
|
def test_choices_name(self):
|
||
|
|
StatusFieldChoicesName()
|