Projet

Général

Profil

0006-idp_oidc-test-templated-claims-37884.patch

Paul Marillonnet, 20 février 2020 16:53

Télécharger (3,67 ko)

Voir les différences:

Subject: [PATCH 6/6] idp_oidc: test templated claims (#37884)

 tests/test_idp_oidc.py | 74 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
tests/test_idp_oidc.py
1115 1115
    assert 'family_name' not in user_info
1116 1116

  
1117 1117

  
1118
def test_claim_templated(oidc_settings, normal_oidc_client, simple_user, app):
1119
    oidc_settings.A2_IDP_OIDC_SCOPES = ['openid', 'profile', 'email']
1120
    OIDCClaim.objects.filter(
1121
            client=normal_oidc_client, name='given_name').delete()
1122
    OIDCClaim.objects.filter(
1123
            client=normal_oidc_client, name='family_name').delete()
1124
    claim1 = OIDCClaim.objects.create(
1125
            client=normal_oidc_client,
1126
            name='given_name',
1127
            value='{{ django_user_first_name|add:"ounet" }}',
1128
            is_templated=True,
1129
            scopes='profile')
1130
    claim2 = OIDCClaim.objects.create(
1131
            client=normal_oidc_client,
1132
            name='family_name',
1133
            value='{{ "Von der "|add:django_user_last_name }}',
1134
            is_templated=True,
1135
            scopes='profile')
1136
    normal_oidc_client.authorization_flow = normal_oidc_client.FLOW_AUTHORIZATION_CODE
1137
    normal_oidc_client.authorization_mode = normal_oidc_client.AUTHORIZATION_MODE_NONE
1138
    normal_oidc_client.save()
1139

  
1140
    utils.login(app, simple_user)
1141

  
1142
    oidc_client = normal_oidc_client
1143
    redirect_uri = oidc_client.redirect_uris.split()[0]
1144

  
1145
    params = {
1146
        'client_id': oidc_client.client_id,
1147
        'scope': 'openid email profile',
1148
        'redirect_uri': redirect_uri,
1149
        'state': 'xxx',
1150
        'nonce': 'yyy',
1151
        'response_type': 'code',
1152
    }
1153

  
1154
    def sso():
1155
        authorize_url = make_url('oidc-authorize', params=params)
1156

  
1157
        response = app.get(authorize_url)
1158
        location = urlparse.urlparse(response['Location'])
1159
        query = urlparse.parse_qs(location.query)
1160
        code = query['code'][0]
1161

  
1162
        token_url = make_url('oidc-token')
1163
        response = app.post(token_url, params={
1164
            'grant_type': 'authorization_code',
1165
            'code': code,
1166
            'redirect_uri': oidc_client.redirect_uris.split()[0],
1167
        }, headers=client_authentication_headers(oidc_client))
1168
        access_token = response.json['access_token']
1169
        id_token = response.json['id_token']
1170

  
1171
        key = JWK(kty='oct', k=base64.b64encode(oidc_client.client_secret.encode('utf-8')))
1172
        jwt = JWT(jwt=id_token, key=key)
1173
        claims = json.loads(jwt.claims)
1174

  
1175
        user_info_url = make_url('oidc-user-info')
1176
        response = app.get(user_info_url, headers=bearer_authentication_headers(access_token))
1177
        return claims, response.json
1178

  
1179
    claims, user_info = sso()
1180

  
1181
    assert claims['given_name'].endswith('ounet')
1182
    assert claims['given_name'].startswith(simple_user.first_name)
1183
    assert claims['family_name'].startswith('Von der')
1184
    assert claims['family_name'].endswith(simple_user.last_name)
1185

  
1186
    assert user_info['given_name'].endswith('ounet')
1187
    assert user_info['given_name'].startswith(simple_user.first_name)
1188
    assert user_info['family_name'].startswith('Von der')
1189
    assert user_info['family_name'].endswith(simple_user.last_name)
1190

  
1191

  
1118 1192
def test_client_is_valid_redirect_uri():
1119 1193
    client = OIDCClient(redirect_uris='''http://example.com
1120 1194
http://example2.com/
1121
-