fix issues in test_all_related.py

piccolo-orm · Dec 23, 2023 · 18a87c0 · 18a87c0
1 parent 92c092b
commit 18a87c0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/columns/foreign_key/test_all_related.py b/tests/columns/foreign_key/test_all_related.py
@@ -38,13 +38,13 @@ def test_all_related_deep(self):
         """
         Make sure ``all_related`` works when the joins are several layers deep.
         """
-        all_related = Ticket.concert.band_1.all_related()
-        self.assertEqual(all_related, [Ticket.concert.band_1.manager])
+        all_related = Ticket.concert._.band_1.all_related()
+        self.assertEqual(all_related, [Ticket.concert._.band_1._.manager])
 
         # Make sure the call chains are also correct.
         self.assertEqual(
             all_related[0]._meta.call_chain,
-            Ticket.concert.band_1.manager._meta.call_chain,
+            Ticket.concert._.band_1._.manager._meta.call_chain,
         )
 
     def test_all_related_exclude(self):
@@ -57,6 +57,6 @@ def test_all_related_exclude(self):
         )
 
         self.assertEqual(
-            Ticket.concert.all_related(exclude=[Ticket.concert.venue]),
+            Ticket.concert.all_related(exclude=[Ticket.concert._.venue]),
             [Ticket.concert.band_1, Ticket.concert.band_2],
         )