Jan Dittberner
c94f20f83f
This commit adds a test suite in tests.py, the added setup.cfg contains configuration for coverage, pep8 and flake8. Output files of coverage are ignored. The commit also adds requirements files for tests.
146 lines
4.3 KiB
Python
146 lines
4.3 KiB
Python
"""
|
|
Tests for gvacommon code.
|
|
|
|
"""
|
|
from __future__ import absolute_import, unicode_literals
|
|
import unittest
|
|
|
|
try:
|
|
from unittest.Mock import MagicMock
|
|
except ImportError:
|
|
from mock import MagicMock
|
|
|
|
import django
|
|
from django.db import connection
|
|
from django.http import HttpResponseForbidden
|
|
from django.conf import settings
|
|
from django.views.generic import View
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from celeryrouters import GvaRouter
|
|
|
|
if not settings.configured:
|
|
settings.configure(
|
|
DEBUG=True,
|
|
ROOT_URLCONF='tests',
|
|
TEMPLATE_DIRS=['.'],
|
|
DATABASES={
|
|
'default': {
|
|
'ENGINE': 'django.db.backends.sqlite3',
|
|
'NAME': 'test',
|
|
}
|
|
},
|
|
INSTALLED_APPS=(
|
|
'django.contrib.contenttypes',
|
|
'django.contrib.auth',),
|
|
)
|
|
|
|
urlpatterns = []
|
|
|
|
|
|
class GvaRouteTest(unittest.TestCase):
|
|
|
|
def test_route_for_task_ldap(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('ldap')
|
|
self.assertEqual(
|
|
route,
|
|
{'exchange': 'ldap', 'exchange_type': 'direct', 'queue': 'ldap'})
|
|
|
|
def test_route_for_task_file(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('file')
|
|
self.assertEqual(
|
|
route,
|
|
{'exchange': 'file', 'exchange_type': 'direct', 'queue': 'file'})
|
|
|
|
def test_route_for_task_mysql(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('mysql')
|
|
self.assertEqual(
|
|
route,
|
|
{'exchange': 'mysql', 'exchange_type': 'direct', 'queue': 'mysql'})
|
|
|
|
def test_route_for_task_pgsql(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('pgsql')
|
|
self.assertEqual(
|
|
route,
|
|
{'exchange': 'pgsql', 'exchange_type': 'direct', 'queue': 'pgsql'})
|
|
|
|
def test_route_for_task_web(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('web')
|
|
self.assertEqual(
|
|
route,
|
|
{'exchange': 'web', 'exchange_type': 'direct', 'queue': 'web'})
|
|
|
|
def test_route_for_task_other(self):
|
|
router = GvaRouter()
|
|
route = router.route_for_task('unknown')
|
|
self.assertIsNone(route)
|
|
|
|
|
|
class StaffOrSelfLoginRequiredMixinTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
User = get_user_model()
|
|
User.objects.create_superuser(
|
|
username='admin', password='admin', email='test@example.org')
|
|
User.objects.create_user(username='test')
|
|
|
|
def setUp(self):
|
|
from viewmixins import StaffOrSelfLoginRequiredMixin
|
|
|
|
class SubjectView(StaffOrSelfLoginRequiredMixin, View):
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
return "success"
|
|
|
|
def get_customer_object(self):
|
|
return get_user_model().objects.get(username='test')
|
|
|
|
self.subject = SubjectView.as_view()
|
|
|
|
def test_dispatch_anonymous(self):
|
|
from django.contrib.auth.models import AnonymousUser
|
|
request = MagicMock(method='GET')
|
|
request.user = AnonymousUser()
|
|
result = self.subject(request)
|
|
self.assertIsInstance(result, HttpResponseForbidden)
|
|
|
|
def test_dispatch_staff_user(self):
|
|
request = MagicMock(method='GET')
|
|
request.user = get_user_model().objects.get(username='admin')
|
|
result = self.subject(request)
|
|
self.assertEqual(result, "success")
|
|
|
|
def test_dispatch_customer_user(self):
|
|
request = MagicMock(method='GET')
|
|
request.user = get_user_model().objects.get(username='test')
|
|
result = self.subject(request)
|
|
self.assertEqual(result, "success")
|
|
|
|
def test_dispatch_other_user(self):
|
|
request = MagicMock(method='GET')
|
|
request.user = get_user_model().objects.create_user('other')
|
|
result = self.subject(request)
|
|
self.assertIsInstance(result, HttpResponseForbidden)
|
|
|
|
def test_get_customer_object_not_implemented(self):
|
|
from viewmixins import StaffOrSelfLoginRequiredMixin
|
|
|
|
class IncompleteView(StaffOrSelfLoginRequiredMixin, View):
|
|
pass
|
|
|
|
view = IncompleteView()
|
|
with self.assertRaises(NotImplementedError):
|
|
view.get_customer_object()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
django.setup()
|
|
connection.creation.create_test_db()
|
|
unittest.main()
|
|
connection.creation.destroy_test_db()
|