def setUp(self):
super(RoutesTestCase, self).setUp()
self.factory = RequestFactory()
self.tm = TenantMiddleware()
self.tenant_domain = 'tenant.test.com'
self.tenant = Tenant(domain_url=self.tenant_domain, schema_name='test')
self.tenant.save()
class TenantClient(Client):
tm = TenantMiddleware()
def __init__(self, tenant, enforce_csrf_checks=False, **defaults):
super(TenantClient, self).__init__(enforce_csrf_checks, **defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantClient, self).get(path, data, **extra)
class TenantRequestFactory(RequestFactory):
tm = TenantMiddleware()
def __init__(self, tenant, **defaults):
super(TenantRequestFactory, self).__init__(**defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
request = super(TenantRequestFactory, self).get(path, data, **extra)
return self.tm.process_request(request)
def setUp(self):
super(RoutesTestCase, self).setUp()
self.factory = RequestFactory()
self.tm = TenantMiddleware()
self.dtm = DefaultTenantMiddleware()
self.tenant_domain = 'tenant.test.com'
self.tenant = Tenant(domain_url=self.tenant_domain, schema_name='test')
self.tenant.save(verbosity=BaseTestCase.get_verbosity())
self.non_existent_domain = 'no-tenant.test.com'
self.non_existent_tenant = Tenant(domain_url=self.non_existent_domain, schema_name='no-tenant')
self.url = '/any/path/'
def setUp(self):
self.factory = RequestFactory()
self.tm = TenantMiddleware()
# settings needs some patching
settings.TENANT_MODEL = 'tenant_schemas.Tenant'
# add the public tenant
self.public_tenant_domain = 'test.com'
self.public_tenant = Tenant(domain_url=self.public_tenant_domain,
schema_name='public')
self.public_tenant.save()
# add a test tenant
self.tenant_domain = 'tenant.test.com'
self.tenant = Tenant(domain_url=self.tenant_domain, schema_name='test')
self.tenant.save()
class TenantClient(Client):
tm = TenantMiddleware()
def __init__(self, tenant, enforce_csrf_checks=False, **defaults):
super(TenantClient, self).__init__(enforce_csrf_checks, **defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantClient, self).get(path, data, **extra)
def post(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantClient, self).post(path, data, **extra)
def patch(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
if 'content_type' not in extra:
extra["content_type"] = "application/json"
return super(TenantClient, self).patch(path, data, **extra)
def put(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
if 'content_type' not in extra:
extra["content_type"] = "application/json"
return super(TenantClient, self).put(path, data, **extra)
def delete(self, path, data='', content_type='application/octet-stream',
**extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantClient, self).delete(path, data, **extra)
class TenantRequestFactory(RequestFactory):
tm = TenantMiddleware()
def __init__(self, tenant, **defaults):
super(TenantRequestFactory, self).__init__(**defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantRequestFactory, self).get(path, data, **extra)
def post(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantRequestFactory, self).post(path, data, **extra)
def patch(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
if 'content_type' not in extra:
extra["content_type"] = "application/json"
return super(TenantRequestFactory, self).patch(path, data, **extra)
def put(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
if 'content_type' not in extra:
extra["content_type"] = "application/json"
return super(TenantRequestFactory, self).put(path, data, **extra)
def delete(self, path, data='', content_type='application/octet-stream',
**extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(TenantRequestFactory, self).delete(path, data, **extra)
class ApiClient(RestAPIClient):
tm = TenantMiddleware()
renderer_classes_list = api_settings.TEST_REQUEST_RENDERER_CLASSES
default_format = api_settings.TEST_REQUEST_DEFAULT_FORMAT
def __init__(self, tenant, enforce_csrf_checks=False, **defaults):
super(ApiClient, self).__init__(enforce_csrf_checks, **defaults)
self.tenant = tenant
self.renderer_classes = {}
for cls in self.renderer_classes_list:
self.renderer_classes[cls.format] = cls
def get(self, path, data=None, **extra):
if 'token' in extra:
extra['HTTP_AUTHORIZATION'] = extra['token']
del extra['token']
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(ApiClient, self).get(path, data=data, **extra)
def post(self, path, data=None, format='json', content_type=None, **extra):
if 'token' in extra:
extra['HTTP_AUTHORIZATION'] = extra['token']
del extra['token']
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(ApiClient, self).post(path,
data=data,
format=format,
content_type=content_type,
**extra)
def put(self, path, data=None, format='json', content_type=None, **extra):
if 'token' in extra:
extra['HTTP_AUTHORIZATION'] = extra['token']
del extra['token']
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(ApiClient, self).put(path,
data=data,
format=format,
content_type=content_type,
**extra)
def patch(self,
path,
data=None,
format='json',
content_type=None,
**extra):
if 'token' in extra:
extra['HTTP_AUTHORIZATION'] = extra['token']
del extra['token']
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(ApiClient, self).patch(path,
data=data,
format=format,
content_type=content_type,
**extra)
def delete(self,
path,
data=None,
format='json',
content_type=None,
**extra):
if 'token' in extra:
extra['HTTP_AUTHORIZATION'] = extra['token']
del extra['token']
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.domain_url
return super(ApiClient, self).delete(path,
data=data,
format=format,
content_type=content_type,
**extra)
