Added a new automation hook ExecuteJavascriptInOOBEWebUI() to execute javascript prior to login on…

chrome/browser/automation/testing_automation_provider_chromeos.cc

 // Copyright (c) 2012 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/automation/testing_automation_provider.h"
 
 #include "base/command_line.h"
 #include "base/i18n/time_formatting.h"
 #include "base/stringprintf.h"
 #include "base/time.h"
@@ skipping 22 matching lines @@
 #include "chrome/browser/policy/cloud_policy_subsystem.h"
 #include "chrome/browser/policy/enterprise_install_attributes.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/dialog_style.h"
 #include "chrome/browser/ui/webui/chromeos/login/oobe_ui.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/power_manager_client.h"
 #include "chromeos/dbus/update_engine_client.h"
+#include "content/public/browser/web_contents.h"
 #include "net/base/network_change_notifier.h"
 #include "policy/policy_constants.h"
 #include "ui/views/widget/widget.h"
 
 using chromeos::CrosLibrary;
 using chromeos::DBusThreadManager;
 using chromeos::NetworkLibrary;
 using chromeos::UpdateEngineClient;
 using chromeos::UserManager;
 
@@ skipping 229 matching lines @@
   std::string username, password;
   if (!args->GetString("username", &username) ||
       !args->GetString("password", &password)) {
     AutomationJSONReply(this, reply_message).SendError(
         "Invalid or missing args.");
     return;
   }
 
   chromeos::ExistingUserController* controller =
       chromeos::ExistingUserController::current_controller();
+  if (!controller) {
+    AutomationJSONReply(this, reply_message).SendError(
+        "Unable to access ExistingUserController");
+    return;
+  }
 
   // Set up an observer (it will delete itself).
   new LoginObserver(controller, this, reply_message);
 
   // WebUI login.
   chromeos::WebUILoginDisplay* webui_login_display =
       static_cast<chromeos::WebUILoginDisplay*>(controller->login_display());
   LOG(ERROR) << "TestingAutomationProvider::Login ShowSigninScreenForCreds("
              << username << ", " << password << ")";
 
@@ skipping 645 matching lines @@
     AutomationJSONReply(this, reply_message).SendError(
         "Unable to access EnterpriseEnrollmentScreen");
     return;
   }
   // Set up an observer (it will delete itself).
   new EnrollmentObserver(this, reply_message, enroll_screen->GetActor(),
                          enroll_screen);
   enroll_screen->GetActor()->SubmitTestCredentials(user, password);
 }
 
+void TestingAutomationProvider::ExecuteJavascriptInLoginWebUI(
+    DictionaryValue* args, IPC::Message* reply_message) {
+  std::string javascript, frame_xpath;
+  const UserManager* user_manager = UserManager::Get();

[dennis_jeffrey, 2012/04/28 16:44:05]

move this down below right before line 977

[craigdh, 2012/04/30 18:41:32]

Done.

+  if (!args->GetString("javascript", &javascript)) {
+    AutomationJSONReply(this, reply_message)
+        .SendError("'javascript' missing or invalid");
+    return;
+  }
+  if (!args->GetString("frame_xpath", &frame_xpath)) {
+    AutomationJSONReply(this, reply_message)
+        .SendError("'frame_xpath' missing or invalid");
+    return;
+  }
+  if (!user_manager) {
+    AutomationJSONReply(this, reply_message).SendError(
+        "No user manager!");
+    return;
+  }
+  if (user_manager->IsUserLoggedIn()) {
+    AutomationJSONReply(this, reply_message).SendError(
+        "User is already logged in.");
+    return;
+  }
+  chromeos::ExistingUserController* controller =
+      chromeos::ExistingUserController::current_controller();
+  chromeos::WebUILoginDisplayHost* webui_login_display_host =
+      static_cast<chromeos::WebUILoginDisplayHost*>(
+          controller->login_display_host());

[dennis_jeffrey, 2012/04/28 16:44:05]

is it safe to assume 'controller' exists here?  You had added a check at line
293 above.

[craigdh, 2012/04/30 18:41:32]

Done.

+  content::WebContents* web_contents =
+      webui_login_display_host->GetOobeUI()->web_ui()->GetWebContents();
+
+  new DomOperationMessageSender(this, reply_message, true);
+  ExecuteJavascriptInRenderViewFrame(ASCIIToUTF16(frame_xpath),
+                                     ASCIIToUTF16(javascript),
+                                     reply_message,
+                                     web_contents->GetRenderViewHost());
+}
+
 void TestingAutomationProvider::GetEnterprisePolicyInfo(
     DictionaryValue* args, IPC::Message* reply_message) {
   AutomationJSONReply reply(this, reply_message);
   scoped_ptr<DictionaryValue> return_value(new DictionaryValue);
 
   policy::BrowserPolicyConnector* connector =
       g_browser_process->browser_policy_connector();
   if (!connector) {
     reply.SendError("Unable to access BrowserPolicyConnector");
     return;
@@ skipping 228 matching lines @@
   chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->
       AddObserver(power_manager_observer_);
 }
 
 void TestingAutomationProvider::RemoveChromeosObservers() {
   chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->
       RemoveObserver(power_manager_observer_);
   delete power_manager_observer_;
   power_manager_observer_ = NULL;
 }