From 85d2b12bdff63a5c4b269b290904dd499f92a40f Mon Sep 17 00:00:00 2001 From: Adrian Holovaty Date: Fri, 29 Jul 2005 16:20:39 +0000 Subject: [PATCH] Added two more model test modules -- get_latest and lookup git-svn-id: http://code.djangoproject.com/svn/django/trunk@339 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- tests/testapp/models/__init__.py | 2 +- tests/testapp/models/get_latest.py | 44 ++++++++++++++++++++ tests/testapp/models/lookup.py | 67 ++++++++++++++++++++++++++++++ 3 files changed, 112 insertions(+), 1 deletion(-) create mode 100644 tests/testapp/models/get_latest.py create mode 100644 tests/testapp/models/lookup.py diff --git a/tests/testapp/models/__init__.py b/tests/testapp/models/__init__.py index 640028864c..08a7671fb3 100644 --- a/tests/testapp/models/__init__.py +++ b/tests/testapp/models/__init__.py @@ -1 +1 @@ -__all__ = ['basic', 'repr', 'custom_methods', 'many_to_one', 'many_to_many', 'ordering'] +__all__ = ['basic', 'repr', 'custom_methods', 'many_to_one', 'many_to_many', 'ordering', 'lookup', 'get_latest'] diff --git a/tests/testapp/models/get_latest.py b/tests/testapp/models/get_latest.py new file mode 100644 index 0000000000..2bd5a6660b --- /dev/null +++ b/tests/testapp/models/get_latest.py @@ -0,0 +1,44 @@ +""" +8. get_latest_by + +Models can have a ``get_latest_by`` attribute, which should be set to the name +of a DateField or DateTimeField. If ``get_latest_by`` exists, the model's +module will get a ``get_latest()`` function, which will return the latest +object in the database according to that field. "Latest" means "having the +date farthest into the future." +""" + +from django.core import meta + +class Article(meta.Model): + fields = ( + meta.CharField('headline', maxlength=100), + meta.DateTimeField('pub_date'), + ) + get_latest_by = 'pub_date' + + def __repr__(self): + return self.headline + +API_TESTS = """ +# Because no Articles exist yet, get_latest() raises ArticleDoesNotExist. +>>> articles.get_latest() +Traceback (most recent call last): + ... +ArticleDoesNotExist: Article does not exist for {'order_by': ('-pub_date',), 'limit': 1} + +# Create a couple of Articles. +>>> from datetime import datetime +>>> a1 = articles.Article(id=None, headline='Article 1', pub_date=datetime(2005, 7, 26)) +>>> a1.save() +>>> a2 = articles.Article(id=None, headline='Article 2', pub_date=datetime(2005, 7, 27)) +>>> a2.save() +>>> a3 = articles.Article(id=None, headline='Article 3', pub_date=datetime(2005, 7, 27)) +>>> a3.save() +>>> a4 = articles.Article(id=None, headline='Article 4', pub_date=datetime(2005, 7, 28)) +>>> a4.save() + +# Get the latest Article. +>>> articles.get_latest() +Article 4 +""" diff --git a/tests/testapp/models/lookup.py b/tests/testapp/models/lookup.py new file mode 100644 index 0000000000..8eb0032036 --- /dev/null +++ b/tests/testapp/models/lookup.py @@ -0,0 +1,67 @@ +""" +7. The lookup API + +""" + +from django.core import meta + +class Article(meta.Model): + fields = ( + meta.CharField('headline', maxlength=100), + meta.DateTimeField('pub_date'), + ) + ordering = ('-pub_date', 'headline') + + def __repr__(self): + return self.headline + +API_TESTS = """ +# Create a couple of Articles. +>>> from datetime import datetime +>>> a1 = articles.Article(id=None, headline='Article 1', pub_date=datetime(2005, 7, 26)) +>>> a1.save() +>>> a2 = articles.Article(id=None, headline='Article 2', pub_date=datetime(2005, 7, 27)) +>>> a2.save() +>>> a3 = articles.Article(id=None, headline='Article 3', pub_date=datetime(2005, 7, 27)) +>>> a3.save() +>>> a4 = articles.Article(id=None, headline='Article 4', pub_date=datetime(2005, 7, 28)) +>>> a4.save() + +# get_iterator() is just like get_list(), but it's a generator. +>>> for a in articles.get_iterator(): +... print a.headline +Article 4 +Article 2 +Article 3 +Article 1 + +# get_iterator() takes the same lookup arguments as get_list(). +>>> for a in articles.get_iterator(headline__endswith='4'): +... print a.headline +Article 4 + +# get_count() returns the number of objects matching search criteria. +>>> articles.get_count() +4L +>>> articles.get_count(pub_date__exact=datetime(2005, 7, 27)) +2L +>>> articles.get_count(headline__startswith='Blah blah') +0L + +# get_in_bulk() takes a list of IDs and returns a dictionary mapping IDs +# to objects. +>>> articles.get_in_bulk([1, 2]) +{1: Article 1, 2: Article 2} +>>> articles.get_in_bulk([3]) +{3: Article 3} +>>> articles.get_in_bulk([1000]) +{} + +# Every DateField and DateTimeField creates get_next_by_FOO() and +# get_previous_by_FOO() methods. +>>> a3.get_next_by_pub_date() +Article 4 +>>> a2.get_previous_by_pub_date() +Article 1 + +"""