1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/test/chromedriver/alert_commands.h"
6
7 #include "base/callback.h"
8 #include "base/values.h"
9 #include "chrome/test/chromedriver/chrome/chrome.h"
10 #include "chrome/test/chromedriver/chrome/devtools_client.h"
11 #include "chrome/test/chromedriver/chrome/javascript_dialog_manager.h"
12 #include "chrome/test/chromedriver/chrome/status.h"
13 #include "chrome/test/chromedriver/chrome/web_view.h"
14 #include "chrome/test/chromedriver/net/timeout.h"
15 #include "chrome/test/chromedriver/session.h"
16
ExecuteAlertCommand(const AlertCommand & alert_command,Session * session,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)17 Status ExecuteAlertCommand(const AlertCommand& alert_command,
18 Session* session,
19 const base::DictionaryValue& params,
20 std::unique_ptr<base::Value>* value) {
21 WebView* web_view = NULL;
22 Status status = session->GetTargetWindow(&web_view);
23 if (status.IsError())
24 return status;
25
26 status = web_view->ConnectIfNecessary();
27 if (status.IsError())
28 return status;
29
30 status = web_view->HandleReceivedEvents();
31 if (status.IsError())
32 return status;
33
34 status = web_view->WaitForPendingNavigations(
35 session->GetCurrentFrameId(), Timeout(session->page_load_timeout), true);
36 if (status.IsError() && status.code() != kUnexpectedAlertOpen)
37 return status;
38
39 return alert_command.Run(session, web_view, params, value);
40 }
41
ExecuteGetAlert(Session * session,WebView * web_view,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)42 Status ExecuteGetAlert(Session* session,
43 WebView* web_view,
44 const base::DictionaryValue& params,
45 std::unique_ptr<base::Value>* value) {
46 value->reset(
47 new base::Value(web_view->GetJavaScriptDialogManager()->IsDialogOpen()));
48 return Status(kOk);
49 }
50
ExecuteGetAlertText(Session * session,WebView * web_view,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)51 Status ExecuteGetAlertText(Session* session,
52 WebView* web_view,
53 const base::DictionaryValue& params,
54 std::unique_ptr<base::Value>* value) {
55 std::string message;
56 Status status =
57 web_view->GetJavaScriptDialogManager()->GetDialogMessage(&message);
58 if (status.IsError())
59 return status;
60 value->reset(new base::Value(message));
61 return Status(kOk);
62 }
63
ExecuteSetAlertText(Session * session,WebView * web_view,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)64 Status ExecuteSetAlertText(Session* session,
65 WebView* web_view,
66 const base::DictionaryValue& params,
67 std::unique_ptr<base::Value>* value) {
68 std::string text;
69 if (!params.GetString("text", &text))
70 return Status(kInvalidArgument, "missing or invalid 'text'");
71
72 JavaScriptDialogManager* dialog_manager =
73 web_view->GetJavaScriptDialogManager();
74
75 if (!dialog_manager->IsDialogOpen())
76 return Status(kNoSuchAlert);
77
78 std::string type;
79 Status status = dialog_manager->GetTypeOfDialog(&type);
80 if (status.IsError())
81 return status;
82
83 if (type == "prompt")
84 session->prompt_text.reset(new std::string(text));
85 else if (type == "alert" || type == "confirm")
86 return Status(kElementNotInteractable,
87 "User dialog does not have a text box input field.");
88 else
89 return Status(kUnsupportedOperation,
90 "Text can only be sent to window.prompt dialogs.");
91 return Status(kOk);
92 }
93
ExecuteAcceptAlert(Session * session,WebView * web_view,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)94 Status ExecuteAcceptAlert(Session* session,
95 WebView* web_view,
96 const base::DictionaryValue& params,
97 std::unique_ptr<base::Value>* value) {
98 Status status = web_view->GetJavaScriptDialogManager()
99 ->HandleDialog(true, session->prompt_text.get());
100 session->prompt_text.reset();
101 return status;
102 }
103
ExecuteDismissAlert(Session * session,WebView * web_view,const base::DictionaryValue & params,std::unique_ptr<base::Value> * value)104 Status ExecuteDismissAlert(Session* session,
105 WebView* web_view,
106 const base::DictionaryValue& params,
107 std::unique_ptr<base::Value>* value) {
108 Status status = web_view->GetJavaScriptDialogManager()
109 ->HandleDialog(false, session->prompt_text.get());
110 session->prompt_text.reset();
111 return status;
112 }
113