26 lines
646 B
Python
26 lines
646 B
Python
|
"""passbook user view tests"""
|
||
|
|
from guardian.shortcuts import get_anonymous_user
|
||
|
|
from django.test import TestCase
|
||
|
|
from django.utils.timezone import now
|
||
|
|
|
||
|
|
from passbook.core.tasks import clean_tokens
|
||
|
|
from passbook.core.models import Token
|
||
|
|
|
||
|
|
|
||
|
|
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_tokens()
|
||
|
|
self.assertEqual(
|
||
|
|
Token.objects.all().count(), 0
|
||
|
|
)
|