38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""passbook util view tests"""
|
|
import string
|
|
from random import SystemRandom
|
|
|
|
from django.test import RequestFactory, TestCase
|
|
|
|
from passbook.core.models import User
|
|
from passbook.core.views.utils import LoadingView, PermissionDeniedView
|
|
|
|
|
|
class TestUtilViews(TestCase):
|
|
"""Test Utility Views"""
|
|
|
|
def setUp(self):
|
|
self.user = User.objects.create_superuser(
|
|
username="unittest user",
|
|
email="unittest@example.com",
|
|
password="".join(
|
|
SystemRandom().choice(string.ascii_uppercase + string.digits)
|
|
for _ in range(8)
|
|
),
|
|
)
|
|
self.factory = RequestFactory()
|
|
|
|
def test_loading_view(self):
|
|
"""Test loading view"""
|
|
request = self.factory.get("something")
|
|
response = LoadingView.as_view(target_url="somestring")(request)
|
|
response.render()
|
|
self.assertIn("somestring", response.content.decode("utf-8"))
|
|
|
|
def test_permission_denied_view(self):
|
|
"""Test PermissionDeniedView"""
|
|
request = self.factory.get("something")
|
|
request.user = self.user
|
|
response = PermissionDeniedView.as_view()(request)
|
|
self.assertEqual(response.status_code, 200)
|