"""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)