From 9edd6953ccb2e173ffc94a8f3eaff8e3fa236e7f Mon Sep 17 00:00:00 2001 From: Alex Gaynor Date: Thu, 18 Nov 2010 23:51:20 +0000 Subject: [PATCH] [1.2.X] Converted the model_regress doctests into unittests. We have always been at war with doctests. Backport of [14614]. git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@14617 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- tests/regressiontests/model_regress/models.py | 116 +------------ tests/regressiontests/model_regress/tests.py | 161 +++++++++++++++++- 2 files changed, 156 insertions(+), 121 deletions(-) diff --git a/tests/regressiontests/model_regress/models.py b/tests/regressiontests/model_regress/models.py index 6c5d7bd59e..f30b3eecd7 100644 --- a/tests/regressiontests/model_regress/models.py +++ b/tests/regressiontests/model_regress/models.py @@ -1,9 +1,6 @@ # coding: utf-8 -import datetime +from django.db import models -from django.conf import settings -from django.db import models, DEFAULT_DB_ALIAS -from django.utils import tzinfo CHOICES = ( (1, 'first'), @@ -60,114 +57,3 @@ class BrokenUnicodeMethod(models.Model): class NonAutoPK(models.Model): name = models.CharField(max_length=10, primary_key=True) - -__test__ = {'API_TESTS': """ -(NOTE: Part of the regression test here is merely parsing the model -declaration. The verbose_name, in particular, did not always work.) - -An empty choice field should return None for the display name. - ->>> from datetime import datetime ->>> a = Article(headline="Look at me!", pub_date=datetime.now()) ->>> a.save() ->>> a.get_status_display() is None -True - -Empty strings should be returned as Unicode ->>> a2 = Article.objects.get(pk=a.id) ->>> a2.misc_data -u'' - -# TextFields can hold more than 4000 characters (this was broken in Oracle). ->>> a3 = Article(headline="Really, really big", pub_date=datetime.now()) ->>> a3.article_text = "ABCDE" * 1000 ->>> a3.save() ->>> a4 = Article.objects.get(pk=a3.id) ->>> len(a4.article_text) -5000 - -# Regression test for #659 ->>> import datetime ->>> p = Party.objects.create(when = datetime.datetime(1999, 12, 31)) ->>> p = Party.objects.create(when = datetime.datetime(1998, 12, 31)) ->>> p = Party.objects.create(when = datetime.datetime(1999, 1, 1)) ->>> [p.when for p in Party.objects.filter(when__month=2)] -[] ->>> [p.when for p in Party.objects.filter(when__month=1)] -[datetime.date(1999, 1, 1)] ->>> [p.when for p in Party.objects.filter(when__month=12)] -[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)] ->>> [p.when for p in Party.objects.filter(when__year=1998)] -[datetime.date(1998, 12, 31)] - -# Regression test for #8510 ->>> [p.when for p in Party.objects.filter(when__day='31')] -[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)] ->>> [p.when for p in Party.objects.filter(when__month='12')] -[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)] ->>> [p.when for p in Party.objects.filter(when__year='1998')] -[datetime.date(1998, 12, 31)] - -# Date filtering was failing with NULL date values in SQLite (regression test -# for #3501, amongst other things). ->>> _ = Party.objects.create() ->>> p = Party.objects.filter(when__month=1)[0] ->>> p.when -datetime.date(1999, 1, 1) ->>> l = Party.objects.filter(pk=p.pk).dates("when", "month") ->>> l[0].month == 1 -True - -# Check that get_next_by_FIELD and get_previous_by_FIELD don't crash when we -# have usecs values stored on the database -# -# [It crashed after the Field.get_db_prep_* refactor, because on most backends -# DateTimeFields supports usecs, but DateTimeField.to_python didn't recognize -# them. (Note that Model._get_next_or_previous_by_FIELD coerces values to -# strings)] -# ->>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 16, 0, 0)) ->>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 6, 1, 1)) ->>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 13, 1, 1)) ->>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 12, 0, 20, 24)) ->>> e.get_next_by_when().when -datetime.datetime(2000, 1, 1, 13, 1, 1) ->>> e.get_previous_by_when().when -datetime.datetime(2000, 1, 1, 6, 1, 1) - -# Check Department and Worker ->>> d = Department(id=10, name='IT') ->>> d.save() ->>> w = Worker(department=d, name='Full-time') ->>> w.save() ->>> w - - -# Models with broken unicode methods should still have a printable repr ->>> b = BrokenUnicodeMethod(name="Jerry") ->>> b.save() ->>> BrokenUnicodeMethod.objects.all() -[] - -"""} - -if settings.DATABASES[DEFAULT_DB_ALIAS]['ENGINE'] not in ( - "django.db.backends.mysql", - "django.db.backends.oracle"): - __test__["timezone-tests"] = """ -# Saving an updating with timezone-aware datetime Python objects. Regression -# test for #10443. - -# The idea is that all these creations and saving should work without crashing. -# It's not rocket science. ->>> Article.objects.all().delete() ->>> dt1 = datetime.datetime(2008, 8, 31, 16, 20, tzinfo=tzinfo.FixedOffset(600)) ->>> dt2 = datetime.datetime(2008, 8, 31, 17, 20, tzinfo=tzinfo.FixedOffset(600)) ->>> obj = Article.objects.create(headline="A headline", pub_date=dt1, article_text="foo") - ->>> obj.pub_date = dt2 ->>> obj.save() ->>> Article.objects.filter(headline="A headline").update(pub_date=dt1) -1 - -""" diff --git a/tests/regressiontests/model_regress/tests.py b/tests/regressiontests/model_regress/tests.py index 6397b90a66..8009839a09 100644 --- a/tests/regressiontests/model_regress/tests.py +++ b/tests/regressiontests/model_regress/tests.py @@ -1,23 +1,172 @@ +import datetime +from operator import attrgetter + +from django.conf import settings from django.core.exceptions import ValidationError +from django.db import DEFAULT_DB_ALIAS from django.test import TestCase +from django.utils import tzinfo -from models import Worker, NonAutoPK +from models import (Worker, Article, Party, Event, Department, + BrokenUnicodeMethod, NonAutoPK) -class RelatedModelOrderedLookupTest(TestCase): - """ - Regression test for #10153: foreign key __gte and __lte lookups. - """ +class ModelTests(TestCase): # The bug is that the following queries would raise: # "TypeError: Related Field has invalid lookup: gte" - def test_related_gte_lookup(self): + """ + Regression test for #10153: foreign key __gte lookups. + """ Worker.objects.filter(department__gte=0) def test_related_lte_lookup(self): + """ + Regression test for #10153: foreign key __lte lookups. + """ Worker.objects.filter(department__lte=0) + def test_empty_choice(self): + # NOTE: Part of the regression test here is merely parsing the model + # declaration. The verbose_name, in particular, did not always work. + a = Article.objects.create( + headline="Look at me!", pub_date=datetime.datetime.now() + ) + # An empty choice field should return None for the display name. + self.assertEqual(a.get_status_display(), None) + + # Empty strings should be returned as Unicode + a = Article.objects.get(pk=a.pk) + self.assertEqual(a.misc_data, u'') + self.assertEqual(type(a.misc_data), unicode) + + def test_long_textfield(self): + # TextFields can hold more than 4000 characters (this was broken in + # Oracle). + a = Article.objects.create( + headline="Really, really big", + pub_date=datetime.datetime.now(), + article_text = "ABCDE" * 1000 + ) + a = Article.objects.get(pk=a.pk) + self.assertEqual + (len(a.article_text), 5000) + + def test_date_lookup(self): + # Regression test for #659 + Party.objects.create(when=datetime.datetime(1999, 12, 31)) + Party.objects.create(when=datetime.datetime(1998, 12, 31)) + Party.objects.create(when=datetime.datetime(1999, 1, 1)) + self.assertQuerysetEqual( + Party.objects.filter(when__month=2), [] + ) + self.assertQuerysetEqual( + Party.objects.filter(when__month=1), [ + datetime.date(1999, 1, 1) + ], + attrgetter("when") + ) + self.assertQuerysetEqual( + Party.objects.filter(when__month=12), [ + datetime.date(1999, 12, 31), + datetime.date(1998, 12, 31), + ], + attrgetter("when") + ) + self.assertQuerysetEqual( + Party.objects.filter(when__year=1998), [ + datetime.date(1998, 12, 31), + ], + attrgetter("when") + ) + # Regression test for #8510 + self.assertQuerysetEqual( + Party.objects.filter(when__day="31"), [ + datetime.date(1999, 12, 31), + datetime.date(1998, 12, 31), + ], + attrgetter("when") + ) + self.assertQuerysetEqual( + Party.objects.filter(when__month="12"), [ + datetime.date(1999, 12, 31), + datetime.date(1998, 12, 31), + ], + attrgetter("when") + ) + self.assertQuerysetEqual( + Party.objects.filter(when__year="1998"), [ + datetime.date(1998, 12, 31), + ], + attrgetter("when") + ) + + def test_date_filter_null(self): + # Date filtering was failing with NULL date values in SQLite + # (regression test for #3501, amongst other things). + Party.objects.create(when=datetime.datetime(1999, 1, 1)) + Party.objects.create() + p = Party.objects.filter(when__month=1)[0] + self.assertEqual(p.when, datetime.date(1999, 1, 1)) + self.assertQuerysetEqual( + Party.objects.filter(pk=p.pk).dates("when", "month"), [ + 1 + ], + attrgetter("month") + ) + + def test_get_next_prev_by_field(self): + # Check that get_next_by_FIELD and get_previous_by_FIELD don't crash + # when we have usecs values stored on the database + # + # It crashed after the Field.get_db_prep_* refactor, because on most + # backends DateTimeFields supports usecs, but DateTimeField.to_python + # didn't recognize them. (Note that + # Model._get_next_or_previous_by_FIELD coerces values to strings) + Event.objects.create(when=datetime.datetime(2000, 1, 1, 16, 0, 0)) + Event.objects.create(when=datetime.datetime(2000, 1, 1, 6, 1, 1)) + Event.objects.create(when=datetime.datetime(2000, 1, 1, 13, 1, 1)) + e = Event.objects.create(when=datetime.datetime(2000, 1, 1, 12, 0, 20, 24)) + + self.assertEqual( + e.get_next_by_when().when, datetime.datetime(2000, 1, 1, 13, 1, 1) + ) + self.assertEqual( + e.get_previous_by_when().when, datetime.datetime(2000, 1, 1, 6, 1, 1) + ) + + def test_primary_key_foreign_key_types(self): + # Check Department and Worker (non-default PK type) + d = Department.objects.create(id=10, name="IT") + w = Worker.objects.create(department=d, name="Full-time") + self.assertEqual(unicode(w), "Full-time") + + def test_broken_unicode(self): + # Models with broken unicode methods should still have a printable repr + b = BrokenUnicodeMethod.objects.create(name="Jerry") + self.assertEqual(repr(b), "") + + if settings.DATABASES[DEFAULT_DB_ALIAS]["ENGINE"] not in [ + "django.db.backends.mysql", + "django.db.backends.oracle" + ]: + def test_timezones(self): + # Saving an updating with timezone-aware datetime Python objects. + # Regression test for #10443. + # The idea is that all these creations and saving should work + # without crashing. It's not rocket science. + dt1 = datetime.datetime(2008, 8, 31, 16, 20, tzinfo=tzinfo.FixedOffset(600)) + dt2 = datetime.datetime(2008, 8, 31, 17, 20, tzinfo=tzinfo.FixedOffset(600)) + obj = Article.objects.create( + headline="A headline", pub_date=dt1, article_text="foo" + ) + obj.pub_date = dt2 + obj.save() + self.assertEqual( + Article.objects.filter(headline="A headline").update(pub_date=dt1), + 1 + ) class ModelValidationTest(TestCase): def test_pk_validation(self):