[LOCATION] Request eu checks if helper method defined

Author: ssunday

lib/action_pack/cloudfront/location_headers.rb

@@ -9,7 +9,9 @@ module LocationHeaders
                               |RS|SK|SI|ES|SE|CH|UA|GB|VA|RS)$/ix
 
       included do
-        helper_method :request_eu?
+        if respond_to?(:helper_method)
+          helper_method :request_eu?
+        end
       end
 
       private

lib/action_pack/cloudfront/railtie.rb

@@ -6,6 +6,7 @@ class Railtie < ::Rails::Railtie
 
       config.action_pack_cloudfront = ActiveSupport::OrderedOptions.new
       config.action_pack_cloudfront.load_proxies = !::Rails.env.test? && !::Rails.env.development?
+      config.action_pack_cloudfront.include_location_headers = true
 
       config.before_initialize do |app|
         if app.config.action_pack_cloudfront.load_proxies
@@ -17,9 +18,10 @@ class Railtie < ::Rails::Railtie
       end
 
       config.to_prepare do
-        ::ApplicationController.send :include, ActionPack::Cloudfront::LocationHeaders
+        if app.config.action_pack_cloudfront.include_location_headers
+          ::ApplicationController.send :include, ActionPack::Cloudfront::LocationHeaders
+        end
       end
-
     end
   end
 end

test/actionpack/cloudfront/location_headers_test.rb

@@ -0,0 +1,16 @@
+require 'test_helper'
+
+class LocationHeadersTest < ActiveSupport::TestCase
+  class ApplicationController < ActionController::Base
+    include ActionPack::Cloudfront::LocationHeaders
+  end
+
+  class ApiController < ActionController::API
+    include ActionPack::Cloudfront::LocationHeaders
+  end
+
+  test '#request_eu? - includes when has helper method' do
+    assert_includes ApplicationController._helper_methods, :request_eu?
+    assert_nil ApiController.try(:_helper_methods)
+  end
+end