[Media Router] Auto-close Media Router dialog after starting or stopping a session.

chrome/test/media_router/media_router_integration_browsertest.cc

 // 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/test/media_router/media_router_integration_browsertest.h"
 
 #include "base/bind.h"
 #include "base/files/file_util.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/json/json_reader.h"
@@ skipping 37 matching lines @@
     "    sinks[i].click();"
     "    break;"
     "}}";
 const char kCloseRouteScript[] =
     "window.document.getElementById('media-router-container').shadowRoot."
     "  getElementById('route-details').shadowRoot.getElementById("
     "    'close-route-button').click()";
 const char kGetRouteLengthScript[] =
     "domAutomationController.send(window.document.getElementById("
     "  'media-router-container').routeList.length)";
-
+const char kClickDialog[] =
+    "window.document.getElementById('media-router-container').click();";
 std::string GetStartedSessionId(content::WebContents* web_contents) {
   std::string session_id;
   CHECK(content::ExecuteScriptAndExtractString(
       web_contents, "window.domAutomationController.send(startedSession.id)",
       &session_id));
   return session_id;
 }
 
 std::string GetDefaultRequestSessionId(content::WebContents* web_contents) {
   std::string session_id;
@@ skipping 67 matching lines @@
 
 void MediaRouterIntegrationBrowserTest::ChooseSink(
     content::WebContents* web_contents,
     const std::string& sink_name) {
   content::WebContents* dialog_contents = GetMRDialog(web_contents);
   std::string script = base::StringPrintf(
       kChooseSinkScript, sink_name.c_str());
   ASSERT_TRUE(content::ExecuteScript(dialog_contents, script));
 }
 
+void MediaRouterIntegrationBrowserTest::ClickDialog() {
+  content::WebContents* web_contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  content::WebContents* dialog_contents = GetMRDialog(web_contents);
+  ASSERT_TRUE(content::ExecuteScript(dialog_contents, kClickDialog));
+}
 content::WebContents* MediaRouterIntegrationBrowserTest::GetMRDialog(
     content::WebContents* web_contents) {
   MediaRouterDialogControllerImpl* controller =
       MediaRouterDialogControllerImpl::GetOrCreateForWebContents(web_contents);
   content::WebContents* dialog_contents = controller->GetMediaRouterDialog();
   CHECK(dialog_contents);
   return dialog_contents;
 }
 
+bool MediaRouterIntegrationBrowserTest::IsDialogClosed(
+    content::WebContents* web_contents) {
+  MediaRouterDialogControllerImpl* controller =
+      MediaRouterDialogControllerImpl::GetOrCreateForWebContents(web_contents);
+  return !controller->GetMediaRouterDialog();
+}
+
+void MediaRouterIntegrationBrowserTest::WaitUntilDialogClosed(
+    content::WebContents* web_contents) {
+  ASSERT_TRUE(ConditionalWait(
+      base::TimeDelta::FromSeconds(5), base::TimeDelta::FromSeconds(1),
+      base::Bind(&MediaRouterIntegrationBrowserTest::IsDialogClosed,
+                 base::Unretained(this), web_contents)));
+}
+
+void MediaRouterIntegrationBrowserTest::CheckDialogRemainsOpen(
+    content::WebContents* web_contents) {
+  ASSERT_FALSE(ConditionalWait(
+      base::TimeDelta::FromSeconds(5), base::TimeDelta::FromSeconds(1),
+      base::Bind(&MediaRouterIntegrationBrowserTest::IsDialogClosed,
+                 base::Unretained(this), web_contents)));
+}
+
 void MediaRouterIntegrationBrowserTest::SetTestData(
     base::FilePath::StringPieceType test_data_file) {
   base::FilePath full_path = GetResourceFile(test_data_file);
   JSONFileValueDeserializer deserializer(full_path);
   int error_code = 0;
   std::string error_message;
   scoped_ptr<base::Value> value =
       deserializer.Deserialize(&error_code, &error_message);
   CHECK(value.get()) << "Deserialize failed: " << error_message;
   std::string test_data_str;
@@ skipping 202 matching lines @@
   content::WebContents* new_web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_TRUE(new_web_contents);
   ExecuteJavaScriptAPI(
       new_web_contents,
       base::StringPrintf("checkReconnectSessionFails('%s');",
                          session_id.c_str()));
 }
 
 }  // namespace media_router