def test_get_logs_out_if_multifactor_invalid(self, logged_in_request):
def valid(self, request):
return False
a = ProcessMultiFactorView.multifactor_is_valid
ProcessMultiFactorView.multifactor_is_valid = valid
with pytest.raises(PermissionDenied):
ProcessMultiFactorView.as_view()(logged_in_request)
ProcessMultiFactorView.multifactor_is_valid = a
def test_loads_data_when_appropriate_with_post(self, logged_in_request):
logged_in_request.session['saml_data'] = {
"type": "POST",
"data": "<html></html>",
}
response = ProcessMultiFactorView.as_view()(logged_in_request)
assert isinstance(response, HttpResponse)
assert response.content == '<html></html>'.encode()
def test_loads_data_when_appropriate_with_redirect(self):
request = get_logged_in_request()
request.session['saml_data'] = {
"type": "REDIRECT",
"data": "https://example.com",
}
response = ProcessMultiFactorView.as_view()(request)
assert isinstance(response, HttpResponseRedirect)
assert response.url == "https://example.com"