Minor fine-tuning

hjmikkon · hjmikkon · commit dd82bcf2ae0b · 2018-07-30T21:39:16.000+03:00
diff --git a/src/main/java/org/oidc/service/oidc/Registration.java b/src/main/java/org/oidc/service/oidc/Registration.java
@@ -67,10 +67,6 @@ public Registration(ServiceContext serviceContext, State state, ServiceConfig se
   @Override
   protected void doUpdateServiceContext(Message response, String stateKey)
       throws MissingRequiredAttributeException, ValueException, InvalidClaimException {
-    if (response == null || !(response instanceof RegistrationResponse)) {
-      throw new ValueException("Unexpected response message type, should be RegistrationResponse");
-    }
-    this.responseMessage = response;
     if (!response.getClaims().containsKey("token_endpoint_auth_method")) {
       response.getClaims().put("token_endpoint_auth_method", "client_secret_basic");
     }
@@ -81,6 +77,7 @@ protected void doUpdateServiceContext(Message response, String stateKey)
     getServiceContext()
         .setRegistrationAccessToken((String) response.getClaims().get("registration_access_token"));
     getServiceContext().setRegistrationResponse((RegistrationResponse) response);
+    this.responseMessage = response;
   }
 
   @Override
