summaryrefslogtreecommitdiff
path: root/tests/foreign_object/tests.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-08 12:09:55 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-08 12:15:38 +0100
commit6a682b38e75d4c975b4c4493565a59f1bc14397c (patch)
tree0bd9cda550bea26238656d9f120d769e8b41bb9e /tests/foreign_object/tests.py
parente73ce08888e6f34d3f050377cfd2fbb733be94a9 (diff)
[4.0.x] Refs #33476 -- Reformatted code with Black.
Backport of 9c19aff7c7561e3a82978a272ecdaad40dda5c00 from main.
Diffstat (limited to 'tests/foreign_object/tests.py')
-rw-r--r--tests/foreign_object/tests.py446
1 files changed, 273 insertions, 173 deletions
diff --git a/tests/foreign_object/tests.py b/tests/foreign_object/tests.py
index 72d50cad6b..99206e837f 100644
--- a/tests/foreign_object/tests.py
+++ b/tests/foreign_object/tests.py
@@ -8,8 +8,16 @@ from django.test.utils import isolate_apps
from django.utils import translation
from .models import (
- Article, ArticleIdea, ArticleTag, ArticleTranslation, Country, Friendship,
- Group, Membership, NewsArticle, Person,
+ Article,
+ ArticleIdea,
+ ArticleTag,
+ ArticleTranslation,
+ Country,
+ Friendship,
+ Group,
+ Membership,
+ NewsArticle,
+ Person,
)
# Note that these tests are testing internal implementation details.
@@ -23,25 +31,28 @@ class MultiColumnFKTests(TestCase):
cls.usa = Country.objects.create(name="United States of America")
cls.soviet_union = Country.objects.create(name="Soviet Union")
# Creating People
- cls.bob = Person.objects.create(name='Bob', person_country=cls.usa)
- cls.jim = Person.objects.create(name='Jim', person_country=cls.usa)
- cls.george = Person.objects.create(name='George', person_country=cls.usa)
+ cls.bob = Person.objects.create(name="Bob", person_country=cls.usa)
+ cls.jim = Person.objects.create(name="Jim", person_country=cls.usa)
+ cls.george = Person.objects.create(name="George", person_country=cls.usa)
- cls.jane = Person.objects.create(name='Jane', person_country=cls.soviet_union)
- cls.mark = Person.objects.create(name='Mark', person_country=cls.soviet_union)
- cls.sam = Person.objects.create(name='Sam', person_country=cls.soviet_union)
+ cls.jane = Person.objects.create(name="Jane", person_country=cls.soviet_union)
+ cls.mark = Person.objects.create(name="Mark", person_country=cls.soviet_union)
+ cls.sam = Person.objects.create(name="Sam", person_country=cls.soviet_union)
# Creating Groups
- cls.kgb = Group.objects.create(name='KGB', group_country=cls.soviet_union)
- cls.cia = Group.objects.create(name='CIA', group_country=cls.usa)
- cls.republican = Group.objects.create(name='Republican', group_country=cls.usa)
- cls.democrat = Group.objects.create(name='Democrat', group_country=cls.usa)
+ cls.kgb = Group.objects.create(name="KGB", group_country=cls.soviet_union)
+ cls.cia = Group.objects.create(name="CIA", group_country=cls.usa)
+ cls.republican = Group.objects.create(name="Republican", group_country=cls.usa)
+ cls.democrat = Group.objects.create(name="Democrat", group_country=cls.usa)
def test_get_succeeds_on_multicolumn_match(self):
# Membership objects have access to their related Person if both
# country_ids match between them
membership = Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ )
person = membership.person
self.assertEqual((person.id, person.name), (self.bob.id, "Bob"))
@@ -50,20 +61,28 @@ class MultiColumnFKTests(TestCase):
# Membership objects returns DoesNotExist error when there is no
# Person with the same id and country_id
membership = Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.jane.id, group_id=self.cia.id)
+ membership_country_id=self.usa.id,
+ person_id=self.jane.id,
+ group_id=self.cia.id,
+ )
with self.assertRaises(Person.DoesNotExist):
- getattr(membership, 'person')
+ getattr(membership, "person")
def test_reverse_query_returns_correct_result(self):
# Creating a valid membership because it has the same country has the person
Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ )
# Creating an invalid membership because it has a different country has the person
Membership.objects.create(
- membership_country_id=self.soviet_union.id, person_id=self.bob.id,
- group_id=self.republican.id)
+ membership_country_id=self.soviet_union.id,
+ person_id=self.bob.id,
+ group_id=self.republican.id,
+ )
with self.assertNumQueries(1):
membership = self.bob.membership_set.get()
@@ -74,20 +93,27 @@ class MultiColumnFKTests(TestCase):
# Creating a to valid memberships
Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.bob.id, group_id=self.cia.id)
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ )
Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.jim.id,
- group_id=self.cia.id)
+ membership_country_id=self.usa.id,
+ person_id=self.jim.id,
+ group_id=self.cia.id,
+ )
# Creating an invalid membership
- Membership.objects.create(membership_country_id=self.soviet_union.id,
- person_id=self.george.id, group_id=self.cia.id)
+ Membership.objects.create(
+ membership_country_id=self.soviet_union.id,
+ person_id=self.george.id,
+ group_id=self.cia.id,
+ )
self.assertQuerysetEqual(
- Membership.objects.filter(person__name__contains='o'), [
- self.bob.id
- ],
- attrgetter("person_id")
+ Membership.objects.filter(person__name__contains="o"),
+ [self.bob.id],
+ attrgetter("person_id"),
)
def test_reverse_query_filters_correctly(self):
@@ -97,191 +123,228 @@ class MultiColumnFKTests(TestCase):
# Creating a to valid memberships
Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.bob.id,
- group_id=self.cia.id, date_joined=timemark - timedelta)
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ date_joined=timemark - timedelta,
+ )
Membership.objects.create(
- membership_country_id=self.usa.id, person_id=self.jim.id,
- group_id=self.cia.id, date_joined=timemark + timedelta)
+ membership_country_id=self.usa.id,
+ person_id=self.jim.id,
+ group_id=self.cia.id,
+ date_joined=timemark + timedelta,
+ )
# Creating an invalid membership
Membership.objects.create(
- membership_country_id=self.soviet_union.id, person_id=self.george.id,
- group_id=self.cia.id, date_joined=timemark + timedelta)
+ membership_country_id=self.soviet_union.id,
+ person_id=self.george.id,
+ group_id=self.cia.id,
+ date_joined=timemark + timedelta,
+ )
self.assertQuerysetEqual(
- Person.objects.filter(membership__date_joined__gte=timemark), [
- 'Jim'
- ],
- attrgetter('name')
+ Person.objects.filter(membership__date_joined__gte=timemark),
+ ["Jim"],
+ attrgetter("name"),
)
def test_forward_in_lookup_filters_correctly(self):
- Membership.objects.create(membership_country_id=self.usa.id, person_id=self.bob.id,
- group_id=self.cia.id)
- Membership.objects.create(membership_country_id=self.usa.id, person_id=self.jim.id,
- group_id=self.cia.id)
+ Membership.objects.create(
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ )
+ Membership.objects.create(
+ membership_country_id=self.usa.id,
+ person_id=self.jim.id,
+ group_id=self.cia.id,
+ )
# Creating an invalid membership
Membership.objects.create(
- membership_country_id=self.soviet_union.id, person_id=self.george.id,
- group_id=self.cia.id)
+ membership_country_id=self.soviet_union.id,
+ person_id=self.george.id,
+ group_id=self.cia.id,
+ )
self.assertQuerysetEqual(
- Membership.objects.filter(person__in=[self.george, self.jim]), [
+ Membership.objects.filter(person__in=[self.george, self.jim]),
+ [
self.jim.id,
],
- attrgetter('person_id')
+ attrgetter("person_id"),
)
self.assertQuerysetEqual(
- Membership.objects.filter(person__in=Person.objects.filter(name='Jim')), [
+ Membership.objects.filter(person__in=Person.objects.filter(name="Jim")),
+ [
self.jim.id,
],
- attrgetter('person_id')
+ attrgetter("person_id"),
)
def test_double_nested_query(self):
- m1 = Membership.objects.create(membership_country_id=self.usa.id, person_id=self.bob.id,
- group_id=self.cia.id)
- m2 = Membership.objects.create(membership_country_id=self.usa.id, person_id=self.jim.id,
- group_id=self.cia.id)
- Friendship.objects.create(from_friend_country_id=self.usa.id, from_friend_id=self.bob.id,
- to_friend_country_id=self.usa.id, to_friend_id=self.jim.id)
+ m1 = Membership.objects.create(
+ membership_country_id=self.usa.id,
+ person_id=self.bob.id,
+ group_id=self.cia.id,
+ )
+ m2 = Membership.objects.create(
+ membership_country_id=self.usa.id,
+ person_id=self.jim.id,
+ group_id=self.cia.id,
+ )
+ Friendship.objects.create(
+ from_friend_country_id=self.usa.id,
+ from_friend_id=self.bob.id,
+ to_friend_country_id=self.usa.id,
+ to_friend_id=self.jim.id,
+ )
self.assertSequenceEqual(
Membership.objects.filter(
person__in=Person.objects.filter(
- from_friend__in=Friendship.objects.filter(to_friend__in=Person.objects.all())
+ from_friend__in=Friendship.objects.filter(
+ to_friend__in=Person.objects.all()
+ )
)
),
- [m1]
+ [m1],
)
self.assertSequenceEqual(
Membership.objects.exclude(
person__in=Person.objects.filter(
- from_friend__in=Friendship.objects.filter(to_friend__in=Person.objects.all())
+ from_friend__in=Friendship.objects.filter(
+ to_friend__in=Person.objects.all()
+ )
)
),
- [m2]
+ [m2],
)
def test_select_related_foreignkey_forward_works(self):
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.democrat
+ )
with self.assertNumQueries(1):
- people = [m.person for m in Membership.objects.select_related('person').order_by('pk')]
+ people = [
+ m.person
+ for m in Membership.objects.select_related("person").order_by("pk")
+ ]
- normal_people = [m.person for m in Membership.objects.all().order_by('pk')]
+ normal_people = [m.person for m in Membership.objects.all().order_by("pk")]
self.assertEqual(people, normal_people)
def test_prefetch_foreignkey_forward_works(self):
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.democrat
+ )
with self.assertNumQueries(2):
people = [
- m.person for m in Membership.objects.prefetch_related('person').order_by('pk')]
+ m.person
+ for m in Membership.objects.prefetch_related("person").order_by("pk")
+ ]
- normal_people = [m.person for m in Membership.objects.order_by('pk')]
+ normal_people = [m.person for m in Membership.objects.order_by("pk")]
self.assertEqual(people, normal_people)
def test_prefetch_foreignkey_reverse_works(self):
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.democrat
+ )
with self.assertNumQueries(2):
membership_sets = [
list(p.membership_set.all())
- for p in Person.objects.prefetch_related('membership_set').order_by('pk')]
+ for p in Person.objects.prefetch_related("membership_set").order_by(
+ "pk"
+ )
+ ]
with self.assertNumQueries(7):
normal_membership_sets = [
- list(p.membership_set.all())
- for p in Person.objects.order_by('pk')
+ list(p.membership_set.all()) for p in Person.objects.order_by("pk")
]
self.assertEqual(membership_sets, normal_membership_sets)
def test_m2m_through_forward_returns_valid_members(self):
# We start out by making sure that the Group 'CIA' has no members.
- self.assertQuerysetEqual(
- self.cia.members.all(),
- []
- )
+ self.assertQuerysetEqual(self.cia.members.all(), [])
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.cia)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.cia
+ )
# Let's check to make sure that it worked. Bob and Jim should be members of the CIA.
self.assertQuerysetEqual(
- self.cia.members.all(), [
- 'Bob',
- 'Jim'
- ], attrgetter("name")
+ self.cia.members.all(), ["Bob", "Jim"], attrgetter("name")
)
def test_m2m_through_reverse_returns_valid_members(self):
# We start out by making sure that Bob is in no groups.
- self.assertQuerysetEqual(
- self.bob.groups.all(),
- []
- )
+ self.assertQuerysetEqual(self.bob.groups.all(), [])
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.bob,
- group=self.republican)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.republican
+ )
# Bob should be in the CIA and a Republican
self.assertQuerysetEqual(
- self.bob.groups.all(), [
- 'CIA',
- 'Republican'
- ], attrgetter("name")
+ self.bob.groups.all(), ["CIA", "Republican"], attrgetter("name")
)
def test_m2m_through_forward_ignores_invalid_members(self):
# We start out by making sure that the Group 'CIA' has no members.
- self.assertQuerysetEqual(
- self.cia.members.all(),
- []
- )
+ self.assertQuerysetEqual(self.cia.members.all(), [])
# Something adds jane to group CIA but Jane is in Soviet Union which isn't CIA's country
- Membership.objects.create(membership_country=self.usa, person=self.jane, group=self.cia)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jane, group=self.cia
+ )
# There should still be no members in CIA
- self.assertQuerysetEqual(
- self.cia.members.all(),
- []
- )
+ self.assertQuerysetEqual(self.cia.members.all(), [])
def test_m2m_through_reverse_ignores_invalid_members(self):
# We start out by making sure that Jane has no groups.
- self.assertQuerysetEqual(
- self.jane.groups.all(),
- []
- )
+ self.assertQuerysetEqual(self.jane.groups.all(), [])
# Something adds jane to group CIA but Jane is in Soviet Union which isn't CIA's country
- Membership.objects.create(membership_country=self.usa, person=self.jane, group=self.cia)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jane, group=self.cia
+ )
# Jane should still not be in any groups
- self.assertQuerysetEqual(
- self.jane.groups.all(),
- []
- )
+ self.assertQuerysetEqual(self.jane.groups.all(), [])
def test_m2m_through_on_self_works(self):
- self.assertQuerysetEqual(
- self.jane.friends.all(),
- []
- )
+ self.assertQuerysetEqual(self.jane.friends.all(), [])
Friendship.objects.create(
- from_friend_country=self.jane.person_country, from_friend=self.jane,
- to_friend_country=self.george.person_country, to_friend=self.george)
+ from_friend_country=self.jane.person_country,
+ from_friend=self.jane,
+ to_friend_country=self.george.person_country,
+ to_friend=self.george,
+ )
self.assertQuerysetEqual(
- self.jane.friends.all(),
- ['George'], attrgetter("name")
+ self.jane.friends.all(), ["George"], attrgetter("name")
)
def test_m2m_through_on_self_ignores_mismatch_columns(self):
@@ -290,74 +353,102 @@ class MultiColumnFKTests(TestCase):
# Note that we use ids instead of instances. This is because instances on ForeignObject
# properties will set all related field off of the given instance
Friendship.objects.create(
- from_friend_id=self.jane.id, to_friend_id=self.george.id,
+ from_friend_id=self.jane.id,
+ to_friend_id=self.george.id,
to_friend_country_id=self.jane.person_country_id,
- from_friend_country_id=self.george.person_country_id)
+ from_friend_country_id=self.george.person_country_id,
+ )
self.assertQuerysetEqual(self.jane.friends.all(), [])
def test_prefetch_related_m2m_forward_works(self):
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.democrat
+ )
with self.assertNumQueries(2):
- members_lists = [list(g.members.all())
- for g in Group.objects.prefetch_related('members')]
+ members_lists = [
+ list(g.members.all()) for g in Group.objects.prefetch_related("members")
+ ]
normal_members_lists = [list(g.members.all()) for g in Group.objects.all()]
self.assertEqual(members_lists, normal_members_lists)
def test_prefetch_related_m2m_reverse_works(self):
- Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
- Membership.objects.create(membership_country=self.usa, person=self.jim, group=self.democrat)
+ Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
+ Membership.objects.create(
+ membership_country=self.usa, person=self.jim, group=self.democrat
+ )
with self.assertNumQueries(2):
- groups_lists = [list(p.groups.all()) for p in Person.objects.prefetch_related('groups')]
+ groups_lists = [
+ list(p.groups.all()) for p in Person.objects.prefetch_related("groups")
+ ]
normal_groups_lists = [list(p.groups.all()) for p in Person.objects.all()]
self.assertEqual(groups_lists, normal_groups_lists)
- @translation.override('fi')
+ @translation.override("fi")
def test_translations(self):
a1 = Article.objects.create(pub_date=datetime.date.today())
- at1_fi = ArticleTranslation(article=a1, lang='fi', title='Otsikko', body='Diipadaapa')
+ at1_fi = ArticleTranslation(
+ article=a1, lang="fi", title="Otsikko", body="Diipadaapa"
+ )
at1_fi.save()
- at2_en = ArticleTranslation(article=a1, lang='en', title='Title', body='Lalalalala')
+ at2_en = ArticleTranslation(
+ article=a1, lang="en", title="Title", body="Lalalalala"
+ )
at2_en.save()
self.assertEqual(Article.objects.get(pk=a1.pk).active_translation, at1_fi)
with self.assertNumQueries(1):
- fetched = Article.objects.select_related('active_translation').get(
- active_translation__title='Otsikko')
- self.assertEqual(fetched.active_translation.title, 'Otsikko')
+ fetched = Article.objects.select_related("active_translation").get(
+ active_translation__title="Otsikko"
+ )
+ self.assertEqual(fetched.active_translation.title, "Otsikko")
a2 = Article.objects.create(pub_date=datetime.date.today())
- at2_fi = ArticleTranslation(article=a2, lang='fi', title='Atsikko', body='Diipadaapa',
- abstract='dipad')
+ at2_fi = ArticleTranslation(
+ article=a2, lang="fi", title="Atsikko", body="Diipadaapa", abstract="dipad"
+ )
at2_fi.save()
a3 = Article.objects.create(pub_date=datetime.date.today())
- at3_en = ArticleTranslation(article=a3, lang='en', title='A title', body='lalalalala',
- abstract='lala')
+ at3_en = ArticleTranslation(
+ article=a3, lang="en", title="A title", body="lalalalala", abstract="lala"
+ )
at3_en.save()
# Test model initialization with active_translation field.
a3 = Article(id=a3.id, pub_date=a3.pub_date, active_translation=at3_en)
a3.save()
self.assertEqual(
- list(Article.objects.filter(active_translation__abstract=None)),
- [a1, a3])
+ list(Article.objects.filter(active_translation__abstract=None)), [a1, a3]
+ )
self.assertEqual(
- list(Article.objects.filter(active_translation__abstract=None,
- active_translation__pk__isnull=False)),
- [a1])
+ list(
+ Article.objects.filter(
+ active_translation__abstract=None,
+ active_translation__pk__isnull=False,
+ )
+ ),
+ [a1],
+ )
- with translation.override('en'):
+ with translation.override("en"):
self.assertEqual(
list(Article.objects.filter(active_translation__abstract=None)),
- [a1, a2])
+ [a1, a2],
+ )
def test_foreign_key_raises_informative_does_not_exist(self):
referrer = ArticleTranslation()
- with self.assertRaisesMessage(Article.DoesNotExist, 'ArticleTranslation has no article'):
+ with self.assertRaisesMessage(
+ Article.DoesNotExist, "ArticleTranslation has no article"
+ ):
referrer.article
def test_foreign_key_related_query_name(self):
@@ -387,29 +478,37 @@ class MultiColumnFKTests(TestCase):
with self.assertRaisesMessage(FieldError, msg):
Article.objects.filter(ideas__name="idea1")
- @translation.override('fi')
+ @translation.override("fi")
def test_inheritance(self):
na = NewsArticle.objects.create(pub_date=datetime.date.today())
ArticleTranslation.objects.create(
- article=na, lang="fi", title="foo", body="bar")
+ article=na, lang="fi", title="foo", body="bar"
+ )
self.assertSequenceEqual(
- NewsArticle.objects.select_related('active_translation'),
- [na]
+ NewsArticle.objects.select_related("active_translation"), [na]
)
with self.assertNumQueries(1):
self.assertEqual(
- NewsArticle.objects.select_related(
- 'active_translation')[0].active_translation.title,
- "foo")
+ NewsArticle.objects.select_related("active_translation")[
+ 0
+ ].active_translation.title,
+ "foo",
+ )
- @skipUnlessDBFeature('has_bulk_insert')
+ @skipUnlessDBFeature("has_bulk_insert")
def test_batch_create_foreign_object(self):
- objs = [Person(name="abcd_%s" % i, person_country=self.usa) for i in range(0, 5)]
+ objs = [
+ Person(name="abcd_%s" % i, person_country=self.usa) for i in range(0, 5)
+ ]
Person.objects.bulk_create(objs, 10)
def test_isnull_lookup(self):
- m1 = Membership.objects.create(membership_country=self.usa, person=self.bob, group_id=None)
- m2 = Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
+ m1 = Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group_id=None
+ )
+ m2 = Membership.objects.create(
+ membership_country=self.usa, person=self.bob, group=self.cia
+ )
self.assertSequenceEqual(
Membership.objects.filter(group__isnull=True),
[m1],
@@ -421,15 +520,14 @@ class MultiColumnFKTests(TestCase):
class TestModelCheckTests(SimpleTestCase):
-
- @isolate_apps('foreign_object')
+ @isolate_apps("foreign_object")
def test_check_composite_foreign_object(self):
class Parent(models.Model):
a = models.PositiveIntegerField()
b = models.PositiveIntegerField()
class Meta:
- unique_together = (('a', 'b'),)
+ unique_together = (("a", "b"),)
class Child(models.Model):
a = models.PositiveIntegerField()
@@ -438,14 +536,14 @@ class TestModelCheckTests(SimpleTestCase):
parent = models.ForeignObject(
Parent,
on_delete=models.SET_NULL,
- from_fields=('a', 'b'),
- to_fields=('a', 'b'),
- related_name='children',
+ from_fields=("a", "b"),
+ to_fields=("a", "b"),
+ related_name="children",
)
- self.assertEqual(Child._meta.get_field('parent').check(from_model=Child), [])
+ self.assertEqual(Child._meta.get_field("parent").check(from_model=Child), [])
- @isolate_apps('foreign_object')
+ @isolate_apps("foreign_object")
def test_check_subset_composite_foreign_object(self):
class Parent(models.Model):
a = models.PositiveIntegerField()
@@ -453,7 +551,7 @@ class TestModelCheckTests(SimpleTestCase):
c = models.PositiveIntegerField()
class Meta:
- unique_together = (('a', 'b'),)
+ unique_together = (("a", "b"),)
class Child(models.Model):
a = models.PositiveIntegerField()
@@ -463,22 +561,24 @@ class TestModelCheckTests(SimpleTestCase):
parent = models.ForeignObject(
Parent,
on_delete=models.SET_NULL,
- from_fields=('a', 'b', 'c'),
- to_fields=('a', 'b', 'c'),
- related_name='children',
+ from_fields=("a", "b", "c"),
+ to_fields=("a", "b", "c"),
+ related_name="children",
)
- self.assertEqual(Child._meta.get_field('parent').check(from_model=Child), [])
+ self.assertEqual(Child._meta.get_field("parent").check(from_model=Child), [])
class TestExtraJoinFilterQ(TestCase):
- @translation.override('fi')
+ @translation.override("fi")
def test_extra_join_filter_q(self):
a = Article.objects.create(pub_date=datetime.datetime.today())
- ArticleTranslation.objects.create(article=a, lang='fi', title='title', body='body')
+ ArticleTranslation.objects.create(
+ article=a, lang="fi", title="title", body="body"
+ )
qs = Article.objects.all()
with self.assertNumQueries(2):
- self.assertEqual(qs[0].active_translation_q.title, 'title')
- qs = qs.select_related('active_translation_q')
+ self.assertEqual(qs[0].active_translation_q.title, "title")
+ qs = qs.select_related("active_translation_q")
with self.assertNumQueries(1):
- self.assertEqual(qs[0].active_translation_q.title, 'title')
+ self.assertEqual(qs[0].active_translation_q.title, "title")