From fe85e286a2f6cffd70e56f6c30475dbc68a5feea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gregor=20M=C3=BCllegger?= Date: Wed, 9 Jun 2010 11:08:58 +0200 Subject: [PATCH] Fixing small bug with CharFields.max_length < 15 --- src/autofixture/base.py | 1 + src/autofixture/tests.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/src/autofixture/base.py b/src/autofixture/base.py index 02e8e0e..b2192f8 100644 --- a/src/autofixture/base.py +++ b/src/autofixture/base.py @@ -307,6 +307,7 @@ def get_generator(self, field): return generators.LoremSentenceGenerator( common=False, max_length=field.max_length) + else: generator = generators.StringGenerator return generator(max_length=field.max_length) if isinstance(field, fields.DecimalField): diff --git a/src/autofixture/tests.py b/src/autofixture/tests.py index 8fa85eb..afbbcf6 100644 --- a/src/autofixture/tests.py +++ b/src/autofixture/tests.py @@ -47,6 +47,7 @@ class NullableFKModel(models.Model): class BasicModel(models.Model): chars = models.CharField(max_length=50) + shortchars = models.CharField(max_length=2) blankchars = models.CharField(max_length=100, blank=True) nullchars = models.CharField(max_length=100, blank=True, null=True) slugfield = models.SlugField() @@ -138,6 +139,8 @@ def test_constraints(self): for obj in filler.create(100): self.assertTrue(len(obj.chars) > 0) self.assertEqual(type(obj.chars), unicode) + self.assertTrue(len(obj.shortchars) <= 2) + self.assertEqual(type(obj.shortchars), unicode) self.assertTrue(type(obj.blankchars), unicode) self.assertEqualOr(type(obj.nullchars), unicode, None) self.assertEqual(type(obj.slugfield), unicode)