19 lines
584 B
Python
19 lines
584 B
Python
|
"""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_expired_models
|
||
|
|
|
||
|
|
|
||
|
|
class TestTasks(TestCase):
|
||
|
|
"""Test Tasks"""
|
||
|
|
|
||
|
|
def test_token_cleanup(self):
|
||
|
|
"""Test Token cleanup task"""
|
||
|
|
Token.objects.create(expires=now(), user=get_anonymous_user())
|
||
|
|
self.assertEqual(Token.objects.all().count(), 1)
|
||
|
|
clean_expired_models()
|
||
|
|
self.assertEqual(Token.objects.all().count(), 0)
|