content/browser/web_package/mock_signed_exchange_handler.h

// Copyright 2018 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 CONTENT_BROWSER_WEB_PACKAGE_MOCK_SIGNED_EXCHANGE_HANDLER_H_
#define CONTENT_BROWSER_WEB_PACKAGE_MOCK_SIGNED_EXCHANGE_HANDLER_H_

#include <string>
#include <vector>

#include "content/browser/web_package/signed_exchange_handler.h"
#include "url/gurl.h"

namespace content {

class SignedExchangeCertFetcherFactory;

class MockSignedExchangeHandlerParams {
 public:
  // |mime_type| and |response_headers| are ignored if |error| is not net::OK.
  MockSignedExchangeHandlerParams(const GURL& outer_url,
                                  SignedExchangeLoadResult result,
                                  net::Error error,
                                  const GURL& inner_url,
                                  const std::string& mime_type,
                                  std::vector<std::string> response_headers);
  MockSignedExchangeHandlerParams(const MockSignedExchangeHandlerParams& other);
  ~MockSignedExchangeHandlerParams();
  const GURL outer_url;
  const SignedExchangeLoadResult result;
  const net::Error error;
  const GURL inner_url;
  const std::string mime_type;
  const std::vector<std::string> response_headers;
};

class MockSignedExchangeHandler final : public SignedExchangeHandler {
 public:
  MockSignedExchangeHandler(const MockSignedExchangeHandlerParams& params,
                            std::unique_ptr<net::SourceStream> body,
                            ExchangeHeadersCallback headers_callback);
  ~MockSignedExchangeHandler();

 private:
  DISALLOW_COPY_AND_ASSIGN(MockSignedExchangeHandler);
};

class MockSignedExchangeHandlerFactory final
    : public SignedExchangeHandlerFactory {
 public:
  using ExchangeHeadersCallback =
      SignedExchangeHandler::ExchangeHeadersCallback;

  // Creates a factory that creates SignedExchangeHandler which fires
  // a headers callback with the matching MockSignedExchangeHandlerParams.
  MockSignedExchangeHandlerFactory(
      std::vector<MockSignedExchangeHandlerParams> params_list);
  ~MockSignedExchangeHandlerFactory() override;

  std::unique_ptr<SignedExchangeHandler> Create(
      const GURL& outer_url,
      std::unique_ptr<net::SourceStream> body,
      ExchangeHeadersCallback headers_callback,
      std::unique_ptr<SignedExchangeCertFetcherFactory> cert_fetcher_factory)
      override;

 private:
  const std::vector<MockSignedExchangeHandlerParams> params_list_;

  DISALLOW_COPY_AND_ASSIGN(MockSignedExchangeHandlerFactory);
};

}  // namespace content

#endif  // CONTENT_BROWSER_WEB_PACKAGE_MOCK_SIGNED_EXCHANGE_HANDLER_H_