blob: 184625436cc34f8054fb540bc5f802754fda11d3 [file] [log] [blame]
// Copyright 2019 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 "base/callback.h"
#include "base/compiler_specific.h"
#include "base/memory/weak_ptr.h"
#include "base/strings/string16.h"
#include "base/strings/string_piece.h"
#include "ui/shell_dialogs/select_file_dialog.h"
#include "ui/shell_dialogs/select_file_dialog_factory.h"
#include "ui/shell_dialogs/shell_dialogs_export.h"
namespace ui {
// A test fake SelectFileDialog. Usage:
// FakeSelectFileDialog::Factory* factory =
// FakeSelectFileDialog::RegisterFactory();
// factory->SetOpenCallback(open_callback);
// Now calls to SelectFileDialog::Create() will create a |FakeSelectFileDialog|,
// and open_callback is invoked when the dialog is opened.
// Once the dialog is opened, use factory->GetLastDialog() to access the dialog
// to query file information and select a file.
class FakeSelectFileDialog : public SelectFileDialog {
// A |FakeSelectFileDialog::Factory| which creates |FakeSelectFileDialog|
// instances.
class Factory : public SelectFileDialogFactory {
~Factory() override;
Factory(const Factory&) = delete;
Factory& operator=(const Factory&) = delete;
// SelectFileDialog::Factory.
ui::SelectFileDialog* Create(
ui::SelectFileDialog::Listener* listener,
std::unique_ptr<ui::SelectFilePolicy> policy) override;
// Returns the last opened dialog, or null if one has not been opened yet.
FakeSelectFileDialog* GetLastDialog() const;
// Sets a callback to be called when a new fake dialog has been opened.
void SetOpenCallback(base::RepeatingClosure callback);
base::RepeatingClosure opened_callback_;
base::WeakPtr<FakeSelectFileDialog> last_dialog_;
// Creates a |Factory| and registers it with |SelectFileDialog::SetFactory|,
// which owns the new factory. This factory will create new
// |FakeSelectFileDialog| instances upon calls to
// |SelectFileDialog::Create()|.
static Factory* RegisterFactory();
FakeSelectFileDialog(const base::RepeatingClosure& opened,
Listener* listener,
std::unique_ptr<SelectFilePolicy> policy);
FakeSelectFileDialog(const FakeSelectFileDialog&) = delete;
FakeSelectFileDialog& operator=(const FakeSelectFileDialog&) = delete;
// SelectFileDialog.
void SelectFileImpl(Type type,
const base::string16& title,
const base::FilePath& default_path,
const FileTypeInfo* file_types,
int file_type_index,
const base::FilePath::StringType& default_extension,
gfx::NativeWindow owning_window,
void* params) override;
bool HasMultipleFileTypeChoicesImpl() override;
bool IsRunning(gfx::NativeWindow owning_window) const override;
void ListenerDestroyed() override {}
// Returns the file title provided to the dialog.
const base::string16& title() const { return title_; }
// Returns the file types provided to the dialog.
const FileTypeInfo& file_types() const { return file_types_; }
// Returns the default file extension provided to the dialog.
const std::string& default_extension() const { return default_extension_; }
// Calls the |FileSelected()| method on listener(). |filter_text| selects
// which file extension filter to report.
bool CallFileSelected(const base::FilePath& file_path,
base::StringPiece filter_text) WARN_UNUSED_RESULT;
base::WeakPtr<FakeSelectFileDialog> GetWeakPtr() {
return weak_ptr_factory_.GetWeakPtr();
~FakeSelectFileDialog() override;
base::RepeatingClosure opened_;
base::string16 title_;
FileTypeInfo file_types_;
std::string default_extension_;
void* params_;
base::WeakPtrFactory<FakeSelectFileDialog> weak_ptr_factory_{this};
} // namespace ui