Index: chrome/test/test_navigation_observer.h |
diff --git a/chrome/test/test_navigation_observer.h b/chrome/test/test_navigation_observer.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..ba76ab5aaff9f5be15f566fa44ea6c63d315af8a |
--- /dev/null |
+++ b/chrome/test/test_navigation_observer.h |
@@ -0,0 +1,90 @@ |
+// Copyright (c) 2011 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 CHROME_TEST_TEST_NAVIGATION_OBSERVER_H_ |
+#define CHROME_TEST_TEST_NAVIGATION_OBSERVER_H_ |
+#pragma once |
+ |
+#include "base/compiler_specific.h" |
+#include "content/common/notification_observer.h" |
+#include "content/common/notification_registrar.h" |
+ |
+class NavigationController; |
+ |
+// In order to support testing of print preview, we need to wait for the tab to |
+// be inserted, and then observe notifications on the newly added tab's |
+// controller to wait for it to be loaded. To support tests registering |
+// javascript WebUI handlers, we need to inject the framework & registration |
+// javascript before the webui page loads by calling back through the |
+// TestTabStripModelObserver::LoadStartObserver when the new page starts |
+// loading. |
+class TestNavigationObserver : public NotificationObserver { |
+ public: |
+ class JsInjectionReadyObserver { |
+ public: |
+ JsInjectionReadyObserver(); |
+ virtual ~JsInjectionReadyObserver(); |
+ |
+ // Called to indicate page entry committed and ready for javascript |
+ // injection. |
+ virtual void OnJsInjectionReady() = 0; |
+ }; |
+ |
+ // Create and register a new TestNavigationObserver against the |
+ // |controller|. When |js_injection_ready_observer| is non-null, notify with |
+ // OnEntryCommitted() after |number_of_navigations| navigations. |
+ // Note: |js_injection_ready_observer| is owned by the caller and should be |
+ // valid until this class is destroyed. |
+ TestNavigationObserver(NavigationController* controller, |
+ JsInjectionReadyObserver* js_injection_ready_observer, |
+ int number_of_navigations); |
+ |
+ virtual ~TestNavigationObserver(); |
+ |
+ // Run the UI message loop until |done_| becomes true. |
+ void WaitForObservation(); |
+ |
+ protected: |
+ // Note: |js_injection_ready_observer| is owned by the caller and should be |
+ // valid until this class is destroyed. |
sky
2011/07/11 22:47:42
You should document that subclasses must explicitl
Sheridan Rawlins
2011/07/12 04:37:49
Done.
|
+ explicit TestNavigationObserver( |
+ JsInjectionReadyObserver* js_injection_ready_observer, |
+ int number_of_navigations); |
+ |
+ // Register this TestNavigationObserver as an observer of the |controller|. |
+ void RegisterAsObserver(NavigationController* controller); |
+ |
+ private: |
+ // NotificationObserver: |
+ virtual void Observe(int type, const NotificationSource& source, |
+ const NotificationDetails& details) OVERRIDE; |
+ |
+ NotificationRegistrar registrar_; |
+ |
+ // If true the navigation has started. |
+ bool navigation_started_; |
+ |
+ // If true the navigation has been committed. |
+ bool navigation_entry_committed_; |
+ |
+ // The number of navigations that have been completed. |
+ int navigations_completed_; |
+ |
+ // The number of navigations to wait for. |
+ int number_of_navigations_; |
+ |
+ // Observer to take some action when the page is ready for javascript |
+ // injection. |
+ JsInjectionReadyObserver* js_injection_ready_observer_; |
+ |
+ // |done_| will get set when this object observes a TabStripModel event. |
+ bool done_; |
+ |
+ // |running_| will be true during WaitForObservation until |done_| is true. |
+ bool running_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TestNavigationObserver); |
+}; |
+ |
+#endif // CHROME_TEST_TEST_NAVIGATION_OBSERVER_H_ |