from __future__ import unicode_literals

import uuid

from django.test import TestCase

from model_utils.fields import UUIDField
from model_utils.exceptions import UUIDVersionException


class UUIDFieldTests(TestCase):

    def test_uuid_version_default(self):
        instance = UUIDField()
        self.assertEqual(instance.default, uuid.uuid4)

    def test_uuid_version_1(self):
        instance = UUIDField(version=1)
        self.assertEqual(instance.default, uuid.uuid1)

    def test_uuid_version_2_error(self):
        self.assertRaises(UUIDVersionException, UUIDField, 'version', 2)

    def test_uuid_version_3(self):
        instance = UUIDField(version=3)
        self.assertEqual(instance.default, uuid.uuid3)

    def test_uuid_version_4(self):
        instance = UUIDField(version=4)
        self.assertEqual(instance.default, uuid.uuid4)

    def test_uuid_version_5(self):
        instance = UUIDField(version=5)
        self.assertEqual(instance.default, uuid.uuid5)