diff options
Diffstat (limited to 'tests/composite_pk/tests.py')
| -rw-r--r-- | tests/composite_pk/tests.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tests/composite_pk/tests.py b/tests/composite_pk/tests.py index 264a1bb7c2..4284f9d627 100644 --- a/tests/composite_pk/tests.py +++ b/tests/composite_pk/tests.py @@ -17,7 +17,7 @@ from django.db.models import CompositePrimaryKey from django.forms import modelform_factory from django.test import TestCase -from .models import Comment, Post, Tenant, TimeStamped, Token, User +from .models import Comment, Post, PostDbDefault, Tenant, TimeStamped, Token, User class CommentForm(forms.ModelForm): @@ -64,6 +64,12 @@ class CompositePKTests(TestCase): self.assertIsNone(user.id) self.assertIs(user._is_pk_set(), False) + def test_pk_not_set_db_default(self): + post = PostDbDefault(tenant=self.tenant) + self.assertEqual(post.tenant_id, self.tenant.pk) + self.assertIsNotNone(post.id) + self.assertIs(post._is_pk_set(), False) + def test_hash(self): self.assertEqual(hash(User(pk=(1, 2))), hash((1, 2))) self.assertEqual(hash(User(tenant_id=2, id=3)), hash((2, 3))) |