diff options
Diffstat (limited to 'parts/django/tests/modeltests/or_lookups/tests.py')
| -rw-r--r-- | parts/django/tests/modeltests/or_lookups/tests.py | 232 |
1 files changed, 0 insertions, 232 deletions
diff --git a/parts/django/tests/modeltests/or_lookups/tests.py b/parts/django/tests/modeltests/or_lookups/tests.py deleted file mode 100644 index ad218cd..0000000 --- a/parts/django/tests/modeltests/or_lookups/tests.py +++ /dev/null @@ -1,232 +0,0 @@ -from datetime import datetime -from operator import attrgetter - -from django.db.models import Q -from django.test import TestCase - -from models import Article - - -class OrLookupsTests(TestCase): - - def setUp(self): - self.a1 = Article.objects.create( - headline='Hello', pub_date=datetime(2005, 11, 27) - ).pk - self.a2 = Article.objects.create( - headline='Goodbye', pub_date=datetime(2005, 11, 28) - ).pk - self.a3 = Article.objects.create( - headline='Hello and goodbye', pub_date=datetime(2005, 11, 29) - ).pk - - def test_filter_or(self): - self.assertQuerysetEqual( - Article.objects.filter(headline__startswith='Hello') | Article.objects.filter(headline__startswith='Goodbye'), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - self.assertQuerysetEqual( - Article.objects.filter(headline__contains='Hello') | Article.objects.filter(headline__contains='bye'), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - self.assertQuerysetEqual( - Article.objects.filter(headline__iexact='Hello') | Article.objects.filter(headline__contains='ood'), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello') | Q(headline__startswith='Goodbye')), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - - def test_stages(self): - # You can shorten this syntax with code like the following, which is - # especially useful if building the query in stages: - articles = Article.objects.all() - self.assertQuerysetEqual( - articles.filter(headline__startswith='Hello') & articles.filter(headline__startswith='Goodbye'), - [] - ) - self.assertQuerysetEqual( - articles.filter(headline__startswith='Hello') & articles.filter(headline__contains='bye'), [ - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - def test_pk_q(self): - self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2)), [ - 'Hello', - 'Goodbye' - ], - attrgetter("headline") - ) - - self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2) | Q(pk=self.a3)), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - def test_pk_in(self): - self.assertQuerysetEqual( - Article.objects.filter(pk__in=[self.a1, self.a2, self.a3]), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - self.assertQuerysetEqual( - Article.objects.filter(pk__in=(self.a1, self.a2, self.a3)), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - self.assertQuerysetEqual( - Article.objects.filter(pk__in=[self.a1, self.a2, self.a3, 40000]), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - def test_q_negated(self): - # Q objects can be negated - self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | ~Q(pk=self.a2)), [ - 'Hello', - 'Hello and goodbye' - ], - attrgetter("headline") - ) - - self.assertQuerysetEqual( - Article.objects.filter(~Q(pk=self.a1) & ~Q(pk=self.a2)), [ - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - # This allows for more complex queries than filter() and exclude() - # alone would allow - self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) & (~Q(pk=self.a2) | Q(pk=self.a3))), [ - 'Hello' - ], - attrgetter("headline"), - ) - - def test_complex_filter(self): - # The 'complex_filter' method supports framework features such as - # 'limit_choices_to' which normally take a single dictionary of lookup - # arguments but need to support arbitrary queries via Q objects too. - self.assertQuerysetEqual( - Article.objects.complex_filter({'pk': self.a1}), [ - 'Hello' - ], - attrgetter("headline"), - ) - - self.assertQuerysetEqual( - Article.objects.complex_filter(Q(pk=self.a1) | Q(pk=self.a2)), [ - 'Hello', - 'Goodbye' - ], - attrgetter("headline"), - ) - - def test_empty_in(self): - # Passing "in" an empty list returns no results ... - self.assertQuerysetEqual( - Article.objects.filter(pk__in=[]), - [] - ) - # ... but can return results if we OR it with another query. - self.assertQuerysetEqual( - Article.objects.filter(Q(pk__in=[]) | Q(headline__icontains='goodbye')), [ - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - def test_q_and(self): - # Q arg objects are ANDed - self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello'), Q(headline__contains='bye')), [ - 'Hello and goodbye' - ], - attrgetter("headline") - ) - # Q arg AND order is irrelevant - self.assertQuerysetEqual( - Article.objects.filter(Q(headline__contains='bye'), headline__startswith='Hello'), [ - 'Hello and goodbye' - ], - attrgetter("headline"), - ) - - self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello') & Q(headline__startswith='Goodbye')), - [] - ) - - def test_q_exclude(self): - self.assertQuerysetEqual( - Article.objects.exclude(Q(headline__startswith='Hello')), [ - 'Goodbye' - ], - attrgetter("headline") - ) - - def test_other_arg_queries(self): - # Try some arg queries with operations other than filter. - self.assertEqual( - Article.objects.get(Q(headline__startswith='Hello'), Q(headline__contains='bye')).headline, - 'Hello and goodbye' - ) - - self.assertEqual( - Article.objects.filter(Q(headline__startswith='Hello') | Q(headline__contains='bye')).count(), - 3 - ) - - self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello'), Q(headline__contains='bye')).values(), [ - {"headline": "Hello and goodbye", "id": self.a3, "pub_date": datetime(2005, 11, 29)}, - ], - lambda o: o, - ) - - self.assertEqual( - Article.objects.filter(Q(headline__startswith='Hello')).in_bulk([self.a1, self.a2]), - {self.a1: Article.objects.get(pk=self.a1)} - ) |