Headless utility classes for making deterministic protocol handlers

headless/public/util/testing/generic_url_request_mocks.h

+// Copyright 2016 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 HEADLESS_PUBLIC_UTIL_TESTING_GENERIC_URL_REQUEST_MOCKS_H_
+#define HEADLESS_PUBLIC_UTIL_TESTING_GENERIC_URL_REQUEST_MOCKS_H_
+
+#include <memory>
+#include <string>
+
+#include "base/macros.h"
+#include "headless/public/util/generic_url_request_job.h"
+#include "headless/public/util/testing/generic_url_request_mocks.h"
+#include "net/cookies/cookie_store.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_job_factory.h"
+
+namespace net {
+class URLRequestJob;
+}  // namespace net
+
+namespace htmlrender_webkit_headless_proto {
+class Resource;
+}  // htmlrender_webkit_headless_proto net
+
+namespace headless {
+
+class MockGenericURLRequestJobDelegate : public GenericURLRequestJob::Delegate {
+ public:
+  MockGenericURLRequestJobDelegate();
+  ~MockGenericURLRequestJobDelegate() override;
+
+  bool BlockOrRewriteRequest(
+      const GURL& url,
+      const std::string& referrer,
+      GenericURLRequestJob::RewriteCallback callback) override;
+
+  const GenericURLRequestJob::HttpResponse* MaybeMatchResource(
+      const GURL& url,
+      const net::HttpRequestHeaders& request_headers) override;
+
+  void OnResourceLoadComplete(const GURL& final_url,
+                              const std::string& mime_type,
+                              int http_response_code) override;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockGenericURLRequestJobDelegate);
+};
+
+// TODO(alexclarke): We may be able to replace this with the CookieMonster.
+class MockCookieStore : public net::CookieStore {
+ public:
+  MockCookieStore();
+  ~MockCookieStore() override;
+
+  // net::CookieStore implementation:
+  void SetCookieWithOptionsAsync(const GURL& url,
+                                 const std::string& cookie_line,
+                                 const net::CookieOptions& options,
+                                 const SetCookiesCallback& callback) override;
+
+  void SetCookieWithDetailsAsync(const GURL& url,
+                                 const std::string& name,
+                                 const std::string& value,
+                                 const std::string& domain,
+                                 const std::string& path,
+                                 base::Time creation_time,
+                                 base::Time expiration_time,
+                                 base::Time last_access_time,
+                                 bool secure,
+                                 bool http_only,
+                                 net::CookieSameSite same_site,
+                                 bool enforce_strict_secure,
+                                 net::CookiePriority priority,
+                                 const SetCookiesCallback& callback) override;
+
+  void GetCookiesWithOptionsAsync(const GURL& url,
+                                  const net::CookieOptions& options,
+                                  const GetCookiesCallback& callback) override;
+
+  void GetCookieListWithOptionsAsync(
+      const GURL& url,
+      const net::CookieOptions& options,
+      const GetCookieListCallback& callback) override;
+
+  void GetAllCookiesAsync(const GetCookieListCallback& callback) override;
+
+  void DeleteCookieAsync(const GURL& url,
+                         const std::string& cookie_name,
+                         const base::Closure& callback) override;
+
+  void DeleteCanonicalCookieAsync(const net::CanonicalCookie& cookie,
+                                  const DeleteCallback& callback) override;
+
+  void DeleteAllCreatedBetweenAsync(const base::Time& delete_begin,
+                                    const base::Time& delete_end,
+                                    const DeleteCallback& callback) override;
+
+  void DeleteAllCreatedBetweenWithPredicateAsync(
+      const base::Time& delete_begin,
+      const base::Time& delete_end,
+      const CookiePredicate& predicate,
+      const DeleteCallback& callback) override;
+
+  void DeleteSessionCookiesAsync(const DeleteCallback&) override;
+
+  void FlushStore(const base::Closure& callback) override;
+
+  void SetForceKeepSessionState() override;
+
+  std::unique_ptr<CookieChangedSubscription> AddCallbackForCookie(
+      const GURL& url,
+      const std::string& name,
+      const CookieChangedCallback& callback) override;
+
+  bool IsEphemeral() override;
+
+  net::CookieList* cookies() { return &cookies_; }
+
+ private:
+  void SendCookies(const GURL& url,
+                   const net::CookieOptions& options,
+                   const GetCookieListCallback& callback);
+
+  net::CookieList cookies_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockCookieStore);
+};
+
+class MockURLRequestDelegate : public net::URLRequest::Delegate {
+ public:
+  MockURLRequestDelegate();
+  ~MockURLRequestDelegate() override;
+
+  void OnResponseStarted(net::URLRequest* request) override;
+  void OnReadCompleted(net::URLRequest* request, int bytes_read) override;
+  const std::string& response_data() const;
+  const net::IOBufferWithSize* metadata() const;
+
+ private:
+  std::string response_data_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockURLRequestDelegate);
+};
+
+}  // namespace headless
+
+#endif  // HEADLESS_PUBLIC_UTIL_TESTING_GENERIC_URL_REQUEST_MOCKS_H_