[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

@@ -19,7 +19,6 @@ class Railtie < ::Rails::Railtie
       config.to_prepare do
         ::ApplicationController.send :include, ActionPack::Cloudfront::LocationHeaders
       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