Rake helpers extended

lib/tasks/graphiti.rake

@@ -1,62 +1,27 @@
 namespace :graphiti do
-  module Graphiti
-    module Rails
-      module RakeHelpers
-        extend RescueRegistry::RailsTestHelpers
-
-        module_function
-
-        def session
-          @session ||= ActionDispatch::Integration::Session.new(::Rails.application)
-        end
-
-        def setup_rails!
-          ::Rails.application.eager_load!
-          ::Rails.application.config.cache_classes = true
-          ::Rails.application.config.action_controller.perform_caching = false
-        end
-
-        def make_request(path, debug = false)
-          if path.split("/").length == 2
-            path = "#{ApplicationResource.endpoint_namespace}#{path}"
-          end
-          path << if path.include?("?")
-            "&cache=bust"
-          else
-            "?cache=bust"
-          end
-          path = "#{path}&debug=true" if debug
-          handle_request_exceptions do
-            headers = {
-              "Authorization": ENV['AUTHORIZATION_HEADER']
-            }.compact
-            session.get(path.to_s, headers: headers)
-          end
-          JSON.parse(session.response.body)
-        end
-      end
-    end
-  end
-
   desc "Execute request without web server."
   task :request, [:path, :debug] => [:environment] do |_, args|
-    Graphiti::Rails::RakeHelpers.setup_rails!
+    require_relative "rake_helpers"
+    extend Graphiti::Rails::RakeHelpers
+
     Graphiti.logger = Graphiti.stdout_logger
     Graphiti::Debugger.preserve = true
     require "pp"
     path, debug = args[:path], args[:debug]
     puts "Graphiti Request: #{path}"
-    json = Graphiti::Rails::RakeHelpers.make_request(path, debug)
+    json = make_request(path, debug)
     pp json
     Graphiti::Debugger.flush if debug
   end
 
   desc "Execute benchmark without web server."
   task :benchmark, [:path, :requests] => [:environment] do |_, args|
-    Graphiti::Rails::RakeHelpers.setup_rails!
+    require_relative "rake_helpers"
+    extend Graphiti::Rails::RakeHelpers
+
     took = Benchmark.ms {
       args[:requests].to_i.times do
-        Graphiti::Rails::RakeHelpers.make_request(args[:path])
+        make_request(args[:path])
       end
     }
     puts "Took: #{(took / args[:requests].to_f).round(2)}ms"

lib/tasks/rake_helpers.rb

@@ -0,0 +1,46 @@
+module Graphiti::Rails::RakeHelpers
+  extend RescueRegistry::RailsTestHelpers
+
+  def self.included(klass)
+    klass.class_eval do
+      setup_rails!
+    end
+  end
+
+  def self.extended(other)
+    other.instance_eval do
+      setup_rails!
+    end
+  end
+
+  module_function
+
+  def session
+    @session ||= ActionDispatch::Integration::Session.new(::Rails.application)
+  end
+
+  def setup_rails!
+    ::Rails.application.eager_load!
+    ::Rails.application.config.cache_classes = true
+    ::Rails.application.config.action_controller.perform_caching = false
+  end
+
+  def make_request(path, debug = false)
+    if path.split("/").length == 2
+      path = "#{ApplicationResource.endpoint_namespace}#{path}"
+    end
+    path << if path.include?("?")
+      "&cache=bust"
+    else
+      "?cache=bust"
+    end
+    path = "#{path}&debug=true" if debug
+    handle_request_exceptions do
+      headers = {
+        "Authorization": ENV['AUTHORIZATION_HEADER']
+      }.compact
+      session.get(path.to_s, headers: headers)
+    end
+    JSON.parse(session.response.body)
+  end
+end