mirror of
https://github.com/Hopiu/django-model-utils.git
synced 2026-03-17 04:10:24 +00:00
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import Mock
|
|
|
|
from django.db.models import NOT_PROVIDED
|
|
from django.test import TestCase
|
|
|
|
from model_utils.fields import UrlsafeTokenField
|
|
|
|
|
|
class UrlsaftTokenFieldTests(TestCase):
|
|
def test_editable_default(self) -> None:
|
|
field = UrlsafeTokenField()
|
|
self.assertFalse(field.editable)
|
|
|
|
def test_editable(self) -> None:
|
|
field = UrlsafeTokenField(editable=True)
|
|
self.assertTrue(field.editable)
|
|
|
|
def test_max_length_default(self) -> None:
|
|
field = UrlsafeTokenField()
|
|
self.assertEqual(field.max_length, 128)
|
|
|
|
def test_max_length(self) -> None:
|
|
field = UrlsafeTokenField(max_length=256)
|
|
self.assertEqual(field.max_length, 256)
|
|
|
|
def test_factory_default(self) -> None:
|
|
field = UrlsafeTokenField()
|
|
self.assertIsNone(field._factory)
|
|
|
|
def test_factory_not_callable(self) -> None:
|
|
with self.assertRaises(TypeError):
|
|
UrlsafeTokenField(factory='INVALID') # type: ignore[arg-type]
|
|
|
|
def test_get_default(self) -> None:
|
|
field = UrlsafeTokenField()
|
|
value = field.get_default()
|
|
self.assertEqual(len(value), field.max_length)
|
|
|
|
def test_get_default_with_non_default_max_length(self) -> None:
|
|
field = UrlsafeTokenField(max_length=64)
|
|
value = field.get_default()
|
|
self.assertEqual(len(value), 64)
|
|
|
|
def test_get_default_with_factory(self) -> None:
|
|
token = 'SAMPLE_TOKEN'
|
|
factory = Mock(return_value=token)
|
|
field = UrlsafeTokenField(factory=factory)
|
|
value = field.get_default()
|
|
|
|
self.assertEqual(value, token)
|
|
factory.assert_called_once_with(field.max_length)
|
|
|
|
def test_no_default_param(self) -> None:
|
|
field = UrlsafeTokenField(default='DEFAULT')
|
|
self.assertIs(field.default, NOT_PROVIDED)
|
|
|
|
def test_deconstruct(self) -> None:
|
|
def test_factory(max_length: int) -> str:
|
|
assert False
|
|
instance = UrlsafeTokenField(factory=test_factory)
|
|
name, path, args, kwargs = instance.deconstruct()
|
|
new_instance = UrlsafeTokenField(*args, **kwargs)
|
|
self.assertIs(instance._factory, new_instance._factory)
|
|
self.assertIs(test_factory, new_instance._factory)
|