diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/many_to_one/tests.py | 10 | ||||
| -rw-r--r-- | tests/one_to_one/tests.py | 10 |
2 files changed, 20 insertions, 0 deletions
diff --git a/tests/many_to_one/tests.py b/tests/many_to_one/tests.py index 0e3395bf3a..a2ff587ab3 100644 --- a/tests/many_to_one/tests.py +++ b/tests/many_to_one/tests.py @@ -656,3 +656,13 @@ class ManyToOneTests(TestCase): self.assertEqual(city.districts.count(), 2) city.districts.remove(d2) self.assertEqual(city.districts.count(), 1) + + def test_cached_relation_invalidated_on_save(self): + """ + Model.save() invalidates stale ForeignKey relations after a primary key + assignment. + """ + self.assertEqual(self.a.reporter, self.r) # caches a.reporter + self.a.reporter_id = self.r2.pk + self.a.save() + self.assertEqual(self.a.reporter, self.r2) diff --git a/tests/one_to_one/tests.py b/tests/one_to_one/tests.py index 314c56a60f..a3c84a9980 100644 --- a/tests/one_to_one/tests.py +++ b/tests/one_to_one/tests.py @@ -507,3 +507,13 @@ class OneToOneTests(TestCase): pointer = ToFieldPointer.objects.create(target=target) self.assertSequenceEqual(ToFieldPointer.objects.filter(target=target), [pointer]) self.assertSequenceEqual(ToFieldPointer.objects.filter(pk__exact=pointer), [pointer]) + + def test_cached_relation_invalidated_on_save(self): + """ + Model.save() invalidates stale OneToOneField relations after a primary + key assignment. + """ + self.assertEqual(self.b1.place, self.p1) # caches b1.place + self.b1.place_id = self.p2.pk + self.b1.save() + self.assertEqual(self.b1.place, self.p2) |