Add UI test for click-to-play.

chrome/test/plugin/plugin_test.cpp

 // Copyright (c) 2010 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.
 
 // Tests for the top plugins to catch regressions in our plugin host code, as
 // well as in the out of process code.  Currently this tests:
 //  Flash
 //  Real
 //  QuickTime
 //  Windows Media Player
@@ skipping 120 matching lines @@
       "npswf32.dll"
 #elif defined(OS_MACOSX)
       "Flash Player.plugin"
 #elif defined(OS_POSIX)
       "libflashplayer.so"
 #endif
       ;
   TestPlugin("flash.html?" + kFlashQuery, action_max_timeout_ms(), false);
 }
 
+class ClickToPlayPluginTest : public PluginTest {
+ public:
+  ClickToPlayPluginTest() {
+    dom_automation_enabled_ = true;
+  }
+};
+
+TEST_F(ClickToPlayPluginTest, Flash) {
+  scoped_refptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
+  ASSERT_TRUE(browser.get());
+  ASSERT_TRUE(browser->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_PLUGINS,
+                                                CONTENT_SETTING_BLOCK));
+
+  GURL url = GetTestUrl("flash-clicktoplay.html", true);
+  NavigateToURL(url);
+
+  scoped_refptr<TabProxy> tab(browser->GetTab(0));
+  ASSERT_TRUE(tab.get());
+
+  ASSERT_TRUE(tab->LoadBlockedPlugins());
+
+  WaitForFinish(action_max_timeout_ms(), true);
+}
+
+TEST_F(ClickToPlayPluginTest, FlashDocument) {
+  scoped_refptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
+  ASSERT_TRUE(browser.get());
+  ASSERT_TRUE(browser->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_PLUGINS,
+                                                CONTENT_SETTING_BLOCK));
+
+  scoped_refptr<TabProxy> tab(browser->GetTab(0));
+  ASSERT_TRUE(tab.get());
+  GURL url = GetTestUrl("js-invoker.swf?callback=done", true);
+  NavigateToURL(url);
+
+  // Inject the callback function into the HTML page generated by the browser.
+  ASSERT_TRUE(tab->ExecuteJavaScript("window.done = function() {"
+                                     "  window.location = \"done.html\";"
+                                     "}"));
+
+  ASSERT_TRUE(tab->LoadBlockedPlugins());
+
+  WaitForFinish(action_max_timeout_ms(), true);
+}
+
 #if defined(OS_WIN)
 // Windows only test
 TEST_F(PluginTest, FlashSecurity) {
   TestPlugin("flash.html", action_max_timeout_ms(), false);
 }
 #endif  // defined(OS_WIN)
 
 #if defined(OS_WIN)
 // TODO(port) Port the following tests to platforms that have the required
 // plugins.
@@ skipping 147 matching lines @@
 TEST_F(PluginInstallerDownloadTest, PluginInstallerDownloadPathTest) {
   MessageLoop loop(MessageLoop::TYPE_IO);
   Start();
   loop.Run();
 
   EXPECT_TRUE(success());
   EXPECT_TRUE(initial_download_path().BaseName().value() ==
               final_download_path().BaseName().value());
 }
 #endif  // defined(OS_WIN)