diff options
Diffstat (limited to 'tests/defer/tests.py')
| -rw-r--r-- | tests/defer/tests.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/defer/tests.py b/tests/defer/tests.py index 1a5f333fa1..ea7703c1e3 100644 --- a/tests/defer/tests.py +++ b/tests/defer/tests.py @@ -219,6 +219,20 @@ class DeferTests(AssertionMixin, TestCase): with self.assertNumQueries(1): p1.value + def test_defer_fk_fetch_mode_fetch_peers(self): + p1, p2 = Primary.objects.fetch_mode(FETCH_PEERS).defer("related") + with self.assertNumQueries(2): + self.assertEqual(p1.related, self.s1) + with self.assertNumQueries(0): + self.assertEqual(p2.related, self.s1) + + def test_only_fk_fetch_mode_fetch_peers(self): + p1, p2 = Primary.objects.fetch_mode(FETCH_PEERS).only("name") + with self.assertNumQueries(2): + self.assertEqual(p1.related, self.s1) + with self.assertNumQueries(0): + self.assertEqual(p2.related, self.s1) + def test_only_fetch_mode_raise(self): p1 = Primary.objects.fetch_mode(RAISE).only("name").get(name="p1") msg = "Fetching of Primary.value blocked." |