def test_tenant_view_post_value_error(self, mock_tenant_form, mock_do_auth,
mock_render):
mock_tenant_form.is_valid = mock.MagicMock(return_value=True)
mock_tenant_form().cleaned_data = {'tenant': 'GitHub'}
mock_backend = mock.MagicMock(spec=MultiTenantMixin)
type(mock_backend).tenant = mock.PropertyMock(side_effect=ValueError)
post_params = {
'sso-tenant-submit': 'submit',
'tenant': 'GitHub',
'remember': False
}
mock_request = mock.MagicMock(
method='POST', POST=post_params,
backend=mock_backend) # valid POST request
url_backend = 'foo'
ret = tenant(mock_request, url_backend)
# Make sure form is bound to POST data
mock_tenant_form.assert_called_with(mock_request.POST)
mock_tenant_form().is_valid.assert_called()
mock_do_auth.assert_not_called()
mock_tenant_form().add_error.assert_called_with(
'tenant', 'Invalid thingy provided.')
mock_render.assert_called_with(
mock_request, 'tethys_portal/accounts/sso_tenant.html', {
'form': mock_tenant_form(),
'form_title': 'Thingy',
'page_title': 'Thingy',
'backend': url_backend
})
self.assertEqual(mock_render(), ret)
def test_tenant_view_post_improperly_configured(self, mock_tenant_form,
mock_do_auth,
mock_redirect, mock_log):
mock_tenant_form.is_valid = mock.MagicMock(return_value=True)
mock_tenant_form().cleaned_data = {'tenant': 'GitHub'}
mock_backend = mock.MagicMock(spec=MultiTenantMixin)
type(mock_backend).tenant = mock.PropertyMock(
side_effect=ImproperlyConfigured('some error message'))
post_params = {
'sso-tenant-submit': 'submit',
'tenant': 'GitHub',
'remember': False
}
mock_request = mock.MagicMock(
method='POST', POST=post_params,
backend=mock_backend) # valid POST request
url_backend = 'foo'
ret = tenant(mock_request, url_backend)
# Make sure form is bound to POST data
mock_tenant_form.assert_called_with(mock_request.POST)
mock_tenant_form().is_valid.assert_called()
mock_do_auth.assert_not_called()
mock_log.error.assert_called_with('some error message')
mock_redirect.assert_called_with('accounts:login')
self.assertEqual(mock_redirect(), ret)
def test_tenant_view_post_no_submit(self, mock_tenant_form):
mock_backend = mock.MagicMock(spec=MultiTenantMixin)
mock_request = mock.MagicMock(method='POST',
POST=dict(),
backend=mock_backend) # Empty POST dict
url_backend = 'foo'
ret = tenant(mock_request, url_backend)
mock_tenant_form.assert_not_called()
self.assertIsInstance(ret, HttpResponseBadRequest)
def test_tenant_get_backend_not_mtm(self, mock_redirect, mock_log):
mock_backend = mock.MagicMock(spec=BaseAuth) # Not a MultiTenantMixin
mock_backend.name = 'other backend'
mock_request = mock.MagicMock(method='GET',
GET=dict(),
backend=mock_backend) # GET request
url_backend = 'foo'
ret = tenant(mock_request, backend=url_backend)
mock_log.error.assert_called_with(
'Backend "other backend" does not support MULTI_TENANT features.')
mock_redirect.assert_called_with('accounts:login')
self.assertEqual(mock_redirect(), ret)
def test_tenant_get(self, mock_render, mock_tenant_form):
mock_backend = mock.MagicMock(spec=MultiTenantMixin)
mock_request = mock.MagicMock(method='GET',
GET=dict(),
backend=mock_backend) # GET request
url_backend = 'foo'
ret = tenant(mock_request, backend=url_backend)
mock_tenant_form.assert_called()
mock_render.assert_called_with(
mock_request, 'tethys_portal/accounts/sso_tenant.html', {
'form': mock_tenant_form(),
'form_title': 'Foo Bar',
'page_title': 'Foo Bar',
'backend': url_backend
})
self.assertEqual(mock_render(), ret)
def test_tenant_view_post_valid(self, mock_tenant_form, mock_do_auth):
mock_tenant_form.is_valid = mock.MagicMock(return_value=True)
mock_tenant_form().cleaned_data = {'tenant': 'GitHub'}
mock_backend = mock.MagicMock(spec=MultiTenantMixin)
post_params = {
'sso-tenant-submit': 'submit',
'tenant': 'GitHub',
'remember': False
}
mock_request = mock.MagicMock(
method='POST', POST=post_params,
backend=mock_backend) # valid POST request
url_backend = 'foo'
ret = tenant(mock_request, url_backend)
# Make sure form is bound to POST data
mock_tenant_form.assert_called_with(mock_request.POST)
mock_tenant_form().is_valid.assert_called()
mock_do_auth.assert_called_with(mock_backend,
redirect_name=REDIRECT_FIELD_NAME)
self.assertEqual('GitHub', mock_backend.tenant)
self.assertEqual(mock_do_auth(), ret)