core: separate expiry logic from tokens and make re-usable
This commit is contained in:
@ -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)
|
||||
|
||||
Reference in New Issue
Block a user