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.
5 #include "chrome/test/chromedriver/chrome/web_view_impl.h"
8 #include "base/files/file_path.h"
9 #include "base/json/json_writer.h"
10 #include "base/logging.h"
11 #include "base/strings/string_util.h"
12 #include "base/strings/stringprintf.h"
13 #include "base/threading/platform_thread.h"
14 #include "base/time/time.h"
15 #include "base/values.h"
16 #include "chrome/test/chromedriver/chrome/debugger_tracker.h"
17 #include "chrome/test/chromedriver/chrome/devtools_client_impl.h"
18 #include "chrome/test/chromedriver/chrome/dom_tracker.h"
19 #include "chrome/test/chromedriver/chrome/frame_tracker.h"
20 #include "chrome/test/chromedriver/chrome/geolocation_override_manager.h"
21 #include "chrome/test/chromedriver/chrome/heap_snapshot_taker.h"
22 #include "chrome/test/chromedriver/chrome/javascript_dialog_manager.h"
23 #include "chrome/test/chromedriver/chrome/js.h"
24 #include "chrome/test/chromedriver/chrome/navigation_tracker.h"
25 #include "chrome/test/chromedriver/chrome/status.h"
26 #include "chrome/test/chromedriver/chrome/ui_events.h"
30 Status GetContextIdForFrame(FrameTracker* tracker,
31 const std::string& frame,
37 Status status = tracker->GetContextIdForFrame(frame, context_id);
43 const char* GetAsString(MouseEventType type) {
45 case kPressedMouseEventType:
46 return "mousePressed";
47 case kReleasedMouseEventType:
48 return "mouseReleased";
49 case kMovedMouseEventType:
56 const char* GetAsString(TouchEventType type) {
69 const char* GetPointStateString(TouchEventType type) {
72 return "touchPressed";
74 return "touchReleased";
82 const char* GetAsString(MouseButton button) {
84 case kLeftMouseButton:
86 case kMiddleMouseButton:
88 case kRightMouseButton:
90 case kNoneMouseButton:
97 const char* GetAsString(KeyEventType type) {
99 case kKeyDownEventType:
101 case kKeyUpEventType:
103 case kRawKeyDownEventType:
114 WebViewImpl::WebViewImpl(const std::string& id,
116 scoped_ptr<DevToolsClient> client)
119 dom_tracker_(new DomTracker(client.get())),
120 frame_tracker_(new FrameTracker(client.get())),
121 navigation_tracker_(new NavigationTracker(client.get())),
122 dialog_manager_(new JavaScriptDialogManager(client.get())),
123 geolocation_override_manager_(
124 new GeolocationOverrideManager(client.get())),
125 heap_snapshot_taker_(new HeapSnapshotTaker(client.get())),
126 debugger_(new DebuggerTracker(client.get())),
127 client_(client.release()) {}
129 WebViewImpl::~WebViewImpl() {}
131 std::string WebViewImpl::GetId() {
135 bool WebViewImpl::WasCrashed() {
136 return client_->WasCrashed();
139 Status WebViewImpl::ConnectIfNecessary() {
140 return client_->ConnectIfNecessary();
143 Status WebViewImpl::HandleReceivedEvents() {
144 return client_->HandleReceivedEvents();
147 Status WebViewImpl::Load(const std::string& url) {
148 // Javascript URLs will cause a hang while waiting for the page to stop
149 // loading, so just disallow.
150 if (StartsWithASCII(url, "javascript:", false))
151 return Status(kUnknownError, "unsupported protocol");
152 base::DictionaryValue params;
153 params.SetString("url", url);
154 return client_->SendCommand("Page.navigate", params);
157 Status WebViewImpl::Reload() {
158 base::DictionaryValue params;
159 params.SetBoolean("ignoreCache", false);
160 return client_->SendCommand("Page.reload", params);
163 Status WebViewImpl::EvaluateScript(const std::string& frame,
164 const std::string& expression,
165 scoped_ptr<base::Value>* result) {
167 Status status = GetContextIdForFrame(frame_tracker_.get(), frame,
169 if (status.IsError())
171 return internal::EvaluateScriptAndGetValue(
172 client_.get(), context_id, expression, result);
175 Status WebViewImpl::CallFunction(const std::string& frame,
176 const std::string& function,
177 const base::ListValue& args,
178 scoped_ptr<base::Value>* result) {
180 base::JSONWriter::Write(&args, &json);
181 // TODO(zachconrad): Second null should be array of shadow host ids.
182 std::string expression = base::StringPrintf(
183 "(%s).apply(null, [null, %s, %s])",
187 scoped_ptr<base::Value> temp_result;
188 Status status = EvaluateScript(frame, expression, &temp_result);
189 if (status.IsError())
192 return internal::ParseCallFunctionResult(*temp_result, result);
195 Status WebViewImpl::CallAsyncFunction(const std::string& frame,
196 const std::string& function,
197 const base::ListValue& args,
198 const base::TimeDelta& timeout,
199 scoped_ptr<base::Value>* result) {
200 return CallAsyncFunctionInternal(
201 frame, function, args, false, timeout, result);
204 Status WebViewImpl::CallUserAsyncFunction(const std::string& frame,
205 const std::string& function,
206 const base::ListValue& args,
207 const base::TimeDelta& timeout,
208 scoped_ptr<base::Value>* result) {
209 return CallAsyncFunctionInternal(
210 frame, function, args, true, timeout, result);
213 Status WebViewImpl::GetFrameByFunction(const std::string& frame,
214 const std::string& function,
215 const base::ListValue& args,
216 std::string* out_frame) {
218 Status status = GetContextIdForFrame(frame_tracker_.get(), frame,
220 if (status.IsError())
224 status = internal::GetNodeIdFromFunction(
225 client_.get(), context_id, function, args, &found_node, &node_id);
226 if (status.IsError())
229 return Status(kNoSuchFrame);
230 return dom_tracker_->GetFrameIdForNode(node_id, out_frame);
233 Status WebViewImpl::DispatchMouseEvents(const std::list<MouseEvent>& events,
234 const std::string& frame) {
235 for (std::list<MouseEvent>::const_iterator it = events.begin();
236 it != events.end(); ++it) {
237 base::DictionaryValue params;
238 params.SetString("type", GetAsString(it->type));
239 params.SetInteger("x", it->x);
240 params.SetInteger("y", it->y);
241 params.SetInteger("modifiers", it->modifiers);
242 params.SetString("button", GetAsString(it->button));
243 params.SetInteger("clickCount", it->click_count);
244 Status status = client_->SendCommand("Input.dispatchMouseEvent", params);
245 if (status.IsError())
247 if (build_no_ < 1569 && it->button == kRightMouseButton &&
248 it->type == kReleasedMouseEventType) {
249 base::ListValue args;
250 args.AppendInteger(it->x);
251 args.AppendInteger(it->y);
252 args.AppendInteger(it->modifiers);
253 scoped_ptr<base::Value> result;
254 status = CallFunction(
255 frame, kDispatchContextMenuEventScript, args, &result);
256 if (status.IsError())
263 Status WebViewImpl::DispatchTouchEvents(const std::list<TouchEvent>& events) {
264 for (std::list<TouchEvent>::const_iterator it = events.begin();
265 it != events.end(); ++it) {
266 base::DictionaryValue params;
267 params.SetString("type", GetAsString(it->type));
268 scoped_ptr<base::ListValue> point_list(new base::ListValue);
269 scoped_ptr<base::DictionaryValue> point(new base::DictionaryValue);
270 point->SetString("state", GetPointStateString(it->type));
271 point->SetInteger("x", it->x);
272 point->SetInteger("y", it->y);
273 point_list->Set(0, point.release());
274 params.Set("touchPoints", point_list.release());
275 Status status = client_->SendCommand("Input.dispatchTouchEvent", params);
276 if (status.IsError())
282 Status WebViewImpl::DispatchKeyEvents(const std::list<KeyEvent>& events) {
283 for (std::list<KeyEvent>::const_iterator it = events.begin();
284 it != events.end(); ++it) {
285 base::DictionaryValue params;
286 params.SetString("type", GetAsString(it->type));
287 if (it->modifiers & kNumLockKeyModifierMask) {
288 params.SetBoolean("isKeypad", true);
289 params.SetInteger("modifiers",
290 it->modifiers & ~kNumLockKeyModifierMask);
292 params.SetInteger("modifiers", it->modifiers);
294 params.SetString("text", it->modified_text);
295 params.SetString("unmodifiedText", it->unmodified_text);
296 params.SetInteger("nativeVirtualKeyCode", it->key_code);
297 params.SetInteger("windowsVirtualKeyCode", it->key_code);
298 Status status = client_->SendCommand("Input.dispatchKeyEvent", params);
299 if (status.IsError())
305 Status WebViewImpl::GetCookies(scoped_ptr<base::ListValue>* cookies) {
306 base::DictionaryValue params;
307 scoped_ptr<base::DictionaryValue> result;
308 Status status = client_->SendCommandAndGetResult(
309 "Page.getCookies", params, &result);
310 if (status.IsError())
312 base::ListValue* cookies_tmp;
313 if (!result->GetList("cookies", &cookies_tmp))
314 return Status(kUnknownError, "DevTools didn't return cookies");
315 cookies->reset(cookies_tmp->DeepCopy());
319 Status WebViewImpl::DeleteCookie(const std::string& name,
320 const std::string& url) {
321 base::DictionaryValue params;
322 params.SetString("cookieName", name);
323 params.SetString("url", url);
324 return client_->SendCommand("Page.deleteCookie", params);
327 Status WebViewImpl::WaitForPendingNavigations(const std::string& frame_id,
328 const base::TimeDelta& timeout,
329 bool stop_load_on_timeout) {
330 VLOG(0) << "Waiting for pending navigations...";
331 Status status = client_->HandleEventsUntil(
332 base::Bind(&WebViewImpl::IsNotPendingNavigation,
333 base::Unretained(this),
336 if (status.code() == kTimeout && stop_load_on_timeout) {
337 VLOG(0) << "Timed out. Stopping navigation...";
338 scoped_ptr<base::Value> unused_value;
339 EvaluateScript(std::string(), "window.stop();", &unused_value);
340 Status new_status = client_->HandleEventsUntil(
341 base::Bind(&WebViewImpl::IsNotPendingNavigation, base::Unretained(this),
343 base::TimeDelta::FromSeconds(10));
344 if (new_status.IsError())
347 VLOG(0) << "Done waiting for pending navigations";
351 Status WebViewImpl::IsPendingNavigation(const std::string& frame_id,
353 return navigation_tracker_->IsPendingNavigation(frame_id, is_pending);
356 JavaScriptDialogManager* WebViewImpl::GetJavaScriptDialogManager() {
357 return dialog_manager_.get();
360 Status WebViewImpl::OverrideGeolocation(const Geoposition& geoposition) {
361 return geolocation_override_manager_->OverrideGeolocation(geoposition);
364 Status WebViewImpl::CaptureScreenshot(std::string* screenshot) {
365 base::DictionaryValue params;
366 scoped_ptr<base::DictionaryValue> result;
367 Status status = client_->SendCommandAndGetResult(
368 "Page.captureScreenshot", params, &result);
369 if (status.IsError())
371 if (!result->GetString("data", screenshot))
372 return Status(kUnknownError, "expected string 'data' in response");
376 Status WebViewImpl::SetFileInputFiles(
377 const std::string& frame,
378 const base::DictionaryValue& element,
379 const std::vector<base::FilePath>& files) {
380 base::ListValue file_list;
381 for (size_t i = 0; i < files.size(); ++i) {
382 if (!files[i].IsAbsolute()) {
383 return Status(kUnknownError,
384 "path is not absolute: " + files[i].AsUTF8Unsafe());
386 if (files[i].ReferencesParent()) {
387 return Status(kUnknownError,
388 "path is not canonical: " + files[i].AsUTF8Unsafe());
390 file_list.AppendString(files[i].value());
394 Status status = GetContextIdForFrame(frame_tracker_.get(), frame,
396 if (status.IsError())
398 base::ListValue args;
399 args.Append(element.DeepCopy());
402 status = internal::GetNodeIdFromFunction(
403 client_.get(), context_id, "function(element) { return element; }",
404 args, &found_node, &node_id);
405 if (status.IsError())
408 return Status(kUnknownError, "no node ID for file input");
409 base::DictionaryValue params;
410 params.SetInteger("nodeId", node_id);
411 params.Set("files", file_list.DeepCopy());
412 return client_->SendCommand("DOM.setFileInputFiles", params);
415 Status WebViewImpl::TakeHeapSnapshot(scoped_ptr<base::Value>* snapshot) {
416 return heap_snapshot_taker_->TakeSnapshot(snapshot);
419 Status WebViewImpl::CallAsyncFunctionInternal(const std::string& frame,
420 const std::string& function,
421 const base::ListValue& args,
422 bool is_user_supplied,
423 const base::TimeDelta& timeout,
424 scoped_ptr<base::Value>* result) {
425 base::ListValue async_args;
426 async_args.AppendString("return (" + function + ").apply(null, arguments);");
427 async_args.Append(args.DeepCopy());
428 async_args.AppendBoolean(is_user_supplied);
429 async_args.AppendInteger(timeout.InMilliseconds());
430 scoped_ptr<base::Value> tmp;
431 Status status = CallFunction(
432 frame, kExecuteAsyncScriptScript, async_args, &tmp);
433 if (status.IsError())
436 const char* kDocUnloadError = "document unloaded while waiting for result";
437 std::string kQueryResult = base::StringPrintf(
439 " var info = document.$chrome_asyncScriptInfo;"
441 " return {status: %d, value: '%s'};"
442 " var result = info.result;"
444 " return {status: 0};"
445 " delete info.result;"
452 base::ListValue no_args;
453 scoped_ptr<base::Value> query_value;
454 Status status = CallFunction(frame, kQueryResult, no_args, &query_value);
455 if (status.IsError()) {
456 if (status.code() == kNoSuchFrame)
457 return Status(kJavaScriptError, kDocUnloadError);
461 base::DictionaryValue* result_info = NULL;
462 if (!query_value->GetAsDictionary(&result_info))
463 return Status(kUnknownError, "async result info is not a dictionary");
465 if (!result_info->GetInteger("status", &status_code))
466 return Status(kUnknownError, "async result info has no int 'status'");
467 if (status_code != kOk) {
469 result_info->GetString("value", &message);
470 return Status(static_cast<StatusCode>(status_code), message);
473 base::Value* value = NULL;
474 if (result_info->Get("value", &value)) {
475 result->reset(value->DeepCopy());
479 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(100));
483 Status WebViewImpl::IsNotPendingNavigation(const std::string& frame_id,
484 bool* is_not_pending) {
487 navigation_tracker_->IsPendingNavigation(frame_id, &is_pending);
488 if (status.IsError())
490 // An alert may block the pending navigation.
491 if (is_pending && dialog_manager_->IsDialogOpen())
492 return Status(kUnexpectedAlertOpen);
494 *is_not_pending = !is_pending;
500 Status EvaluateScript(DevToolsClient* client,
502 const std::string& expression,
503 EvaluateScriptReturnType return_type,
504 scoped_ptr<base::DictionaryValue>* result) {
505 base::DictionaryValue params;
506 params.SetString("expression", expression);
508 params.SetInteger("contextId", context_id);
509 params.SetBoolean("returnByValue", return_type == ReturnByValue);
510 scoped_ptr<base::DictionaryValue> cmd_result;
511 Status status = client->SendCommandAndGetResult(
512 "Runtime.evaluate", params, &cmd_result);
513 if (status.IsError())
517 if (!cmd_result->GetBoolean("wasThrown", &was_thrown))
518 return Status(kUnknownError, "Runtime.evaluate missing 'wasThrown'");
520 std::string description = "unknown";
521 cmd_result->GetString("result.description", &description);
522 return Status(kUnknownError,
523 "Runtime.evaluate threw exception: " + description);
526 base::DictionaryValue* unscoped_result;
527 if (!cmd_result->GetDictionary("result", &unscoped_result))
528 return Status(kUnknownError, "evaluate missing dictionary 'result'");
529 result->reset(unscoped_result->DeepCopy());
533 Status EvaluateScriptAndGetObject(DevToolsClient* client,
535 const std::string& expression,
537 std::string* object_id) {
538 scoped_ptr<base::DictionaryValue> result;
539 Status status = EvaluateScript(client, context_id, expression, ReturnByObject,
541 if (status.IsError())
543 if (!result->HasKey("objectId")) {
547 if (!result->GetString("objectId", object_id))
548 return Status(kUnknownError, "evaluate has invalid 'objectId'");
553 Status EvaluateScriptAndGetValue(DevToolsClient* client,
555 const std::string& expression,
556 scoped_ptr<base::Value>* result) {
557 scoped_ptr<base::DictionaryValue> temp_result;
558 Status status = EvaluateScript(client, context_id, expression, ReturnByValue,
560 if (status.IsError())
564 if (!temp_result->GetString("type", &type))
565 return Status(kUnknownError, "Runtime.evaluate missing string 'type'");
567 if (type == "undefined") {
568 result->reset(base::Value::CreateNullValue());
571 if (!temp_result->Get("value", &value))
572 return Status(kUnknownError, "Runtime.evaluate missing 'value'");
573 result->reset(value->DeepCopy());
578 Status ParseCallFunctionResult(const base::Value& temp_result,
579 scoped_ptr<base::Value>* result) {
580 const base::DictionaryValue* dict;
581 if (!temp_result.GetAsDictionary(&dict))
582 return Status(kUnknownError, "call function result must be a dictionary");
584 if (!dict->GetInteger("status", &status_code)) {
585 return Status(kUnknownError,
586 "call function result missing int 'status'");
588 if (status_code != kOk) {
590 dict->GetString("value", &message);
591 return Status(static_cast<StatusCode>(status_code), message);
593 const base::Value* unscoped_value;
594 if (!dict->Get("value", &unscoped_value)) {
595 return Status(kUnknownError,
596 "call function result missing 'value'");
598 result->reset(unscoped_value->DeepCopy());
602 Status GetNodeIdFromFunction(DevToolsClient* client,
604 const std::string& function,
605 const base::ListValue& args,
609 base::JSONWriter::Write(&args, &json);
610 // TODO(zachconrad): Second null should be array of shadow host ids.
611 std::string expression = base::StringPrintf(
612 "(%s).apply(null, [null, %s, %s, true])",
618 std::string element_id;
619 Status status = internal::EvaluateScriptAndGetObject(
620 client, context_id, expression, &got_object, &element_id);
621 if (status.IsError())
628 scoped_ptr<base::DictionaryValue> cmd_result;
630 base::DictionaryValue params;
631 params.SetString("objectId", element_id);
632 status = client->SendCommandAndGetResult(
633 "DOM.requestNode", params, &cmd_result);
636 // Release the remote object before doing anything else.
637 base::DictionaryValue params;
638 params.SetString("objectId", element_id);
639 Status release_status =
640 client->SendCommand("Runtime.releaseObject", params);
641 if (release_status.IsError()) {
642 LOG(ERROR) << "Failed to release remote object: "
643 << release_status.message();
646 if (status.IsError())
649 if (!cmd_result->GetInteger("nodeId", node_id))
650 return Status(kUnknownError, "DOM.requestNode missing int 'nodeId'");
655 } // namespace internal