diff options
Diffstat (limited to 'tests/basic/tests.py')
| -rw-r--r-- | tests/basic/tests.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/basic/tests.py b/tests/basic/tests.py index 6d34a95805..cb267be0b1 100644 --- a/tests/basic/tests.py +++ b/tests/basic/tests.py @@ -661,6 +661,31 @@ class ModelTest(TestCase): headline__startswith="Area", ) + def test_is_pk_unset(self): + cases = [ + Article(), + Article(id=None), + ] + for case in cases: + with self.subTest(case=case): + self.assertIs(case._is_pk_set(), False) + + def test_is_pk_set(self): + def new_instance(): + a = Article(pub_date=datetime.today()) + a.save() + return a + + cases = [ + Article(id=1), + Article(id=0), + Article.objects.create(pub_date=datetime.today()), + new_instance(), + ] + for case in cases: + with self.subTest(case=case): + self.assertIs(case._is_pk_set(), True) + class ModelLookupTest(TestCase): @classmethod |
