components/pdf/browser/mock_url_loader_client.h

// Copyright 2021 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_PDF_BROWSER_MOCK_URL_LOADER_CLIENT_H_
#define COMPONENTS_PDF_BROWSER_MOCK_URL_LOADER_CLIENT_H_

#include "services/network/public/mojom/early_hints.mojom.h"
#include "services/network/public/mojom/url_loader.mojom.h"
#include "services/network/public/mojom/url_response_head.mojom.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace pdf {

class MockURLLoaderClient : public network::mojom::URLLoaderClient {
 public:
  MockURLLoaderClient();
  MockURLLoaderClient(const MockURLLoaderClient&) = delete;
  MockURLLoaderClient& operator=(const MockURLLoaderClient&) = delete;
  ~MockURLLoaderClient() override;

  MOCK_METHOD(void,
              OnReceiveEarlyHints,
              (network::mojom::EarlyHintsPtr early_hints),
              (override));
  MOCK_METHOD(void,
              OnReceiveResponse,
              (network::mojom::URLResponseHeadPtr head),
              (override));
  MOCK_METHOD(void,
              OnReceiveRedirect,
              (const net::RedirectInfo& redirect_info,
               network::mojom::URLResponseHeadPtr head),
              (override));
  MOCK_METHOD(void,
              OnUploadProgress,
              (int64_t current_position,
               int64_t total_size,
               OnUploadProgressCallback ack_callback),
              (override));
  MOCK_METHOD(void,
              OnReceiveCachedMetadata,
              (mojo_base::BigBuffer data),
              (override));
  MOCK_METHOD(void,
              OnTransferSizeUpdated,
              (int32_t transfer_size_diff),
              (override));
  MOCK_METHOD(void,
              OnStartLoadingResponseBody,
              (mojo::ScopedDataPipeConsumerHandle body),
              (override));
  MOCK_METHOD(void,
              OnComplete,
              (const network::URLLoaderCompletionStatus& status),
              (override));
};

}  // namespace pdf

#endif  // COMPONENTS_PDF_BROWSER_MOCK_URL_LOADER_CLIENT_H_