summaryrefslogtreecommitdiff
path: root/tests/admin_ordering
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/admin_ordering
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
downloaddjango-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/admin_ordering')
-rw-r--r--tests/admin_ordering/models.py13
-rw-r--r--tests/admin_ordering/tests.py83
2 files changed, 57 insertions, 39 deletions
diff --git a/tests/admin_ordering/models.py b/tests/admin_ordering/models.py
index fbddeaa283..ef72dd43d9 100644
--- a/tests/admin_ordering/models.py
+++ b/tests/admin_ordering/models.py
@@ -8,17 +8,17 @@ class Band(models.Model):
rank = models.IntegerField()
class Meta:
- ordering = ('name',)
+ ordering = ("name",)
class Song(models.Model):
band = models.ForeignKey(Band, models.CASCADE)
name = models.CharField(max_length=100)
duration = models.IntegerField()
- other_interpreters = models.ManyToManyField(Band, related_name='covers')
+ other_interpreters = models.ManyToManyField(Band, related_name="covers")
class Meta:
- ordering = ('name',)
+ ordering = ("name",)
class SongInlineDefaultOrdering(admin.StackedInline):
@@ -27,13 +27,12 @@ class SongInlineDefaultOrdering(admin.StackedInline):
class SongInlineNewOrdering(admin.StackedInline):
model = Song
- ordering = ('duration',)
+ ordering = ("duration",)
class DynOrderingBandAdmin(admin.ModelAdmin):
-
def get_ordering(self, request):
if request.user.is_superuser:
- return ['rank']
+ return ["rank"]
else:
- return ['name']
+ return ["name"]
diff --git a/tests/admin_ordering/tests.py b/tests/admin_ordering/tests.py
index d47cf16d46..d79e747765 100644
--- a/tests/admin_ordering/tests.py
+++ b/tests/admin_ordering/tests.py
@@ -5,7 +5,10 @@ from django.db.models import F
from django.test import RequestFactory, TestCase
from .models import (
- Band, DynOrderingBandAdmin, Song, SongInlineDefaultOrdering,
+ Band,
+ DynOrderingBandAdmin,
+ Song,
+ SongInlineDefaultOrdering,
SongInlineNewOrdering,
)
@@ -34,15 +37,18 @@ class TestAdminOrdering(TestCase):
in ModelAdmin rather that ordering defined in the model's inner Meta
class.
"""
+
request_factory = RequestFactory()
@classmethod
def setUpTestData(cls):
- Band.objects.bulk_create([
- Band(name='Aerosmith', bio='', rank=3),
- Band(name='Radiohead', bio='', rank=1),
- Band(name='Van Halen', bio='', rank=2),
- ])
+ Band.objects.bulk_create(
+ [
+ Band(name="Aerosmith", bio="", rank=3),
+ Band(name="Radiohead", bio="", rank=1),
+ Band(name="Van Halen", bio="", rank=2),
+ ]
+ )
def test_default_ordering(self):
"""
@@ -51,40 +57,43 @@ class TestAdminOrdering(TestCase):
"""
ma = ModelAdmin(Band, site)
names = [b.name for b in ma.get_queryset(request)]
- self.assertEqual(['Aerosmith', 'Radiohead', 'Van Halen'], names)
+ self.assertEqual(["Aerosmith", "Radiohead", "Van Halen"], names)
def test_specified_ordering(self):
"""
Let's use a custom ModelAdmin that changes the ordering, and make sure
it actually changes.
"""
+
class BandAdmin(ModelAdmin):
- ordering = ('rank',) # default ordering is ('name',)
+ ordering = ("rank",) # default ordering is ('name',)
+
ma = BandAdmin(Band, site)
names = [b.name for b in ma.get_queryset(request)]
- self.assertEqual(['Radiohead', 'Van Halen', 'Aerosmith'], names)
+ self.assertEqual(["Radiohead", "Van Halen", "Aerosmith"], names)
def test_specified_ordering_by_f_expression(self):
class BandAdmin(ModelAdmin):
- ordering = (F('rank').desc(nulls_last=True),)
+ ordering = (F("rank").desc(nulls_last=True),)
+
band_admin = BandAdmin(Band, site)
names = [b.name for b in band_admin.get_queryset(request)]
- self.assertEqual(['Aerosmith', 'Van Halen', 'Radiohead'], names)
+ self.assertEqual(["Aerosmith", "Van Halen", "Radiohead"], names)
def test_dynamic_ordering(self):
"""
Let's use a custom ModelAdmin that changes the ordering dynamically.
"""
- super_user = User.objects.create(username='admin', is_superuser=True)
- other_user = User.objects.create(username='other')
- request = self.request_factory.get('/')
+ super_user = User.objects.create(username="admin", is_superuser=True)
+ other_user = User.objects.create(username="other")
+ request = self.request_factory.get("/")
request.user = super_user
ma = DynOrderingBandAdmin(Band, site)
names = [b.name for b in ma.get_queryset(request)]
- self.assertEqual(['Radiohead', 'Van Halen', 'Aerosmith'], names)
+ self.assertEqual(["Radiohead", "Van Halen", "Aerosmith"], names)
request.user = other_user
names = [b.name for b in ma.get_queryset(request)]
- self.assertEqual(['Aerosmith', 'Radiohead', 'Van Halen'], names)
+ self.assertEqual(["Aerosmith", "Radiohead", "Van Halen"], names)
class TestInlineModelAdminOrdering(TestCase):
@@ -95,12 +104,14 @@ class TestInlineModelAdminOrdering(TestCase):
@classmethod
def setUpTestData(cls):
- cls.band = Band.objects.create(name='Aerosmith', bio='', rank=3)
- Song.objects.bulk_create([
- Song(band=cls.band, name='Pink', duration=235),
- Song(band=cls.band, name='Dude (Looks Like a Lady)', duration=264),
- Song(band=cls.band, name='Jaded', duration=214),
- ])
+ cls.band = Band.objects.create(name="Aerosmith", bio="", rank=3)
+ Song.objects.bulk_create(
+ [
+ Song(band=cls.band, name="Pink", duration=235),
+ Song(band=cls.band, name="Dude (Looks Like a Lady)", duration=264),
+ Song(band=cls.band, name="Jaded", duration=214),
+ ]
+ )
def test_default_ordering(self):
"""
@@ -109,7 +120,7 @@ class TestInlineModelAdminOrdering(TestCase):
"""
inline = SongInlineDefaultOrdering(self.band, site)
names = [s.name for s in inline.get_queryset(request)]
- self.assertEqual(['Dude (Looks Like a Lady)', 'Jaded', 'Pink'], names)
+ self.assertEqual(["Dude (Looks Like a Lady)", "Jaded", "Pink"], names)
def test_specified_ordering(self):
"""
@@ -117,14 +128,14 @@ class TestInlineModelAdminOrdering(TestCase):
"""
inline = SongInlineNewOrdering(self.band, site)
names = [s.name for s in inline.get_queryset(request)]
- self.assertEqual(['Jaded', 'Pink', 'Dude (Looks Like a Lady)'], names)
+ self.assertEqual(["Jaded", "Pink", "Dude (Looks Like a Lady)"], names)
class TestRelatedFieldsAdminOrdering(TestCase):
@classmethod
def setUpTestData(cls):
- cls.b1 = Band.objects.create(name='Pink Floyd', bio='', rank=1)
- cls.b2 = Band.objects.create(name='Foo Fighters', bio='', rank=5)
+ cls.b1 = Band.objects.create(name="Pink Floyd", bio="", rank=1)
+ cls.b2 = Band.objects.create(name="Foo Fighters", bio="", rank=5)
def setUp(self):
# we need to register a custom ModelAdmin (instead of just using
@@ -132,6 +143,7 @@ class TestRelatedFieldsAdminOrdering(TestCase):
# for the related model
class SongAdmin(admin.ModelAdmin):
pass
+
site.register(Song, SongAdmin)
def tearDown(self):
@@ -140,8 +152,12 @@ class TestRelatedFieldsAdminOrdering(TestCase):
site.unregister(Band)
def check_ordering_of_field_choices(self, correct_ordering):
- fk_field = site._registry[Song].formfield_for_foreignkey(Song.band.field, request=None)
- m2m_field = site._registry[Song].formfield_for_manytomany(Song.other_interpreters.field, request=None)
+ fk_field = site._registry[Song].formfield_for_foreignkey(
+ Song.band.field, request=None
+ )
+ m2m_field = site._registry[Song].formfield_for_manytomany(
+ Song.other_interpreters.field, request=None
+ )
self.assertEqual(list(fk_field.queryset), correct_ordering)
self.assertEqual(list(m2m_field.queryset), correct_ordering)
@@ -152,6 +168,7 @@ class TestRelatedFieldsAdminOrdering(TestCase):
def test_admin_with_no_ordering_fallback_to_model_ordering(self):
class NoOrderingBandAdmin(admin.ModelAdmin):
pass
+
site.register(Band, NoOrderingBandAdmin)
# should be ordered by name (as defined by the model)
@@ -159,7 +176,8 @@ class TestRelatedFieldsAdminOrdering(TestCase):
def test_admin_ordering_beats_model_ordering(self):
class StaticOrderingBandAdmin(admin.ModelAdmin):
- ordering = ('rank',)
+ ordering = ("rank",)
+
site.register(Band, StaticOrderingBandAdmin)
# should be ordered by rank (defined by the ModelAdmin)
@@ -167,20 +185,21 @@ class TestRelatedFieldsAdminOrdering(TestCase):
def test_custom_queryset_still_wins(self):
"""Custom queryset has still precedence (#21405)"""
+
class SongAdmin(admin.ModelAdmin):
# Exclude one of the two Bands from the querysets
def formfield_for_foreignkey(self, db_field, request, **kwargs):
- if db_field.name == 'band':
+ if db_field.name == "band":
kwargs["queryset"] = Band.objects.filter(rank__gt=2)
return super().formfield_for_foreignkey(db_field, request, **kwargs)
def formfield_for_manytomany(self, db_field, request, **kwargs):
- if db_field.name == 'other_interpreters':
+ if db_field.name == "other_interpreters":
kwargs["queryset"] = Band.objects.filter(rank__gt=2)
return super().formfield_for_foreignkey(db_field, request, **kwargs)
class StaticOrderingBandAdmin(admin.ModelAdmin):
- ordering = ('rank',)
+ ordering = ("rank",)
site.unregister(Song)
site.register(Song, SongAdmin)