AC::TestResponse becomes AD::TestResponse in Rails v5

kymmt90 · Feb 12, 2017 · 3ffc3ec · 3ffc3ec
1 parent 1289157
commit 3ffc3ec
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/spec/controllers/authorizations_controller_spec.rb b/spec/controllers/authorizations_controller_spec.rb
@@ -3,11 +3,22 @@
 describe Doorkeeper::AuthorizationsController, 'implicit grant flow' do
   include AuthorizationRequestHelper
 
-  class ActionController::TestResponse
-    def query_params
-      @_query_params ||= begin
-        fragment = URI.parse(location).fragment
-        Rack::Utils.parse_query(fragment)
+  if Rails::VERSION::MAJOR == 5
+    class ActionDispatch::TestResponse
+      def query_params
+        @_query_params ||= begin
+          fragment = URI.parse(location).fragment
+          Rack::Utils.parse_query(fragment)
+        end
+      end
+    end
+  else
+    class ActionController::TestResponse
+      def query_params
+        @_query_params ||= begin
+          fragment = URI.parse(location).fragment
+          Rack::Utils.parse_query(fragment)
+        end
       end
     end
   end