mirror of
https://github.com/Hopiu/django-model-utils.git
synced 2026-03-16 20:00:23 +00:00
20 lines
689 B
Python
20 lines
689 B
Python
from __future__ import annotations
|
|
|
|
from django.test import TestCase
|
|
|
|
from tests.models import CustomNotPrimaryUUIDModel, CustomUUIDModel
|
|
|
|
|
|
class UUIDFieldTests(TestCase):
|
|
|
|
def test_uuid_model_with_uuid_field_as_primary_key(self) -> None:
|
|
instance = CustomUUIDModel()
|
|
instance.save()
|
|
self.assertEqual(instance.id.__class__.__name__, 'UUID')
|
|
self.assertEqual(instance.id, instance.pk)
|
|
|
|
def test_uuid_model_with_uuid_field_as_not_primary_key(self) -> None:
|
|
instance = CustomNotPrimaryUUIDModel()
|
|
instance.save()
|
|
self.assertEqual(instance.uuid.__class__.__name__, 'UUID')
|
|
self.assertNotEqual(instance.uuid, instance.pk)
|