1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/views/select_file_dialog_extension.h"
#include <memory>
#include "base/files/file_path.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/shell_dialogs/select_file_policy.h"
#include "ui/shell_dialogs/selected_file_info.h"
namespace {
const SelectFileDialogExtension::RoutingID kDefaultRoutingID =
SelectFileDialogExtension::RoutingID();
} // namespace
// Must be a class so it can be a friend of SelectFileDialogExtension.
class SelectFileDialogExtensionTest : public ::testing::Test {
public:
SelectFileDialogExtensionTest() = default;
SelectFileDialogExtensionTest(const SelectFileDialogExtensionTest&) = delete;
SelectFileDialogExtensionTest& operator=(
const SelectFileDialogExtensionTest&) = delete;
static SelectFileDialogExtension* CreateDialog(
ui::SelectFileDialog::Listener* listener) {
SelectFileDialogExtension* dialog =
new SelectFileDialogExtension(listener, nullptr);
// Simulate the dialog opening.
EXPECT_FALSE(SelectFileDialogExtension::PendingExists(kDefaultRoutingID));
dialog->AddPending(kDefaultRoutingID);
EXPECT_TRUE(SelectFileDialogExtension::PendingExists(kDefaultRoutingID));
return dialog;
}
};
// Test listener for a SelectFileDialog.
class TestListener : public ui::SelectFileDialog::Listener {
public:
TestListener() : selected_(false), file_index_(-1) {}
TestListener(const TestListener&) = delete;
TestListener& operator=(const TestListener&) = delete;
~TestListener() override {}
bool selected() const { return selected_; }
int file_index() const { return file_index_; }
// ui::SelectFileDialog::Listener implementation
void FileSelected(const base::FilePath& path,
int index,
void* params) override {
selected_ = true;
file_index_ = index;
}
private:
bool selected_;
int file_index_;
};
// Client of a SelectFileDialog that deletes itself whenever the dialog
// is closed. This is a common pattern in UI code.
class SelfDeletingClient : public ui::SelectFileDialog::Listener {
public:
SelfDeletingClient() {
dialog_ = SelectFileDialogExtensionTest::CreateDialog(this);
}
~SelfDeletingClient() override {
if (dialog_.get())
dialog_->ListenerDestroyed();
}
SelectFileDialogExtension* dialog() const { return dialog_.get(); }
// ui::SelectFileDialog::Listener implementation
void FileSelected(const base::FilePath& path,
int index,
void* params) override {
delete this;
}
private:
scoped_refptr<SelectFileDialogExtension> dialog_;
};
TEST_F(SelectFileDialogExtensionTest, FileSelected) {
const int kFileIndex = 5;
auto listener = std::make_unique<TestListener>();
scoped_refptr<SelectFileDialogExtension> dialog =
CreateDialog(listener.get());
// Simulate selecting a file.
ui::SelectedFileInfo info;
SelectFileDialogExtension::OnFileSelected(kDefaultRoutingID, info,
kFileIndex);
// Simulate closing the dialog so the listener gets invoked.
dialog->OnSystemDialogWillClose();
EXPECT_TRUE(listener->selected());
EXPECT_EQ(kFileIndex, listener->file_index());
}
TEST_F(SelectFileDialogExtensionTest, FileSelectionCanceled) {
auto listener = std::make_unique<TestListener>();
scoped_refptr<SelectFileDialogExtension> dialog =
CreateDialog(listener.get());
// Simulate cancelling the dialog.
SelectFileDialogExtension::OnFileSelectionCanceled(kDefaultRoutingID);
// Simulate closing the dialog so the listener gets invoked.
dialog->OnSystemDialogWillClose();
EXPECT_FALSE(listener->selected());
EXPECT_EQ(-1, listener->file_index());
}
TEST_F(SelectFileDialogExtensionTest, SelfDeleting) {
SelfDeletingClient* client = new SelfDeletingClient();
// Ensure we don't crash or trip an Address Sanitizer warning about
// use-after-free.
ui::SelectedFileInfo file_info;
SelectFileDialogExtension::OnFileSelected(kDefaultRoutingID, file_info, 0);
// Simulate closing the dialog so the listener gets invoked.
client->dialog()->OnSystemDialogWillClose();
}
|
