75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
|
from decimal import Decimal
|
||
|
|
|
||
|
|
from django.core import validators
|
||
|
|
from django.core.exceptions import ValidationError
|
||
|
|
from django.db import connection, models
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
from .models import BigD, Foo
|
||
|
|
|
||
|
|
|
||
|
|
class DecimalFieldTests(TestCase):
|
||
|
|
|
||
|
|
def test_to_python(self):
|
||
|
|
f = models.DecimalField(max_digits=4, decimal_places=2)
|
||
|
|
self.assertEqual(f.to_python(3), Decimal('3'))
|
||
|
|
self.assertEqual(f.to_python('3.14'), Decimal('3.14'))
|
||
|
|
with self.assertRaises(ValidationError):
|
||
|
|
f.to_python('abc')
|
||
|
|
|
||
|
|
def test_default(self):
|
||
|
|
f = models.DecimalField(default=Decimal('0.00'))
|
||
|
|
self.assertEqual(f.get_default(), Decimal('0.00'))
|
||
|
|
|
||
|
|
def test_format(self):
|
||
|
|
f = models.DecimalField(max_digits=5, decimal_places=1)
|
||
|
|
self.assertEqual(f._format(f.to_python(2)), '2.0')
|
||
|
|
self.assertEqual(f._format(f.to_python('2.6')), '2.6')
|
||
|
|
self.assertEqual(f._format(None), None)
|
||
|
|
|
||
|
|
def test_get_db_prep_lookup(self):
|
||
|
|
f = models.DecimalField(max_digits=5, decimal_places=1)
|
||
|
|
self.assertEqual(f.get_db_prep_lookup('exact', None, connection=connection), [None])
|
||
|
|
|
||
|
|
def test_filter_with_strings(self):
|
||
|
|
"""
|
||
|
|
Should be able to filter decimal fields using strings (#8023).
|
||
|
|
"""
|
||
|
|
foo = Foo.objects.create(a='abc', d=Decimal('12.34'))
|
||
|
|
self.assertEqual(list(Foo.objects.filter(d='12.34')), [foo])
|
||
|
|
|
||
|
|
def test_save_without_float_conversion(self):
|
||
|
|
"""
|
||
|
|
Ensure decimals don't go through a corrupting float conversion during
|
||
|
|
save (#5079).
|
||
|
|
"""
|
||
|
|
bd = BigD(d='12.9')
|
||
|
|
bd.save()
|
||
|
|
bd = BigD.objects.get(pk=bd.pk)
|
||
|
|
self.assertEqual(bd.d, Decimal('12.9'))
|
||
|
|
|
||
|
|
def test_lookup_really_big_value(self):
|
||
|
|
"""
|
||
|
|
Really big values can be used in a filter statement.
|
||
|
|
"""
|
||
|
|
# This should not crash.
|
||
|
|
Foo.objects.filter(d__gte=100000000000)
|
||
|
|
|
||
|
|
def test_max_digits_validation(self):
|
||
|
|
field = models.DecimalField(max_digits=2)
|
||
|
|
expected_message = validators.DecimalValidator.messages['max_digits'] % {'max': 2}
|
||
|
|
with self.assertRaisesMessage(ValidationError, expected_message):
|
||
|
|
field.clean(100, None)
|
||
|
|
|
||
|
|
def test_max_decimal_places_validation(self):
|
||
|
|
field = models.DecimalField(decimal_places=1)
|
||
|
|
expected_message = validators.DecimalValidator.messages['max_decimal_places'] % {'max': 1}
|
||
|
|
with self.assertRaisesMessage(ValidationError, expected_message):
|
||
|
|
field.clean(Decimal('0.99'), None)
|
||
|
|
|
||
|
|
def test_max_whole_digits_validation(self):
|
||
|
|
field = models.DecimalField(max_digits=3, decimal_places=1)
|
||
|
|
expected_message = validators.DecimalValidator.messages['max_whole_digits'] % {'max': 2}
|
||
|
|
with self.assertRaisesMessage(ValidationError, expected_message):
|
||
|
|
field.clean(Decimal('999'), None)
|