123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- // 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 NET_CERT_MOCK_CERT_NET_FETCHER_H_
- #define NET_CERT_MOCK_CERT_NET_FETCHER_H_
- #include "net/cert/cert_net_fetcher.h"
- #include "testing/gmock/include/gmock/gmock.h"
- #include "third_party/boringssl/src/include/openssl/base.h"
- namespace net {
- // MockCertNetFetcher is an implementation of CertNetFetcher for testing.
- class MockCertNetFetcher : public CertNetFetcher {
- public:
- MockCertNetFetcher();
- MOCK_METHOD0(Shutdown, void());
- MOCK_METHOD3(FetchCaIssuers,
- std::unique_ptr<Request>(const GURL& url,
- int timeout_milliseconds,
- int max_response_bytes));
- MOCK_METHOD3(FetchCrl,
- std::unique_ptr<Request>(const GURL& url,
- int timeout_milliseconds,
- int max_response_bytes));
- MOCK_METHOD3(FetchOcsp,
- std::unique_ptr<Request>(const GURL& url,
- int timeout_milliseconds,
- int max_response_bytes));
- protected:
- // Protected since CertNetFetcher is refcounted.
- ~MockCertNetFetcher() override;
- };
- // MockCertNetFetcherRequest gives back the indicated error and bytes.
- class MockCertNetFetcherRequest : public CertNetFetcher::Request {
- public:
- MockCertNetFetcherRequest(Error error, std::vector<uint8_t> bytes);
- ~MockCertNetFetcherRequest() override;
- // Creates a CertNetFetcher::Request that completes with an error.
- static std::unique_ptr<CertNetFetcher::Request> Create(Error error);
- // Creates a CertNetFetcher::Request that completes with OK error code and
- // the specified bytes.
- static std::unique_ptr<CertNetFetcher::Request> Create(
- std::vector<uint8_t> bytes);
- // Creates a CertNetFetcher::Request that completes with OK error code and
- // the specified CRYPTO_BUFFER data.
- static std::unique_ptr<CertNetFetcher::Request> Create(
- const CRYPTO_BUFFER* buffer);
- void WaitForResult(Error* error, std::vector<uint8_t>* bytes) override;
- private:
- Error error_;
- std::vector<uint8_t> bytes_;
- bool did_consume_result_ = false;
- };
- } // namespace net
- #endif // NET_CERT_MOCK_CERT_NET_FETCHER_H_
|