adds classmethod decorator to generate_key and test to confirm change is acceptable

This commit is contained in:
checkroth 2020-08-28 13:42:17 +09:00
parent 7f3a3557a0
commit 5c86a31481
2 changed files with 5 additions and 0 deletions

View File

@ -32,6 +32,7 @@ class Token(models.Model):
self.key = self.generate_key()
return super().save(*args, **kwargs)
@classmethod
def generate_key(self):
return binascii.hexlify(os.urandom(20)).decode()

View File

@ -397,6 +397,10 @@ class TokenAuthTests(BaseTokenAuthTests, TestCase):
key = token.generate_key()
assert isinstance(key, str)
def test_generate_key_accessible_as_classmethod(self):
key = self.model.generate_key()
assert isinstance(key, str)
def test_token_login_json(self):
"""Ensure token login view using JSON POST works."""
client = APIClient(enforce_csrf_checks=True)