Adding the capacity to interact with modal dialogs to the automation framework.

This change will be used by Ojan to implement some unload handler tests.

TEST=Run the ui tests.
Review URL: http://codereview.chromium.org/13113

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@6402 0039d316-1c4b-4281-b951-d872f2087c98

2 files changed, 37 insertions, 2 deletions

diff --git a/chrome/browser/automation/automation_provider.cc b/chrome/browser/automation/automation_provider.cc
index 146883b..c1d4a15 100644
--- a/chrome/browser/automation/automation_provider.cc
+++ b/chrome/browser/automation/automation_provider.cc
@@ -30,6 +30,8 @@
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/pref_service.h"
 #include "chrome/test/automation/automation_messages.h"
+#include "chrome/views/app_modal_dialog_delegate.h"
+#include "chrome/views/window.h"
 #include "net/base/cookie_monster.h"
 #include "net/url_request/url_request_filter.h"
 
@@ -811,6 +813,8 @@ void AutomationProvider::OnMessageReceived(const IPC::Message& message) {
                         SetIntPreference)
     IPC_MESSAGE_HANDLER(AutomationMsg_ShowingAppModalDialogRequest,
                         GetShowingAppModalDialog)
+    IPC_MESSAGE_HANDLER(AutomationMsg_ClickAppModalDialogButtonRequest,
+                        ClickAppModalDialogButton)
   IPC_END_MESSAGE_MAP()
 }
 
@@ -1074,9 +1078,38 @@ void AutomationProvider::GetBrowserWindowCount(const IPC::Message& message) {
 }
 
 void AutomationProvider::GetShowingAppModalDialog(const IPC::Message& message) {
+  views::AppModalDialogDelegate* dialog_delegate =
+      BrowserList::GetShowingAppModalDialog();
   Send(new AutomationMsg_ShowingAppModalDialogResponse(
-           message.routing_id(),
-           static_cast<bool>(BrowserList::IsShowingAppModalDialog())));
+           message.routing_id(), dialog_delegate != NULL,
+           dialog_delegate ? dialog_delegate->GetDialogButtons() :
+                             views::DialogDelegate::DIALOGBUTTON_NONE));
+}
+
+void AutomationProvider::ClickAppModalDialogButton(const IPC::Message& message,
+                                                   int button) {
+  bool success = false;
+
+  views::AppModalDialogDelegate* dialog_delegate =
+      BrowserList::GetShowingAppModalDialog();
+  if (dialog_delegate &&
+      (dialog_delegate->GetDialogButtons() & button) == button) {
+    views::DialogClientView* client_view =
+        dialog_delegate->window()->client_view()->AsDialogClientView();
+    if ((button & views::DialogDelegate::DIALOGBUTTON_OK) ==
+        views::DialogDelegate::DIALOGBUTTON_OK) {
+      client_view->AcceptWindow();
+      success =  true;
+    }
+    if ((button & views::DialogDelegate::DIALOGBUTTON_CANCEL) ==
+        views::DialogDelegate::DIALOGBUTTON_CANCEL) {
+      DCHECK(!success) << "invalid param, OK and CANCEL specified";
+      client_view->CancelWindow();
+      success =  true;
+    }
+  }
+  Send(new AutomationMsg_ClickAppModalDialogButtonResponse(
+      message.routing_id(), success));
 }
 
 void AutomationProvider::GetBrowserWindow(const IPC::Message& message,
diff --git a/chrome/browser/automation/automation_provider.h b/chrome/browser/automation/automation_provider.h
index 401e0b5..b0e21d3 100644
--- a/chrome/browser/automation/automation_provider.h
+++ b/chrome/browser/automation/automation_provider.h
@@ -106,6 +106,8 @@ class AutomationProvider : public base::RefCounted<AutomationProvider>,
                  int handle);
   void GetBrowserWindowCount(const IPC::Message& message);
   void GetShowingAppModalDialog(const IPC::Message& message);
+  void ClickAppModalDialogButton(const IPC::Message& message,
+                                 int button);
   void GetBrowserWindow(const IPC::Message& message, int index);
   void GetLastActiveBrowserWindow(const IPC::Message& message);
   void GetActiveWindow(const IPC::Message& message);