diff options
author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
---|---|---|
committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/get_earliest_or_latest | |
parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
download | django-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/get_earliest_or_latest')
-rw-r--r-- | tests/get_earliest_or_latest/models.py | 4 | ||||
-rw-r--r-- | tests/get_earliest_or_latest/tests.py | 132 |
2 files changed, 87 insertions, 49 deletions
diff --git a/tests/get_earliest_or_latest/models.py b/tests/get_earliest_or_latest/models.py index 55096f5a2c..936011ab6b 100644 --- a/tests/get_earliest_or_latest/models.py +++ b/tests/get_earliest_or_latest/models.py @@ -7,7 +7,7 @@ class Article(models.Model): expire_date = models.DateField() class Meta: - get_latest_by = 'pub_date' + get_latest_by = "pub_date" class Person(models.Model): @@ -18,11 +18,13 @@ class Person(models.Model): # Ticket #23555 - model with an intentionally broken QuerySet.__iter__ method. + class IndexErrorQuerySet(models.QuerySet): """ Emulates the case when some internal code raises an unexpected IndexError. """ + def __iter__(self): raise IndexError diff --git a/tests/get_earliest_or_latest/tests.py b/tests/get_earliest_or_latest/tests.py index 0234deb195..59ee21aef7 100644 --- a/tests/get_earliest_or_latest/tests.py +++ b/tests/get_earliest_or_latest/tests.py @@ -22,39 +22,46 @@ class EarliestOrLatestTests(TestCase): Article.objects.earliest() a1 = Article.objects.create( - headline="Article 1", pub_date=datetime(2005, 7, 26), - expire_date=datetime(2005, 9, 1) + headline="Article 1", + pub_date=datetime(2005, 7, 26), + expire_date=datetime(2005, 9, 1), ) a2 = Article.objects.create( - headline="Article 2", pub_date=datetime(2005, 7, 27), - expire_date=datetime(2005, 7, 28) + headline="Article 2", + pub_date=datetime(2005, 7, 27), + expire_date=datetime(2005, 7, 28), ) a3 = Article.objects.create( - headline="Article 3", pub_date=datetime(2005, 7, 28), - expire_date=datetime(2005, 8, 27) + headline="Article 3", + pub_date=datetime(2005, 7, 28), + expire_date=datetime(2005, 8, 27), ) a4 = Article.objects.create( - headline="Article 4", pub_date=datetime(2005, 7, 28), - expire_date=datetime(2005, 7, 30) + headline="Article 4", + pub_date=datetime(2005, 7, 28), + expire_date=datetime(2005, 7, 30), ) # Get the earliest Article. self.assertEqual(Article.objects.earliest(), a1) # Get the earliest Article that matches certain filters. self.assertEqual( - Article.objects.filter(pub_date__gt=datetime(2005, 7, 26)).earliest(), - a2 + Article.objects.filter(pub_date__gt=datetime(2005, 7, 26)).earliest(), a2 ) # Pass a custom field name to earliest() to change the field that's used # to determine the earliest object. - self.assertEqual(Article.objects.earliest('expire_date'), a2) - self.assertEqual(Article.objects.filter( - pub_date__gt=datetime(2005, 7, 26)).earliest('expire_date'), a2) + self.assertEqual(Article.objects.earliest("expire_date"), a2) + self.assertEqual( + Article.objects.filter(pub_date__gt=datetime(2005, 7, 26)).earliest( + "expire_date" + ), + a2, + ) # earliest() overrides any other ordering specified on the query. # Refs #11283. - self.assertEqual(Article.objects.order_by('id').earliest(), a1) + self.assertEqual(Article.objects.order_by("id").earliest(), a1) # Error is raised if the user forgot to add a get_latest_by # in the Model.Meta @@ -62,27 +69,33 @@ class EarliestOrLatestTests(TestCase): with self.assertRaisesMessage( ValueError, "earliest() and latest() require either fields as positional " - "arguments or 'get_latest_by' in the model's Meta." + "arguments or 'get_latest_by' in the model's Meta.", ): Article.objects.earliest() # Earliest publication date, earliest expire date. self.assertEqual( - Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest('pub_date', 'expire_date'), + Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest( + "pub_date", "expire_date" + ), a4, ) # Earliest publication date, latest expire date. self.assertEqual( - Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest('pub_date', '-expire_date'), + Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest( + "pub_date", "-expire_date" + ), a3, ) # Meta.get_latest_by may be a tuple. - Article.objects.model._meta.get_latest_by = ('pub_date', 'expire_date') - self.assertEqual(Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest(), a4) + Article.objects.model._meta.get_latest_by = ("pub_date", "expire_date") + self.assertEqual( + Article.objects.filter(pub_date=datetime(2005, 7, 28)).earliest(), a4 + ) def test_earliest_sliced_queryset(self): - msg = 'Cannot change a query once a slice has been taken.' + msg = "Cannot change a query once a slice has been taken." with self.assertRaisesMessage(TypeError, msg): Article.objects.all()[0:5].earliest() @@ -92,64 +105,78 @@ class EarliestOrLatestTests(TestCase): Article.objects.latest() a1 = Article.objects.create( - headline="Article 1", pub_date=datetime(2005, 7, 26), - expire_date=datetime(2005, 9, 1) + headline="Article 1", + pub_date=datetime(2005, 7, 26), + expire_date=datetime(2005, 9, 1), ) a2 = Article.objects.create( - headline="Article 2", pub_date=datetime(2005, 7, 27), - expire_date=datetime(2005, 7, 28) + headline="Article 2", + pub_date=datetime(2005, 7, 27), + expire_date=datetime(2005, 7, 28), ) a3 = Article.objects.create( - headline="Article 3", pub_date=datetime(2005, 7, 27), - expire_date=datetime(2005, 8, 27) + headline="Article 3", + pub_date=datetime(2005, 7, 27), + expire_date=datetime(2005, 8, 27), ) a4 = Article.objects.create( - headline="Article 4", pub_date=datetime(2005, 7, 28), - expire_date=datetime(2005, 7, 30) + headline="Article 4", + pub_date=datetime(2005, 7, 28), + expire_date=datetime(2005, 7, 30), ) # Get the latest Article. self.assertEqual(Article.objects.latest(), a4) # Get the latest Article that matches certain filters. self.assertEqual( - Article.objects.filter(pub_date__lt=datetime(2005, 7, 27)).latest(), - a1 + Article.objects.filter(pub_date__lt=datetime(2005, 7, 27)).latest(), a1 ) # Pass a custom field name to latest() to change the field that's used # to determine the latest object. - self.assertEqual(Article.objects.latest('expire_date'), a1) + self.assertEqual(Article.objects.latest("expire_date"), a1) self.assertEqual( - Article.objects.filter(pub_date__gt=datetime(2005, 7, 26)).latest('expire_date'), + Article.objects.filter(pub_date__gt=datetime(2005, 7, 26)).latest( + "expire_date" + ), a3, ) # latest() overrides any other ordering specified on the query (#11283). - self.assertEqual(Article.objects.order_by('id').latest(), a4) + self.assertEqual(Article.objects.order_by("id").latest(), a4) # Error is raised if get_latest_by isn't in Model.Meta. Article.objects.model._meta.get_latest_by = None with self.assertRaisesMessage( ValueError, "earliest() and latest() require either fields as positional " - "arguments or 'get_latest_by' in the model's Meta." + "arguments or 'get_latest_by' in the model's Meta.", ): Article.objects.latest() # Latest publication date, latest expire date. - self.assertEqual(Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest('pub_date', 'expire_date'), a3) + self.assertEqual( + Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest( + "pub_date", "expire_date" + ), + a3, + ) # Latest publication date, earliest expire date. self.assertEqual( - Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest('pub_date', '-expire_date'), + Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest( + "pub_date", "-expire_date" + ), a2, ) # Meta.get_latest_by may be a tuple. - Article.objects.model._meta.get_latest_by = ('pub_date', 'expire_date') - self.assertEqual(Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest(), a3) + Article.objects.model._meta.get_latest_by = ("pub_date", "expire_date") + self.assertEqual( + Article.objects.filter(pub_date=datetime(2005, 7, 27)).latest(), a3 + ) def test_latest_sliced_queryset(self): - msg = 'Cannot change a query once a slice has been taken.' + msg = "Cannot change a query once a slice has been taken." with self.assertRaisesMessage(TypeError, msg): Article.objects.all()[0:5].latest() @@ -168,29 +195,37 @@ class EarliestOrLatestTests(TestCase): class TestFirstLast(TestCase): - def test_first(self): p1 = Person.objects.create(name="Bob", birthday=datetime(1950, 1, 1)) p2 = Person.objects.create(name="Alice", birthday=datetime(1961, 2, 3)) self.assertEqual(Person.objects.first(), p1) - self.assertEqual(Person.objects.order_by('name').first(), p2) - self.assertEqual(Person.objects.filter(birthday__lte=datetime(1955, 1, 1)).first(), p1) - self.assertIsNone(Person.objects.filter(birthday__lte=datetime(1940, 1, 1)).first()) + self.assertEqual(Person.objects.order_by("name").first(), p2) + self.assertEqual( + Person.objects.filter(birthday__lte=datetime(1955, 1, 1)).first(), p1 + ) + self.assertIsNone( + Person.objects.filter(birthday__lte=datetime(1940, 1, 1)).first() + ) def test_last(self): p1 = Person.objects.create(name="Alice", birthday=datetime(1950, 1, 1)) p2 = Person.objects.create(name="Bob", birthday=datetime(1960, 2, 3)) # Note: by default PK ordering. self.assertEqual(Person.objects.last(), p2) - self.assertEqual(Person.objects.order_by('-name').last(), p1) - self.assertEqual(Person.objects.filter(birthday__lte=datetime(1955, 1, 1)).last(), p1) - self.assertIsNone(Person.objects.filter(birthday__lte=datetime(1940, 1, 1)).last()) + self.assertEqual(Person.objects.order_by("-name").last(), p1) + self.assertEqual( + Person.objects.filter(birthday__lte=datetime(1955, 1, 1)).last(), p1 + ) + self.assertIsNone( + Person.objects.filter(birthday__lte=datetime(1940, 1, 1)).last() + ) def test_index_error_not_suppressed(self): """ #23555 -- Unexpected IndexError exceptions in QuerySet iteration shouldn't be suppressed. """ + def check(): # We know that we've broken the __iter__ method, so the queryset # should always raise an exception. @@ -205,7 +240,8 @@ class TestFirstLast(TestCase): # And it does not matter if there are any records in the DB. IndexErrorArticle.objects.create( - headline="Article 1", pub_date=datetime(2005, 7, 26), - expire_date=datetime(2005, 9, 1) + headline="Article 1", + pub_date=datetime(2005, 7, 26), + expire_date=datetime(2005, 9, 1), ) check() |