def test_handle_registration_response(self):
federation_key = sym_key()
op_root_key = rsa_key()
op_intermediate_key = rsa_key()
op_signed_intermediate_key = JWS(json.dumps(op_intermediate_key.serialize(private=False)),
alg=op_root_key.alg).sign_compact(keys=[op_root_key])
op_software_statement = Federation(federation_key).create_software_statement(
dict(issuer=ISSUER, root_key=op_root_key.serialize(private=False),
scopes_supported=["openid", "test_scope"]))
rp = RP(None, sym_key(), [], [federation_key], None)
signed_jwks_uri = "{}/signed_jwks".format(ISSUER)
# fake provider discovery
rp.client.provider_info = FederationProviderConfigurationResponse(
**dict(signing_key=op_signed_intermediate_key, signed_jwks_uri=signed_jwks_uri,
issuer=ISSUER))
# signed_jwks_uri
expected_kid = "OP key 1"
keys = [RSAKey(key=RSA.generate(1024), kid=expected_kid).serialize(private=False)]
jwks = json.dumps(dict(keys=keys))
jws = JWS(jwks, alg=op_intermediate_key.alg).sign_compact(keys=[op_intermediate_key])
responses.add(responses.GET, signed_jwks_uri, body=jws, status=200,
content_type="application/jose")
resp_args = dict(provider_software_statement=op_software_statement, client_id="foo")
reg_resp = FederationRegistrationResponse(**resp_args)
rp._handle_registration_response(reg_resp)
assert set(rp.client.provider_info["scopes_supported"]) == {"openid", "test_scope"}
assert rp.client.client_id == "foo"
assert rp.client.keyjar[ISSUER][0].keys()[0].kid == expected_kid
def test_handle_registration_response_fail_when_wrong_software_statement(self):
rp = RP(None, sym_key(), [], None, None)
rp.client.provider_info = FederationProviderConfigurationResponse(
**dict(signing_key="whatever")) # fake provider discovery
resp_args = dict(provider_software_statement="abcdef")
reg_resp = FederationRegistrationResponse(**resp_args)
with pytest.raises(OIDCFederationError) as exc:
rp._handle_registration_response(reg_resp)
assert "software statement" in str(exc.value)
