require 'webmock'

module GdsApi
  module TestHelpers
    module ContentApi
      class ArtefactStub
        include WebMock::API
        # This is ugly, but the nicest way we found to get access to artefact_for_slug
        include GdsApi::TestHelpers::ContentApi

        attr_accessor :slug, :query_parameters, :response_body, :response_status

        def initialize(slug)
          @slug = slug
          @query_parameters = {}
          @response_body = artefact_for_slug(slug)
          @response_status = 200
        end
        
        def with_query_parameters(hash)
          @query_parameters = hash
          self
        end

        def with_response_body(response_body)
          @response_body = response_body
          self
        end

        def with_response_status(response_status)
          @response_status = response_status
          self
        end

        # Nothing is stubbed until this is called
        def stub
          stub_request(:get, url_without_query)
              .with(query: hash_including(comparable_query_params))
              .to_return(status: @response_status, body: @response_body.to_json)
        end
        
        private
          def url_without_query
            "#{CONTENT_API_ENDPOINT}/#{CGI.escape(slug)}.json"
          end

          # Ensure that all keys and values are strings 
          # because Webmock doesn't handle symbols
          def comparable_query_params
            @query_parameters.each_with_object({}) do |(k,v),hash| 
              hash[k.to_s] = v.nil? ? v : v.to_s
            end
          end
      end
    end
  end
end