Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(878)

Unified Diff: content/public/test/browser_test_utils.h

Issue 2132603002: [page_load_metrics] Add a NavigationThrottle for richer abort metrics (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: nasko@ nits Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: content/public/test/browser_test_utils.h
diff --git a/content/public/test/browser_test_utils.h b/content/public/test/browser_test_utils.h
index 891d40bc183c1365df08be859677b5913bd90944..964cf670722fe525c0cf0d91b80292d103c2d9f5 100644
--- a/content/public/test/browser_test_utils.h
+++ b/content/public/test/browser_test_utils.h
@@ -61,6 +61,7 @@ namespace content {
class BrowserContext;
class MessageLoopRunner;
+class NavigationHandle;
class RenderViewHost;
class RenderWidgetHost;
class WebContents;
@@ -592,6 +593,57 @@ class FrameFocusedObserver {
DISALLOW_COPY_AND_ASSIGN(FrameFocusedObserver);
};
+// 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.
+// Note: This class is one time use only! After it successfully tracks a
+// navigation it will ignore all subsequent navigations. Explicitly create
+// mutliple instances of this class if you want to pause multiple navigations.
+class TestNavigationManager : public WebContentsObserver {
+ public:
+ // 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. Returns false if the request was aborted before starting.
+ WARN_UNUSED_RESULT bool WaitForWillStartRequest();
+
+ // Waits until the navigation has been finished. Will automatically resume
+ // navigations paused before this point.
+ void WaitForNavigationFinished();
+
+ protected:
+ // Derived classes can override if they want to filter out navigations. This
+ // is called from DidStartNavigation.
+ virtual bool ShouldMonitorNavigation(NavigationHandle* handle);
+
+ 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();
+
+ const GURL url_;
+ bool navigation_paused_;
+ NavigationHandle* handle_;
+ bool handled_navigation_;
+ scoped_refptr<MessageLoopRunner> will_start_loop_runner_;
+ scoped_refptr<MessageLoopRunner> did_finish_loop_runner_;
+
+ base::WeakPtrFactory<TestNavigationManager> weak_factory_;
+
+ DISALLOW_COPY_AND_ASSIGN(TestNavigationManager);
+};
+
} // namespace content
#endif // CONTENT_PUBLIC_TEST_BROWSER_TEST_UTILS_H_
« no previous file with comments | « content/browser/web_contents/web_contents_impl_browsertest.cc ('k') | content/public/test/browser_test_utils.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698