module Pact::Provider::TestMethods

def get_provider_state provider_state_name, consumer
  unless provider_state = ProviderState.get(provider_state_name, :for => consumer)
    extra = consumer ? " for consumer \"#{consumer}\"" : ""
    error_msg = <<-eos
not find a provider state named \"#{provider_state_name}\"#{extra}.
ou required the provider states file for this consumer in your pact_helper.rb?
the name in the Pact.provider_states_for definition is exactly \"#{consumer}\"
    raise error_msg
  end
  provider_state
end

def parse_entity_from_response response
  case response.headers['Content-Type']
  when /json/
    JSON.load(response.body)
  else
    response.body
  end
end

def replay_interaction interaction
  request = interaction.request
  path = request_path(request)
  args = [path, request_body(request)]
  if !request.headers.nil?
    args << request_headers(request)
  end
  logger.debug "Sending #{request.method} with #{args}"
  self.send(request.method, *args)
end

def request_body request
  body = request.body
  if !body.nil? #TODO - write test for this where this is a NullExpectation
    body = JSON.dump(Pact::Reification.from_term(body))
  else
    body = ""
  end
end

def request_headers request
  request_headers = {}
  request.headers.each do |key, value|
    key = key.upcase
    if key.match(/CONTENT.TYPE/)
      request_headers['CONTENT_TYPE'] = value
    else
      request_headers['HTTP_' + key.to_s] = value
    end
  end
  request_headers
end

def request_path request
  path = request.path
  query = request.query
  if query && !query.empty?
    query = query.generate if query.kind_of? Pact::Term
    path += "?" + query
  end
  path
end

def set_up_provider_state provider_state_name, consumer
  if provider_state_name
    get_provider_state(provider_state_name, consumer).set_up
  end
end

def tear_down_provider_state provider_state_name, consumer
  if provider_state_name
    get_provider_state(provider_state_name, consumer).tear_down
  end
end