module Rails
  module Controller
    module Testing
      module Integration
        http_verbs = %w(get post patch put head delete get_via_redirect post_via_redirect)
        http_verbs.push('xhr', 'xml_http_request') if Rails::VERSION::STRING.to_f <= 5.0

        http_verbs.each do |method|
          define_method(method) do |*args|
            reset_template_assertion
            super(*args)
          end
        end
      end
    end
  end
end

module Rails
  module Controller
    module Testing
      module Integration
        http_verbs = %w(get post patch put head delete get_via_redirect post_via_redirect)
        http_verbs.push('xhr', 'xml_http_request') if ActionPack.version < Gem::Version.new('5.1')

        http_verbs.each do |method|
          define_method(method) do |*args|
            reset_template_assertion
            super(*args)
          end
        end
      end
    end
  end
end

Use the right framework to check the version
Use the right framework to check the version
