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/session.h" 15 16Status ExecuteAlertCommand( 17 const AlertCommand& alert_command, 18 Session* session, 19 const base::DictionaryValue& params, 20 scoped_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(), 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 42Status ExecuteGetAlert( 43 Session* session, 44 WebView* web_view, 45 const base::DictionaryValue& params, 46 scoped_ptr<base::Value>* value) { 47 value->reset(new base::FundamentalValue( 48 web_view->GetJavaScriptDialogManager()->IsDialogOpen())); 49 return Status(kOk); 50} 51 52Status ExecuteGetAlertText( 53 Session* session, 54 WebView* web_view, 55 const base::DictionaryValue& params, 56 scoped_ptr<base::Value>* value) { 57 std::string message; 58 Status status = 59 web_view->GetJavaScriptDialogManager()->GetDialogMessage(&message); 60 if (status.IsError()) 61 return status; 62 value->reset(new base::StringValue(message)); 63 return Status(kOk); 64} 65 66Status ExecuteSetAlertValue( 67 Session* session, 68 WebView* web_view, 69 const base::DictionaryValue& params, 70 scoped_ptr<base::Value>* value) { 71 std::string text; 72 if (!params.GetString("text", &text)) 73 return Status(kUnknownError, "missing or invalid 'text'"); 74 75 if (!web_view->GetJavaScriptDialogManager()->IsDialogOpen()) 76 return Status(kNoAlertOpen); 77 78 session->prompt_text.reset(new std::string(text)); 79 return Status(kOk); 80} 81 82Status ExecuteAcceptAlert( 83 Session* session, 84 WebView* web_view, 85 const base::DictionaryValue& params, 86 scoped_ptr<base::Value>* value) { 87 Status status = web_view->GetJavaScriptDialogManager() 88 ->HandleDialog(true, session->prompt_text.get()); 89 session->prompt_text.reset(); 90 return status; 91} 92 93Status ExecuteDismissAlert( 94 Session* session, 95 WebView* web_view, 96 const base::DictionaryValue& params, 97 scoped_ptr<base::Value>* value) { 98 Status status = web_view->GetJavaScriptDialogManager() 99 ->HandleDialog(false, session->prompt_text.get()); 100 session->prompt_text.reset(); 101 return status; 102} 103