[Media Router] Implement MediaRouterAction + MediaRouterUI tests.

chrome/browser/ui/views/media_router/media_router_ui_browsertest.cc

Index: chrome/browser/ui/views/media_router/media_router_ui_browsertest.cc
diff --git a/chrome/browser/ui/views/media_router/media_router_ui_browsertest.cc b/chrome/browser/ui/views/media_router/media_router_ui_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..249042b03c39315d87a1cb9901c8b54b95f20bf8
--- /dev/null
+++ b/chrome/browser/ui/views/media_router/media_router_ui_browsertest.cc
@@ -0,0 +1,89 @@
+// Copyright 2015 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.
+
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_commands.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/browser/ui/toolbar/media_router_action.h"
+#include "chrome/browser/ui/toolbar/toolbar_action_view_delegate.h"
+#include "chrome/browser/ui/views/frame/browser_view.h"
+#include "chrome/browser/ui/views/toolbar/browser_actions_container.h"
+#include "chrome/browser/ui/views/toolbar/toolbar_action_view.h"
+#include "chrome/browser/ui/views/toolbar/toolbar_view.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/test/test_navigation_observer.h"
+#include "content/public/test/test_utils.h"
+
+namespace media_router {
+
+class MediaRouterUIBrowserTest : public InProcessBrowserTest {
+ public:
+  MediaRouterUIBrowserTest() {}
+  ~MediaRouterUIBrowserTest() override {}
+
+  void SetUpOnMainThread() override {
+    InProcessBrowserTest::SetUpOnMainThread();
+
+    BrowserActionsContainer* browser_actions_container =
+        BrowserView::GetBrowserViewForBrowser(browser())
+            ->toolbar()
+            ->browser_actions();
+    ASSERT_TRUE(browser_actions_container);
+
+    media_router_action_.reset(new MediaRouterAction);
+
+    // Sets delegate on |media_router_action_|.
+    toolbar_action_view_.reset(
+        new ToolbarActionView(media_router_action_.get(), browser()->profile(),
+                              browser_actions_container));
+  }
+
+  void OpenMediaRouterDialogAndWaitForNewWebContents() {
+    content::TestNavigationObserver nav_observer(NULL);
+    nav_observer.StartWatchingNewWebContents();
+    media_router_action_->ExecuteAction(true);
+    nav_observer.Wait();
+    nav_observer.StopWatchingNewWebContents();
+  }
+
+ protected:
+  // Must be initialized after |InProcessBrowserTest::SetUpOnMainThread|.
+  scoped_ptr<MediaRouterAction> media_router_action_;
+
+  // ToolbarActionView constructed to set the delegate on |mr_action|.
+  scoped_ptr<ToolbarActionView> toolbar_action_view_;
+};
+
+IN_PROC_BROWSER_TEST_F(MediaRouterUIBrowserTest,

[Peter Kasting, 2015/06/04 07:13:03]

This test is better now, but it still wouldn't go amiss to add a comment above
the test summarizing it, e.g. talking about that this tests various different
types of navigations that should trigger creation of a media router dialog and
ensures they also remove any previously-visible dialogs, or whatever would
actually be accurate.

[imcheng (use chromium acct), 2015/06/04 16:45:43]

Done.

+                       OpenDialogWithMediaRouterAction) {
+  // We start off at about:blank page.
+  // Make sure there is 1 tab and media router is enabled.
+  ASSERT_EQ(1, browser()->tab_strip_model()->count());
+
+  OpenMediaRouterDialogAndWaitForNewWebContents();
+
+  // Reload the browser and wait.
+  content::TestNavigationObserver reload_observer(
+      browser()->tab_strip_model()->GetActiveWebContents());
+  chrome::Reload(browser(), CURRENT_TAB);
+  reload_observer.Wait();
+
+  // The reload should have removed the previously created dialog.
+  // We expect a new dialog WebContents to be created by calling this.

[Peter Kasting, 2015/06/04 07:13:03]

If the previous dialog wasn't removed or a new WebContents isn't created, will
the test hang indefinitely, or eventually figure this out and time out or
something?  It would be nice to fail fast, if doing so is feasible.  (Maybe it's
not.)

[imcheng (use chromium acct), 2015/06/04 16:45:43]

I believe browser tests have a builtin timeout (something like 60s but is
configurable) so this will not hang indefinitely. We could add a timeout to say
the TestNavigationObserver if we wanted a stricter timeout, but it is not needed
right now.

+  OpenMediaRouterDialogAndWaitForNewWebContents();
+
+  // Navigate away and wait.
+  content::TestNavigationObserver nav_observer(
+      browser()->tab_strip_model()->GetActiveWebContents(), 1);
+  ui_test_utils::NavigateToURL(browser(), GURL("about:blank"));
+  nav_observer.Wait();
+
+  // The navigation should have removed the previously created dialog.
+  // We expect a new dialog WebContents to be created by calling this.
+  OpenMediaRouterDialogAndWaitForNewWebContents();
+}
+
+}  // namespace media_router