Allow new access token class extending AccessToken class to get access token instance of the its own class when refreshed

Author: styd

lib/oauth2/access_token.rb

@@ -79,11 +79,11 @@ def expired?
     #
     # @return [AccessToken] a new AccessToken
     # @note options should be carried over to the new AccessToken
-    def refresh!(params = {})
+    def refresh!(params = {}, access_token_opts = {}, access_token_class = self.class)
       raise('A refresh_token is not available') unless refresh_token
       params[:grant_type] = 'refresh_token'
       params[:refresh_token] = refresh_token
-      new_token = @client.get_token(params)
+      new_token = @client.get_token(params, access_token_opts, access_token_class)
       new_token.options = options
       new_token.refresh_token = refresh_token unless new_token.refresh_token
       new_token

spec/oauth2/access_token_spec.rb

@@ -154,13 +154,22 @@ def assert_initialized_token(target) # rubocop:disable Metrics/AbcSize
                                      :expires_in     => 600,
                                      :param_name     => 'o_param')
     end
+    let(:new_access) do
+      NewAccessToken = Class.new(AccessToken)
+      NewAccessToken.new(client, token, :refresh_token  => 'abaca')
+    end
 
     it 'returns a refresh token with appropriate values carried over' do
       refreshed = access.refresh!
       expect(access.client).to eq(refreshed.client)
       expect(access.options[:param_name]).to eq(refreshed.options[:param_name])
     end
 
+    it 'returns a refresh token of the same access token class' do
+      refreshed = new_access.refresh!
+      expect(new_access.class).to eq(refreshed.class)
+    end
+
     context 'with a nil refresh_token in the response' do
       let(:refresh_body) { MultiJson.encode(:access_token => 'refreshed_foo', :expires_in => 600, :refresh_token => nil) }