我已经为我们的 django 后端应用程序 (SP) 实现了通过 SAML 登录的可能性,因为 IDP im 使用 Keycloak。它工作正常,但我想编写测试以确保所有逻辑都正确执行。为此,我想生成一个以 SAML 为主体的发布请求,并模拟 (unittest.mock.patch) 真实请求。但我卡住了。
这是我的 django 视图,当我尝试通过 SAML 登录时,它接受 get 和 post 请求:
class SamlLoginView(View):
@staticmethod
def prepare_django_request(request):
if 'HTTP_X_FORWARDED_FOR' in request.META:
server_port = 443
else:
server_port = request.META.get('SERVER_PORT')
result = {
'https': 'on' if request.is_secure() else 'off',
'http_host': request.META['HTTP_HOST'],
'script_name': request.META['PATH_INFO'],
'server_port': server_port,
'get_data': request.GET.copy(),
'post_data': request.POST.copy(),
}
return result
@never_cache
def get(self, *args, **kwargs):
req = SamlLoginView.prepare_django_request(self.request)
auth = OneLogin_Saml2_Auth(req, settings.SAML_IDP_SETTINGS)
return_url = self.request.GET.get('next') or settings.LOGIN_REDIRECT_URL
return HttpResponseRedirect(auth.login(return_to=return_url))
@never_cache
def post(self, *args, **kwargs):
req = SamlLoginView.prepare_django_request(self.request)
print(req['post_data']['SAMLResponse'])
auth = OneLogin_Saml2_Auth(req, settings.SAML_IDP_SETTINGS)
auth.process_response()
errors = auth.get_errors()
if not errors:
if auth.is_authenticated():
logger.info("Login", extra={'action': 'login',
'userid': auth.get_nameid()})
user = authenticate(request=self.request,
saml_authentication=auth)
login(self.request, user)
return HttpResponseRedirect("/")
else:
raise PermissionDenied()
else:
return HttpResponseBadRequest("Error when processing SAML Response: %s" % (', '.join(errors)))
在我的测试中,我想直接调用post方法,里面会有一个saml:
class TestSamlLogin(TestCase):
def test_saml_auth(self, prepare):
client = APIClient()
url = reverse_lazy("miri_auth:samllogin")
saml_resp='<xml with saml response>'
resp = client.post(url, data=saml_resp)
但显然它表明 request.POST 是空的。
然后我决定为这个prepare_django_request
函数做一个模拟,并手动插入 saml:
def mocked_prepare_request(request):
post_query_dict = QueryDict(mutable=True)
post_data = {
'SAMLResponse': saml_xml,
'RelayState': '/accounts/profile/'
}
post_query_dict.update(post_data)
result = {
'https': 'on',
'http_host': '<http-host>',
'script_name': '/api/auth/samllogin/',
'server_port': '443',
'get_data': {},
'post_data': post_query_dict,
}
return result
class TestSamlLogin(TestCase):
@patch('miri_auth.views.SamlLoginView.prepare_django_request', side_effect=mocked_prepare_request)
def test_saml_auth(self, prepare):
client = APIClient()
url = reverse_lazy("miri_auth:samllogin")
saml_resp='<xml with saml response>'
resp = client.post(url, data=saml_resp)
并且取决于我如何传递saml_xml
它会引发不同的错误,如果我将它定义为字符串:
with open(os.path.join(TEST_FILES_PATH, 'saml.xml')) as f:
saml_xml = " ".join([x.strip() for x in f])
它返回:lxml.etree.XMLSyntaxError: Start tag expected, '<' not found, line 1, column 1
,虽然我检查了saml_xml
xml 验证器中的输出,它说 xml 是有效的。当我尝试提前将文件解析为 xml 时,我稍后会收到另一个错误,我尝试解析的库:
import xml.etree.ElementTree as ET
from xml.dom import minidom
from lxml import etree
tree = etree.parse(os.path.join(TEST_FILES_PATH, 'saml.xml'))
它返回:
TypeError: argument should be a bytes-like object or ASCII string, not '_ElementTree'
调试这些错误并没有让我找到任何解决方案。
如果有人对如何实现这一点有任何想法(使用 SAML 模拟响应),或者我在哪里犯了错误,我会很高兴听到。
预先感谢