Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-17 04:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added deconstruct

Browse source
This commit is contained in:
Nafees Anwar 2021-04-02 13:28:51 +05:00
parent a56d07cd68
commit 9deb39dc46
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
model_utils/fields.py
View file

@ -349,3 +349,8 @@ class UrlsafeTokenField(models.CharField):
# generate a token of length x1.33 approx. trim up to max length
token = secrets.token_urlsafe(self.max_length)[:self.max_length]
return token
def deconstruct(self):
name, path, args, kwargs = super().deconstruct()
kwargs['factory'] = self._factory
return name, path, args, kwargs

9
tests/test_fields/test_urlsafe_token_field.py
View file

@ -53,3 +53,12 @@ class UrlsaftTokenFieldTests(TestCase):
def test_no_default_param(self):
field = UrlsafeTokenField(default='DEFAULT')
self.assertIs(field.default, NOT_PROVIDED)
def test_deconstruct(self):
def test_factory():
pass
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)