app/models/sso/session.rb in doorkeeper_sso-0.0.4 vs app/models/sso/session.rb in doorkeeper_sso-0.1.0.pre.alpha

@@ -61,16 +61,16 @@
           error { "#update_master_with_grant : FAILED to update oauth_grant" }
         end
       end
 
       def update_master_with_access_token(grant_token, access_token)
-        oauth_grant  = Doorkeeper::AccessGrant.by_token(grant_token)
-        oauth_token  = Doorkeeper::AccessToken.by_token(access_token)
+        oauth_grant  = ::Doorkeeper::AccessGrant.by_token(grant_token)
+        oauth_token  = ::Doorkeeper::AccessToken.by_token(access_token)
         return false if oauth_token.blank? or oauth_grant.blank?
 
         master_sso_session = active.master.find_by!(access_grant_id: oauth_grant.id)
 
-        if master_sso_session.update_attribute(:access_token_id, oauth_token.id)
+        if master_sso_session.update_attributes(access_token_id: oauth_token.id, application_id: oauth_token.application_id)
           debug { "#register_access_token : #{master_sso_session.id} with Access Token ID #{oauth_token.id} which is #{oauth_token.token}" }
         else
           error { "#register_access_token : FAILED to update oauth_access_token_id" }
         end
         master_sso_session