tests: simplify idp_oidc tests

This commit is contained in:
Benjamin Dauvergne 2021-03-03 19:26:13 +01:00
parent e2fa4ca689
commit 5a534ec62a
1 changed files with 4 additions and 24 deletions

View File

@ -332,18 +332,8 @@ def test_authorization_code_sso(login_first, do_not_ask_again, oidc_client, oidc
id_token = query['id_token'][0]
if oidc_client.idtoken_algo in (oidc_client.ALGO_RSA, oidc_client.ALGO_EC):
keyset = JWKSet.from_json(app.get(reverse('oidc-certs')).content)
for k in keyset.get('keys'):
if {
'RSA': oidc_client.ALGO_RSA,
'EC': oidc_client.ALGO_EC
}.get(k.key_type) == oidc_client.idtoken_algo:
algs=[{
oidc_client.ALGO_RSA: 'RS256',
oidc_client.ALGO_EC: 'ES256'
}.get(oidc_client.idtoken_algo)]
key = k
break
key = JWKSet.from_json(app.get(reverse('oidc-certs')).content)
algs = ['RS256', 'ES256']
elif oidc_client.idtoken_algo == oidc_client.ALGO_HMAC:
k = base64.b64encode(oidc_client.client_secret.encode('utf-8'))
key = JWK(kty='oct', k=force_text(k))
@ -986,18 +976,8 @@ def test_role_control_access(login_first, oidc_settings, oidc_client, simple_use
id_token = query['id_token'][0]
if oidc_client.idtoken_algo in (oidc_client.ALGO_RSA, oidc_client.ALGO_EC):
keyset = JWKSet.from_json(app.get(reverse('oidc-certs')).content)
for k in keyset.get('keys'):
if {
'RSA': oidc_client.ALGO_RSA,
'EC': oidc_client.ALGO_EC
}.get(k.key_type) == oidc_client.idtoken_algo:
algs=[{
oidc_client.ALGO_RSA: 'RS256',
oidc_client.ALGO_EC: 'ES256'
}.get(oidc_client.idtoken_algo)]
key = k
break
key = JWKSet.from_json(app.get(reverse('oidc-certs')).content)
algs = ['RS256', 'ES256']
elif oidc_client.idtoken_algo == oidc_client.ALGO_HMAC:
k = base64.b64encode(oidc_client.client_secret.encode('utf-8'))
key = JWK(kty='oct', k=force_text(k))