core: separate expiry logic from tokens and make re-usable

This commit is contained in:
Jens Langhammer
2020-07-20 10:57:12 +02:00
parent 2be7d3191f
commit c60d1e1f9a
4 changed files with 36 additions and 31 deletions

View File

@ -1,10 +1,10 @@
"""passbook user view tests"""
"""passbook core task tests"""
from django.test import TestCase
from django.utils.timezone import now
from guardian.shortcuts import get_anonymous_user
from passbook.core.models import Token
from passbook.core.tasks import clean_tokens
from passbook.core.tasks import clean_expired_models
class TestTasks(TestCase):
@ -14,5 +14,5 @@ class TestTasks(TestCase):
"""Test Token cleanup task"""
Token.objects.create(expires=now(), user=get_anonymous_user())
self.assertEqual(Token.objects.all().count(), 1)
clean_tokens()
clean_expired_models()
self.assertEqual(Token.objects.all().count(), 0)