module GdsApi::TestHelpers::AssetManager

def stub_any_asset_manager_call
  stub_request(:any, %r{\A#{ASSET_MANAGER_ENDPOINT}}).to_return(status: 200)
end

def stub_asset_manager_delete_asset(asset_id, body = {})
  stub_request(:delete, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}")
    .to_return(body: body.to_json, status: 200)
end

def stub_asset_manager_delete_asset_failure(asset_id)
  stub_request(:delete, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}").to_return(status: 500)
end

def stub_asset_manager_deletes_any_asset(body = {})
  stub_request(:delete, %r{\A#{ASSET_MANAGER_ENDPOINT}/assets})
    .to_return(body: body.to_json, status: 200)
end

def stub_asset_manager_does_not_have_a_whitehall_asset(legacy_url_path)
  response = {
    "_response_info" => { "status" => "not found" },
  }
  stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/whitehall_assets/#{legacy_url_path}")
    .to_return(body: response.to_json, status: 404)
end

def stub_asset_manager_does_not_have_an_asset(id)
  response = {
    "_response_info" => { "status" => "not found" },
  }
  stub_request(:any, "#{ASSET_MANAGER_ENDPOINT}/assets/#{id}")
    .to_return(body: response.to_json, status: 404)
end

def stub_asset_manager_has_a_whitehall_asset(legacy_url_path, atts)
  response = atts.merge("_response_info" => { "status" => "ok" })
  stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/whitehall_assets/#{legacy_url_path}")
    .to_return(body: response.to_json, status: 200)
end

def stub_asset_manager_has_an_asset(id, atts)
  response = atts.merge("_response_info" => { "status" => "ok" })
  stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/assets/#{id}")
    .to_return(body: response.to_json, status: 200)
end

def stub_asset_manager_isnt_available
  stub_request(:any, %r{\A#{ASSET_MANAGER_ENDPOINT}}).to_return(status: 503)
end

def stub_asset_manager_receives_an_asset(response_url = {})
  stub_request(:post, "#{ASSET_MANAGER_ENDPOINT}/assets").to_return do
    if response_url.is_a?(String)
      file_url = response_url
    else
      options = {
        id: SecureRandom.uuid,
        filename: SecureRandom.hex(8),
      }.merge(response_url)
      file_url = "#{ASSET_MANAGER_ENDPOINT}/media/#{options[:id]}/#{options[:filename]}"
    end
    { body: { file_url: file_url }.to_json, status: 200 }
  end
end

def stub_asset_manager_update_asset(asset_id, body = {})
  stub_request(:put, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}")
    .to_return(body: body.to_json, status: 200)
end

def stub_asset_manager_update_asset_failure(asset_id)
  stub_request(:put, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}").to_return(status: 500)
end

def stub_asset_manager_updates_any_asset(body = {})
  stub_request(:put, %r{\A#{ASSET_MANAGER_ENDPOINT}/assets})
    .to_return(body: body.to_json, status: 200)
end

def stub_asset_manager_upload_failure
  stub_request(:post, "#{ASSET_MANAGER_ENDPOINT}/assets").to_return(status: 500)
end