mirror of
https://github.com/Hopiu/django-model-utils.git
synced 2026-03-17 04:10:24 +00:00
ffa1a85dc7
* Modernize Python syntax, add Python 3.8 * Update Python version & dist in TravisCI * Add postgresql as addon * Switch to psycopg2-binary * Drop django.utils.six
30 lines
983 B
Python
30 lines
983 B
Python
from django.test import TestCase
|
|
|
|
from model_utils.fields import StatusField
|
|
from 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()
|