Add ExecuteJavascript() method to PyUITestBase

chrome/test/pyautolib/pyautolib.cc

 // 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.
 
 #include "base/scoped_ptr.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/test/automation/extension_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
 #include "chrome/test/pyautolib/pyautolib.h"
@@ skipping 277 matching lines @@
   scoped_refptr<BrowserProxy> browser_proxy =
       automation()->GetBrowserWindow(window_index);
   EXPECT_TRUE(browser_proxy.get());
   std::string response;
   if (browser_proxy.get()) {
     EXPECT_TRUE(browser_proxy->SendJSONRequest(request, &response));
   }
   return response;
 }
 
+std::wstring PyUITestBase::ExecuteJavascript(const std::wstring& script,
+                                             int window_index,
+                                             int tab_index,
+                                             const std::wstring& frame_xpath) {
+  scoped_refptr<BrowserProxy> browser_proxy =
+      automation()->GetBrowserWindow(window_index);
+  EXPECT_TRUE(browser_proxy.get());
+  std::wstring response;
+  if (!browser_proxy.get())
+    return response;
+  scoped_refptr<TabProxy> tab_proxy =
+      browser_proxy->GetTab(tab_index);
+  EXPECT_TRUE(tab_proxy.get());
+  if (!tab_proxy.get())
+    return response;
+
+  EXPECT_TRUE(tab_proxy->ExecuteAndExtractString(frame_xpath, script,
+                                                 &response));
+  return response;
+}
+
+std::wstring PyUITestBase::GetDOMValue(const std::wstring& expr,
+                                       int window_index,
+                                       int tab_index,
+                                       const std::wstring& frame_xpath) {
+  std::wstring script = std::wstring(L"window.domAutomationController.send(") +
+      expr + std::wstring(L")");
+  return ExecuteJavascript(script, window_index, tab_index, frame_xpath);
+}
+
 bool PyUITestBase::ResetToDefaultTheme() {
   return automation()->ResetToDefaultTheme();
 }
 
 bool PyUITestBase::SetCookie(const GURL& cookie_url,
                              const std::string& value,
                              int window_index,
                              int tab_index) {
   scoped_refptr<BrowserProxy> browser_proxy = GetBrowserWindow(window_index);
   EXPECT_TRUE(browser_proxy.get());
@@ skipping 15 matching lines @@
   // TODO(phadjan.jr): figure out a way to unambiguously report error.
   if (!browser_proxy.get())
     return cookie_val;
   scoped_refptr<TabProxy> tab_proxy = browser_proxy->GetTab(tab_index);
   EXPECT_TRUE(tab_proxy.get());
   if (!tab_proxy.get())
     return cookie_val;
   EXPECT_TRUE(tab_proxy->GetCookies(cookie_url, &cookie_val));
   return cookie_val;
 }