Jan Dittberner
e6c38b632b
This commit adds tests for userdbs.views.AddUserDatabase, the view itself has been modified so that it is only reachable when the hosting package actually has database options available.
203 lines
7.4 KiB
Python
203 lines
7.4 KiB
Python
"""
|
|
This module provides tests for :py:mod:`userdbs.views`.
|
|
|
|
"""
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
try:
|
|
from unittest.mock import patch, MagicMock
|
|
except ImportError:
|
|
from mock import patch, MagicMock
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django.test import TestCase
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from hostingpackages.models import (
|
|
CustomerHostingPackage,
|
|
CustomerUserDatabaseOption,
|
|
HostingPackageTemplate,
|
|
UserDatabaseOption,
|
|
)
|
|
|
|
from userdbs.models import DB_TYPES, UserDatabase
|
|
from userdbs.views import AddUserDatabase
|
|
|
|
|
|
User = get_user_model()
|
|
|
|
TEST_USER = 'test'
|
|
TEST_PASSWORD = 'secret'
|
|
TEST_EMAIL = 'test@example.org'
|
|
|
|
|
|
class HostingPackageAwareTestMixin(object):
|
|
|
|
def _setup_hosting_package(self, customer):
|
|
template = HostingPackageTemplate.objects.create(
|
|
name='testpackagetemplate', mailboxcount=10, diskspace=1,
|
|
diskspace_unit=0)
|
|
package = CustomerHostingPackage.objects.create_from_template(
|
|
customer, template, 'testpackage')
|
|
with patch('hostingpackages.models.settings') as hmsettings:
|
|
hmsettings.OSUSER_DEFAULT_GROUPS = []
|
|
package.save()
|
|
return package
|
|
|
|
|
|
class CustomerUserDatabaseOptionAwareTestMixin(object):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(CustomerUserDatabaseOptionAwareTestMixin, self).__init__(
|
|
*args, **kwargs)
|
|
self._templates = {}
|
|
|
|
def _setup_userdatabaseoption(self, number, dbtype):
|
|
key = "{}_{}".format(dbtype, number)
|
|
if key not in self._templates:
|
|
self._templates[key] = UserDatabaseOption.objects.create(
|
|
number=number, db_type=dbtype)
|
|
return self._templates[key]
|
|
|
|
def _create_userdatabase_option(self, number=1, dbtype=DB_TYPES.pgsql):
|
|
return CustomerUserDatabaseOption.objects.create(
|
|
template=self._setup_userdatabaseoption(number, dbtype),
|
|
number=number, db_type=dbtype, hosting_package=self.package)
|
|
|
|
|
|
class AddUserDatabaseTest(
|
|
HostingPackageAwareTestMixin, CustomerUserDatabaseOptionAwareTestMixin,
|
|
TestCase
|
|
):
|
|
|
|
def setUp(self):
|
|
self.customer = User.objects.create_user(
|
|
username=TEST_USER, password=TEST_PASSWORD)
|
|
self.package = self._setup_hosting_package(self.customer)
|
|
|
|
def _get_url(self):
|
|
return reverse(
|
|
'add_userdatabase', kwargs={'package': self.package.id})
|
|
|
|
def test_get_anonymous(self):
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_get_regular_user_nodboption(self):
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_get_regular_user(self):
|
|
self._create_userdatabase_option()
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_get_other_regular_user(self):
|
|
User.objects.create_user(
|
|
'test2', password=TEST_PASSWORD)
|
|
self.client.login(username='test2', password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_get_staff_user_nodboption(self):
|
|
User.objects.create_superuser(
|
|
'admin', email=TEST_EMAIL, password=TEST_PASSWORD)
|
|
self.client.login(username='admin', password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_get_staff_user(self):
|
|
self._create_userdatabase_option()
|
|
User.objects.create_superuser(
|
|
'admin', email=TEST_EMAIL, password=TEST_PASSWORD)
|
|
self.client.login(username='admin', password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_get_regular_user_nofree_db(self):
|
|
db_option = self._create_userdatabase_option()
|
|
UserDatabase.objects.create_userdatabase_with_user(
|
|
db_option.db_type, self.package.osuser)
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_get_form_kwargs(self):
|
|
db_option = self._create_userdatabase_option()
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
view = AddUserDatabase(
|
|
request=MagicMock(), kwargs={'package': str(self.package.pk)})
|
|
the_kwargs = view.get_form_kwargs()
|
|
self.assertIn('hostingpackage', the_kwargs)
|
|
self.assertEqual(the_kwargs['hostingpackage'], self.package)
|
|
self.assertIn('dbtypes', the_kwargs)
|
|
self.assertEqual(
|
|
the_kwargs['dbtypes'],
|
|
[(db_option.db_type, DB_TYPES[db_option.db_type])],
|
|
)
|
|
|
|
def test_get_template(self):
|
|
self._create_userdatabase_option()
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.get(self._get_url())
|
|
self.assertTemplateUsed(response, 'userdbs/userdatabase_create.html')
|
|
|
|
def test_form_valid_redirect(self):
|
|
db_option = self._create_userdatabase_option()
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.post(
|
|
self._get_url(),
|
|
data={
|
|
'db_type': db_option.db_type, 'password1': TEST_PASSWORD,
|
|
'password2': TEST_PASSWORD})
|
|
self.assertRedirects(response, self.package.get_absolute_url())
|
|
|
|
def test_form_valid_message(self):
|
|
db_option = self._create_userdatabase_option()
|
|
self.client.login(username=TEST_USER, password=TEST_PASSWORD)
|
|
response = self.client.post(
|
|
self._get_url(), follow=True,
|
|
data={
|
|
'db_type': db_option.db_type, 'password1': TEST_PASSWORD,
|
|
'password2': TEST_PASSWORD})
|
|
db = UserDatabase.objects.filter(
|
|
db_user__osuser=self.package.osuser).get()
|
|
messages = list(response.context['messages'])
|
|
self.assertEqual(len(messages), 1)
|
|
self.assertEqual(
|
|
str(messages[0]), (
|
|
'Successfully create new {type} database {dbname} for user '
|
|
'{dbuser}').format(
|
|
type=db.db_user.db_type, dbname=db.db_name,
|
|
dbuser=db.db_user))
|
|
|
|
|
|
class ChangeDatabaseUserPasswordTest(HostingPackageAwareTestMixin, TestCase):
|
|
|
|
def setUp(self):
|
|
self.customer = User.objects.create_user(
|
|
username=TEST_USER, password=TEST_PASSWORD)
|
|
self.package = self._setup_hosting_package(self.customer)
|
|
|
|
def _get_url(self, userdatabase):
|
|
return reverse(
|
|
'change_dbuser_password', kwargs={
|
|
'package': self.package.id,
|
|
'slug': userdatabase.db_name})
|
|
|
|
|
|
class DeleteUserDatabaseTest(HostingPackageAwareTestMixin, TestCase):
|
|
|
|
def setUp(self):
|
|
self.customer = User.objects.create_user(
|
|
username=TEST_USER, password=TEST_PASSWORD)
|
|
self.package = self._setup_hosting_package(self.customer)
|
|
|
|
def _get_url(self, userdatabase):
|
|
return reverse(
|
|
'delete_userdatabase', kwargs={
|
|
'package': self.package.id,
|
|
'slug': userdatabase.db_name})
|