1
0
mirror of https://github.com/django/django.git synced 2025-07-05 18:29:11 +00:00

Converted modeltests.choices to unittests

git-svn-id: http://code.djangoproject.com/svn/django/branches/soc2010/test-refactor@13344 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Paul McMillan 2010-06-09 22:13:03 +00:00
parent e3bad40d3e
commit 7289013cf1
3 changed files with 49 additions and 26 deletions

View File

@ -0,0 +1,18 @@
[
{
"pk": 1,
"model": "choices.person",
"fields": {
"gender": "M",
"name": "Adrian"
}
},
{
"pk": 2,
"model": "choices.person",
"fields": {
"gender": "F",
"name": "Sara"
}
}
]

View File

@ -22,29 +22,3 @@ class Person(models.Model):
def __unicode__(self):
return self.name
__test__ = {'API_TESTS':"""
>>> a = Person(name='Adrian', gender='M')
>>> a.save()
>>> s = Person(name='Sara', gender='F')
>>> s.save()
>>> a.gender
'M'
>>> s.gender
'F'
>>> a.get_gender_display()
u'Male'
>>> s.get_gender_display()
u'Female'
# If the value for the field doesn't correspond to a valid choice,
# the value itself is provided as a display value.
>>> a.gender = ''
>>> a.get_gender_display()
u''
>>> a.gender = 'U'
>>> a.get_gender_display()
u'U'
"""}

View File

@ -0,0 +1,31 @@
from django.test import TestCase
from models import Person
class ChoicesFieldTestCase(TestCase):
fixtures = ['initial_data.json']
def setUp(self):
self.a = Person.objects.get(name='Adrian')
self.s = Person.objects.get(name='Sara')
def test_choice_storage(self):
self.assertEqual(self.a.gender,
'M')
self.assertEqual(self.s.gender,
'F')
def test_gender_display(self):
self.assertEqual(self.a.get_gender_display(),
u'Male')
self.assertEqual(self.s.get_gender_display(),
u'Female')
# If the value for the field doesn't correspond to a valid choice,
# the value itself is provided as a display value.
self.a.gender = ''
self.assertEqual(self.a.get_gender_display(),
u'')
self.a.gender = 'U'
self.assertEqual(self.a.get_gender_display(),
u'U')