Moved tests for model Field.get_choices().
This commit is contained in:
parent
11b8c30b9e
commit
e35004966b
|
@ -3,7 +3,6 @@ from unittest import skipIf
|
|||
from django.core.exceptions import ValidationError
|
||||
from django.db import connection, models
|
||||
from django.test import SimpleTestCase, TestCase
|
||||
from django.utils.functional import lazy
|
||||
|
||||
from .models import Post
|
||||
|
||||
|
@ -50,17 +49,6 @@ class ValidationTests(SimpleTestCase):
|
|||
with self.assertRaises(ValidationError):
|
||||
f.clean('not a', None)
|
||||
|
||||
def test_charfield_get_choices_with_blank_defined(self):
|
||||
f = models.CharField(choices=[('', '<><>'), ('a', 'A')])
|
||||
self.assertEqual(f.get_choices(True), [('', '<><>'), ('a', 'A')])
|
||||
|
||||
def test_charfield_get_choices_doesnt_evaluate_lazy_strings(self):
|
||||
# Regression test for #23098
|
||||
# Will raise ZeroDivisionError if lazy is evaluated
|
||||
lazy_func = lazy(lambda x: 0 / 0, int)
|
||||
f = models.CharField(choices=[(lazy_func('group'), (('a', 'A'), ('b', 'B')))])
|
||||
self.assertEqual(f.get_choices(True)[0], ('', '---------'))
|
||||
|
||||
def test_charfield_raises_error_on_empty_input(self):
|
||||
f = models.CharField(null=False)
|
||||
with self.assertRaises(ValidationError):
|
||||
|
|
|
@ -3,6 +3,7 @@ import pickle
|
|||
from django import forms
|
||||
from django.db import models
|
||||
from django.test import SimpleTestCase, TestCase
|
||||
from django.utils.functional import lazy
|
||||
|
||||
from .models import (
|
||||
Foo, RenamedField, VerboseNameField, Whiz, WhizIter, WhizIterEmpty,
|
||||
|
@ -130,3 +131,16 @@ class ChoicesTests(SimpleTestCase):
|
|||
self.assertEqual(WhizIterEmpty(c="b").c, "b") # Invalid value
|
||||
self.assertIsNone(WhizIterEmpty(c=None).c) # Blank value
|
||||
self.assertEqual(WhizIterEmpty(c='').c, '') # Empty value
|
||||
|
||||
|
||||
class GetChoicesTests(SimpleTestCase):
|
||||
|
||||
def test_blank_in_choices(self):
|
||||
choices = [('', '<><>'), ('a', 'A')]
|
||||
f = models.CharField(choices=choices)
|
||||
self.assertEqual(f.get_choices(include_blank=True), choices)
|
||||
|
||||
def test_lazy_strings_not_evaluated(self):
|
||||
lazy_func = lazy(lambda x: 0 / 0, int) # raises ZeroDivisionError if evaluated.
|
||||
f = models.CharField(choices=[(lazy_func('group'), (('a', 'A'), ('b', 'B')))])
|
||||
self.assertEqual(f.get_choices(include_blank=True)[0], ('', '---------'))
|
||||
|
|
Loading…
Reference in New Issue