1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
// Copyright 2018 The Chromium Authors
// 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 "base/time/time.h"
#include "content/browser/web_package/signed_exchange_handler.h"
#include "net/base/hash_value.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.
// If |signature_expire_time| is a null Time, we treat as one day after now.
MockSignedExchangeHandlerParams(
const GURL& outer_url,
SignedExchangeLoadResult result,
net::Error error,
const GURL& inner_url,
const std::string& mime_type,
std::vector<std::pair<std::string, std::string>> response_headers,
const net::SHA256HashValue& header_integrity,
const base::Time& signature_expire_time = base::Time());
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::pair<std::string, std::string>> response_headers;
const net::SHA256HashValue header_integrity;
const base::Time signature_expire_time;
};
class MockSignedExchangeHandler final : public SignedExchangeHandler {
public:
MockSignedExchangeHandler(const MockSignedExchangeHandlerParams& params,
std::unique_ptr<net::SourceStream> body,
ExchangeHeadersCallback headers_callback);
MockSignedExchangeHandler(const MockSignedExchangeHandler&) = delete;
MockSignedExchangeHandler& operator=(const MockSignedExchangeHandler&) =
delete;
~MockSignedExchangeHandler() override;
bool GetSignedExchangeInfoForPrefetchCache(
PrefetchedSignedExchangeCacheEntry& entry) const override;
// The mocked, simulated responses need to include the prefix below, to help
// reinforce that SXG format doesn't really sniff as HTML (because of
// CBOR/binary encoding). Ensuring that tests more closely mimic real
// behavior seems desirable, even if in the long-term SXG might avoid
// `no-cors` mode (see https://crbug.com/1316660).
inline static const std::string kMockSxgPrefix = "MOCK-SXG-PREFIX: ";
private:
const net::SHA256HashValue header_integrity_;
const base::Time signature_expire_time_;
const GURL cert_url_;
const net::IPAddress cert_server_ip_address_;
};
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(const MockSignedExchangeHandlerFactory&) =
delete;
MockSignedExchangeHandlerFactory& operator=(
const MockSignedExchangeHandlerFactory&) = delete;
~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_;
};
} // namespace content
#endif // CONTENT_BROWSER_WEB_PACKAGE_MOCK_SIGNED_EXCHANGE_HANDLER_H_
|