diff options
author | Karl Hobley <karlhobley10@gmail.com> | 2015-03-16 19:28:53 +0000 |
---|---|---|
committer | Tim Graham <timograham@gmail.com> | 2015-03-18 19:00:09 -0400 |
commit | 81e1a35c364e5353d2bf99368ad30a4184fbb653 (patch) | |
tree | 2cbfcc6a605600d4f4c387d4c49d17dfe300b91f /tests/many_to_one | |
parent | 02d78bb1a80706d941ffc6c892cc75208eb6b782 (diff) | |
download | django-81e1a35c364e5353d2bf99368ad30a4184fbb653.tar.gz |
Fixed #24495 -- Allowed unsaved model instance assignment check to be bypassed.
Diffstat (limited to 'tests/many_to_one')
-rw-r--r-- | tests/many_to_one/tests.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/many_to_one/tests.py b/tests/many_to_one/tests.py index fdb77b4711..af7503fba8 100644 --- a/tests/many_to_one/tests.py +++ b/tests/many_to_one/tests.py @@ -159,6 +159,31 @@ class ManyToOneTests(TestCase): self.assertFalse(hasattr(self.r2.article_set, 'remove')) self.assertFalse(hasattr(self.r2.article_set, 'clear')) + def test_assign_unsaved_check_override(self): + """ + #24495 - Assigning an unsaved object to a ForeignKey + should be allowed when the allow_unsaved_instance_assignment + attribute has been set to True. + """ + class UnsavedForeignKey(models.ForeignKey): + # A ForeignKey which can point to an unsaved object + allow_unsaved_instance_assignment = True + + class Band(models.Model): + name = models.CharField(max_length=50) + + class BandMember(models.Model): + band = UnsavedForeignKey(Band) + first_name = models.CharField(max_length=50) + last_name = models.CharField(max_length=50) + + beatles = Band(name='The Beatles') + john = BandMember(first_name='John', last_name='Lennon') + # This should not raise an exception as the ForeignKey between member + # and band has allow_unsaved_instance_assignment=True. + john.band = beatles + self.assertEqual(john.band, beatles) + def test_selects(self): self.r.article_set.create(headline="John's second story", pub_date=datetime.date(2005, 7, 29)) |