summaryrefslogtreecommitdiff
path: root/tests/get_earliest_or_latest
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/get_earliest_or_latest
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
downloaddjango-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.py4
-rw-r--r--tests/get_earliest_or_latest/tests.py132
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()