blob: 35bab148bfc21596ebf6cf07d613812537adb89b [file] [log] [blame]
// Copyright 2017 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 <memory>
#include <vector>
#import "ios/web/public/ui/java_script_dialog_presenter.h"
namespace web {
struct FakeJavaScriptDialog {
WebState* web_state = nullptr;
GURL origin_url;
JavaScriptDialogType java_script_dialog_type;
NSString* message_text;
NSString* default_prompt_text;
DialogClosedCallback callback;
// Fake presenter to check that the JavaScriptDialogPresenter methods are called
// as expected. |RunJavaScriptDialog| always calls |callback| with
// |callback_success_argument| and |callback_user_input_argument| values.
class FakeJavaScriptDialogPresenter : public JavaScriptDialogPresenter {
~FakeJavaScriptDialogPresenter() override;
// JavaScriptDialogPresenter overrides:
void RunJavaScriptDialog(WebState* web_state,
const GURL& origin_url,
JavaScriptDialogType java_script_dialog_type,
NSString* message_text,
NSString* default_prompt_text,
DialogClosedCallback callback) override;
void CancelDialogs(WebState* web_state) override;
// Whether callback execution is paused. If a dialog presentation is
// requested while true, the callback will not be executed until unpaused.
bool callback_execution_paused() const { return callback_execution_paused_; }
void set_callback_execution_paused(bool callback_execution_paused) {
if (callback_execution_paused_ == callback_execution_paused)
callback_execution_paused_ = callback_execution_paused;
if (!callback_execution_paused_)
// True if the JavaScriptDialogPresenter CancelDialogs method has been called.
bool cancel_dialogs_called() const { return cancel_dialogs_called_; }
// Returns a vector of requested dialogs.
const std::vector<std::unique_ptr<FakeJavaScriptDialog>>& requested_dialogs()
const {
return requested_dialogs_;
// Sets |success| argument to be used for RunJavaScriptDialog callback.
void set_callback_success_argument(bool success) {
callback_success_argument_ = success;
// Sets |user_input| argument to be used for RunJavaScriptDialog callback.
void set_callback_user_input_argument(NSString* user_input) {
callback_user_input_argument_ = user_input;
// Executes all non-executed callbacks in |requested_dialogs_|.
void ExecuteAllDialogCallbacks();
// Executes the callback for |dialog|.
void ExecuteDialogCallback(FakeJavaScriptDialog* dialog);
bool callback_execution_paused_ = false;
bool cancel_dialogs_called_ = false;
std::vector<std::unique_ptr<FakeJavaScriptDialog>> requested_dialogs_;
bool callback_success_argument_ = false;
NSString* callback_user_input_argument_;
} // namespace web