PlzNavigate: make all ResourceDispatcherHostTests pass

content/test/test_navigation_url_loader_delegate.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 CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_DELEGATE_H_
+#define CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_DELEGATE_H_
+
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/time/time.h"
+#include "content/browser/loader/navigation_url_loader_delegate.h"
+#include "net/url_request/redirect_info.h"
+
+namespace base {
+class RunLoop;
+}
+
+namespace content {
+
+class StreamHandle;
+struct ResourceResponse;
+
+// PlzNavigate
+// Test implementation of NavigationURLLoaderDelegate to monitor navigation
+// progress in the network stack.
+class TestNavigationURLLoaderDelegate : public NavigationURLLoaderDelegate {
+ public:
+  TestNavigationURLLoaderDelegate();
+  ~TestNavigationURLLoaderDelegate() override;
+
+  const net::RedirectInfo& redirect_info() const { return redirect_info_; }
+  ResourceResponse* redirect_response() const {
+    return redirect_response_.get();
+  }
+  ResourceResponse* response() const { return response_.get(); }
+  StreamHandle* body() const { return body_.get(); }
+  int net_error() const { return net_error_; }
+  int on_request_handled_counter() const { return on_request_handled_counter_; }
+
+  // Waits for various navigation events.
+  // Note: if the event already happened, the functions will hang.
+  // TODO(clamy): Make the functions not hang if they are called after the
+  // event happened.
+  void WaitForRequestRedirected();
+  void WaitForResponseStarted();
+  void WaitForRequestFailed();
+  void WaitForRequestStarted();
+
+  void ReleaseBody();
+
+  // NavigationURLLoaderDelegate implementation.
+  void OnRequestRedirected(
+      const net::RedirectInfo& redirect_info,
+      const scoped_refptr<ResourceResponse>& response) override;
+  void OnResponseStarted(const scoped_refptr<ResourceResponse>& response,
+                         scoped_ptr<StreamHandle> body) override;
+  void OnRequestFailed(bool in_cache, int net_error) override;
+  void OnRequestStarted(base::TimeTicks timestamp) override;
+
+ private:
+  net::RedirectInfo redirect_info_;
+  scoped_refptr<ResourceResponse> redirect_response_;
+  scoped_refptr<ResourceResponse> response_;
+  scoped_ptr<StreamHandle> body_;
+  int net_error_;
+  int on_request_handled_counter_;
+
+  scoped_ptr<base::RunLoop> request_redirected_;
+  scoped_ptr<base::RunLoop> response_started_;
+  scoped_ptr<base::RunLoop> request_failed_;
+  scoped_ptr<base::RunLoop> request_started_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestNavigationURLLoaderDelegate);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_DELEGATE_H