django-model-utils/tests/test_models/test_uuid_model.py
2019-02-26 17:36:35 +01:00

20 lines
678 B
Python

from __future__ import unicode_literals
from django.test import TestCase
from tests.models import CustomUUIDModel, CustomNotPrimaryUUIDModel
class UUIDFieldTests(TestCase):
def test_uuid_model_with_uuid_field_as_primary_key(self):
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):
instance = CustomNotPrimaryUUIDModel()
instance.save()
self.assertEqual(instance.uuid.__class__.__name__, 'UUID')
self.assertNotEqual(instance.uuid, instance.pk)