dogweather/rspec-webservice_matchers

# frozen_string_literal: true
require 'web_test/util'


module RSpec
  module WebserviceMatchers
    module RedirectHelpers
      def redirect_failure_message(exception, status, actual_location, kind)
        return WebTest::Util.error_message(exception) if exception

        errors = []
        if redirect? status
          unless redirect? status, kind: kind
            errors << "received a #{kind_for(status)} redirect"
          end
          unless locations_match? expected, actual_location
            errors << "received location #{actual_location}"
          end
        else
          errors << "not a redirect: received status #{status}"
        end

        WebTest::Util.error_message(errors)
      end

      def redirects_correctly?(status, actual_loc, expected_loc, kind)
        redirect?(status, kind: kind) && locations_match?(expected_loc, actual_loc)
      end

      def redirect_result(url_or_domain_name)
        status, headers = WebTest::Util.head(url_or_domain_name)
        [status, headers['location']]
      end

      def locations_match?(expected, actual)
        actual =~ %r{#{expected}/?}
      end

      def redirect?(status, kind: nil)
        case kind
        when :permanent
          permanent_redirect?(status)
        when :temporary
          temp_redirect?(status)
        else
          temp_redirect?(status) || permanent_redirect?(status)
        end
      end

      def temp_redirect?(status)
        [302, 307].include?(status)
      end

      def permanent_redirect?(status)
        status == 301
      end

      def kind_for(status)
        {
          301 => 'permanent',
          302 => 'temporary',
          307 => 'temporary'
        }[status]
      end
    end
  end
end