[page_load_metrics] Add a NavigationThrottle for richer abort metrics

content/test/content_browser_test_utils_internal.h

 // Copyright 2014 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_CONTENT_BROWSER_TEST_UTILS_INTERNAL_H_
 #define CONTENT_TEST_CONTENT_BROWSER_TEST_UTILS_INTERNAL_H_
 
 // A collection of functions designed for use with content_shell based browser
 // tests internal to the content/ module.
 // Note: If a function here also works with browser_tests, it should be in
 // the content public API.
 
 #include <string>
 #include <vector>
 
 #include "base/files/file_path.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "cc/surfaces/surface_id.h"
-#include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/resource_dispatcher_host_delegate.h"
 #include "content/public/browser/web_contents_delegate.h"
-#include "content/public/browser/web_contents_observer.h"
+#include "content/public/test/browser_test_utils.h"
 #include "url/gurl.h"
 
 namespace cc {
 class SurfaceManager;
 }
 
 namespace content {
 
 class FrameTreeNode;
 class MessageLoopRunner;
@@ skipping 76 matching lines @@
   void RequestBeginning(
       net::URLRequest* request,
       content::ResourceContext* resource_context,
       content::AppCacheService* appcache_service,
       ResourceType resource_type,
       ScopedVector<content::ResourceThrottle>* throttles) override;
 
   GURL url_;
 };
 
-// This class can be used to pause and resume navigations, based on a URL
-// match. Note that it only keeps track of one navigation at a time.
-// Navigations are paused automatically before hitting the network, and are
-// resumed automatically if a Wait method is called for a future event.
-class TestNavigationManager : public WebContentsObserver {
- public:
-  // Monitors notifications within the given frame tree node. Use the other
-  // constructor if the manager should monitor all frames, which is equivalent
-  // to passing kFrameTreeNodeInvalidId for |frame_tree_node_id|.
-  TestNavigationManager(int frame_tree_node_id,
-                        WebContents* web_contents,
-                        const GURL& url);
-
-  // Monitors any frame in WebContents.
-  TestNavigationManager(WebContents* web_contents, const GURL& url);
-
-  ~TestNavigationManager() override;
-
-  // Waits until the navigation request is ready to be sent to the network
-  // stack.
-  void WaitForWillStartRequest();
-
-  // Waits until the navigation has been finished. Will automatically resume
-  // navigations paused before this point.
-  void WaitForNavigationFinished();
-
- private:
-  // WebContentsObserver:
-  void DidStartNavigation(NavigationHandle* handle) override;
-  void DidFinishNavigation(NavigationHandle* handle) override;
-
-  // Called when the NavigationThrottle pauses the navigation in
-  // WillStartRequest.
-  void OnWillStartRequest();
-
-  // Resumes the navigation.
-  void ResumeNavigation();
-
-  // If this member is not |kFrameTreeNodeInvalidId|, notifications are filtered
-  // so only this frame is monitored.
-  int filtering_frame_tree_node_id_;
-
-  const GURL url_;
-  bool navigation_paused_;
-  NavigationHandle* handle_;
-  scoped_refptr<MessageLoopRunner> will_start_loop_runner_;
-  scoped_refptr<MessageLoopRunner> did_finish_loop_runner_;
-
-  base::WeakPtrFactory<TestNavigationManager> weak_factory_;
-};
-
 // Helper class to assist with hit testing surfaces in multiple processes.
 // WaitForSurfaceReady() will only return after a Surface from |target_view|
 // has been composited in the top-level frame's Surface. At that point,
 // browser process hit testing to target_view's Surface can succeed.
 class SurfaceHitTestReadyNotifier {
  public:
   SurfaceHitTestReadyNotifier(RenderWidgetHostViewChildFrame* target_view);
   ~SurfaceHitTestReadyNotifier() {}
 
   void WaitForSurfaceReady();
@@ skipping 20 matching lines @@
                       const FileChooserParams& params) override;
 
   // Whether the file dialog was shown.
   bool file_chosen() { return file_chosen_; }
 
  private:
   base::FilePath file_;
   bool file_chosen_;
 };
 
+// This class is a TestNavigationManager that only monitors notifications within
+// the given frame tree node.
+class PerFrameTestNavigationManager : public TestNavigationManager {

[nasko, 2016/08/02 16:22:16]

nit: Drop the "Per" prefix. We usually prefix something with "Frame" when it is
per frame, so it will be consistent with naming of other classes.

[Charlie Harrison, 2016/08/02 16:47:13]

Done.

+ public:
+  PerFrameTestNavigationManager(int frame_tree_node_id,
+                                WebContents* web_contents,
+                                const GURL& url);
+ private:
+  // TestNavigationManager:
+  bool ShouldMonitorNavigation(NavigationHandle* handle) override;
+
+  // Notifications are filtered so only this frame is monitored.
+  int filtering_frame_tree_node_id_;
+
+  DISALLOW_COPY_AND_ASSIGN(PerFrameTestNavigationManager);
+};
+
 }  // namespace content
 
 #endif  // CONTENT_TEST_CONTENT_BROWSER_TEST_UTILS_INTERNAL_H_