📦 charettes / django-tenancy

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98from __future__ import unicode_literals

from unittest import skipIf, skipUnless

import django
from django.core.exceptions import ImproperlyConfigured
from django.test.utils import override_settings
from django.utils.encoding import force_bytes

from tenancy.middleware import TenantHostMiddleware
from tenancy.models import Tenant

from .utils import MIDDLEWARE_SETTING, TenancyTestCase

try:
    import django_hosts
except ImportError:
    django_hosts = None


def django_hosts_installed_setup(func):
    func = override_settings(
        DEFAULT_HOST='default',
        ROOT_HOSTCONF='tests.hosts',
        **{MIDDLEWARE_SETTING: [
            'django_hosts.middleware.HostsRequestMiddleware',
            'tenancy.middleware.TenantHostMiddleware',
            'django_hosts.middleware.HostsResponseMiddleware',
        ]}
    )(func)
    return skipUnless(
        django_hosts,
        'django-hosts is not installed.'
    )(func)


@override_settings(ROOT_URLCONF='tests.urls')
class TenantHostMiddlewareTest(TenancyTestCase):
    @classmethod
    def tenant_client(cls, tenant):
        domain = "%s.testserver" % tenant.name
        return cls.client_class(SERVER_NAME=domain)

    @skipIf(django_hosts, 'django-hosts is installed.')
    def test_not_installed(self):
        self.assertRaisesMessage(
            ImproperlyConfigured,
            'You must install django-hosts in order to use '
            '`TenantHostMiddleware`.',
            TenantHostMiddleware
        )

    @skipUnless(django.VERSION >= (1, 10), 'New-style middleware was introduced in Django 1.10.')
    @skipUnless(django_hosts, 'django-hosts is not installed.')
    @override_settings(
        MIDDLEWARE=(
            'tenancy.middleware.TenantHostMiddleware',
            'django_hosts.middleware.HostsRequestMiddleware',
        ),
    )
    def test_wrong_middleware_order(self):
        message = (
            "Make sure 'django_hosts.middleware.HostsRequestMiddleware' appears before "
            "'tenancy.middleware.TenantHostMiddleware' in your `MIDDLEWARE` setting."
        )
        self.assertRaisesMessage(ImproperlyConfigured, message, TenantHostMiddleware)

    @skipUnless(django.VERSION < (2, 0), 'Old-style middleware support was removed in Django 2.0.')
    @skipUnless(django_hosts, 'django-hosts is not installed.')
    @override_settings(
        MIDDLEWARE_CLASSES=(
            'tenancy.middleware.TenantHostMiddleware',
            'django_hosts.middleware.HostsRequestMiddleware',
        )
    )
    def test_wrong_middleware_classes_order(self):
        message = (
            "Make sure 'django_hosts.middleware.HostsRequestMiddleware' appears before "
            "'tenancy.middleware.TenantHostMiddleware' in your `MIDDLEWARE_CLASSES` setting."
        )
        self.assertRaisesMessage(ImproperlyConfigured, message, TenantHostMiddleware)

    @django_hosts_installed_setup
    def test_tenant_not_found(self):
        tenant = Tenant(name='inexistent')
        client = self.tenant_client(tenant)
        with self.settings(ALLOWED_HOSTS=[client.defaults['SERVER_NAME']]):
            response = client.get('/')
        self.assertEqual(response.status_code, 404)

    @django_hosts_installed_setup
    def test_tenant_found(self):
        client = self.tenant_client(self.tenant)
        with self.settings(ALLOWED_HOSTS=[client.defaults['SERVER_NAME']]):
            response = client.get('/')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.content, force_bytes(self.tenant.name))