entrouvert
/
hobo
16
0
Fork 0
Code Issues Pull Requests 18 Releases Activity

tests: dynamically change ALLOWED_HOSTS in multitenants tests

This commit is contained in:
Emmanuel Cazenave 2018-03-21 13:25:19 +01:00 committed by Frédéric Péters
parent 2fde2a61b8
commit fe017d9deb
3 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests_multitenant/test_request_context_filter.py
View File

@ -18,6 +18,7 @@ def test_request_context_filter(caplog, settings, tenants, client):
user.saml_identifiers.create(name_id='ab' * 16, issuer='https://idp.example.com')
for tenant in tenants:
settings.ALLOWED_HOSTS.append(tenant.domain_url)
with tenant_context(tenant):
client.login(username='john.doe', password='john.doe')
client.get('/', SERVER_NAME=tenant.domain_url,

4
tests_multitenant/test_settings.py
View File

@ -13,10 +13,12 @@ from tenant_schemas.utils import tenant_context
import utilities
def test_tenant_middleware(tenants, client):
def test_tenant_middleware(tenants, client, settings):
settings.ALLOWED_HOSTS.append('invalid.example.net')
res = client.get('/', SERVER_NAME='invalid.example.net')
assert res.status_code == 404
for tenant in tenants:
settings.ALLOWED_HOSTS.append(tenant.domain_url)
res = client.get('/', SERVER_NAME=tenant.domain_url)
assert res.status_code != 404
assert res.wsgi_request.tenant.schema_name == tenant.schema_name

1
tests_multitenant/test_xforwardedfor.py
View File

@ -4,6 +4,7 @@ from django.db import connection
def test_xforwardedfor(settings, tenants, client):
settings.USE_X_FORWARDED_FOR = True
for tenant in tenants:
settings.ALLOWED_HOSTS.append(tenant.domain_url)
response = client.get('/', SERVER_NAME=tenant.domain_url,
HTTP_X_FORWARDED_FOR='99.99.99.99, 127.0.0.1')
assert '99.99.99.99' in response.content