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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "remoting/host/file_transfer/file_chooser_chromeos.h"
#include <algorithm>
#include <memory>
#include "base/check_deref.h"
#include "base/files/file_path.h"
#include "base/functional/bind.h"
#include "base/location.h"
#include "base/logging.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/task/single_thread_task_runner.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "content/public/test/file_system_chooser_test_helpers.h"
#include "remoting/host/chromeos/scoped_fake_ash_proxy.h"
#include "remoting/host/file_transfer/file_chooser.h"
#include "remoting/protocol/file_transfer_helpers.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/aura/env.h"
#include "ui/shell_dialogs/select_file_dialog.h"
#include "ui/shell_dialogs/select_file_dialog_factory.h"
using base::test::TestFuture;
namespace remoting {
namespace {
const base::FilePath kTestFilePath("/path/to/a/test/file.txt");
} // namespace
class FileChooserChromeOsTest : public testing::Test {
public:
FileChooserChromeOsTest() = default;
FileChooserChromeOsTest(const FileChooserChromeOsTest&) = delete;
FileChooserChromeOsTest& operator=(const FileChooserChromeOsTest&) = delete;
// `testing::Test` implementation.
void TearDown() override { ui::SelectFileDialog::SetFactory(nullptr); }
void CreateAndShowFileChooser(FileChooser::ResultCallback callback) {
file_chooser_ = FileChooser::Create(ui_task_runner(), std::move(callback));
file_chooser_->Show();
}
void SetResultFileSelectionFactory(base::FilePath file_path) {
ui::SelectFileDialog::SetFactory(
std::make_unique<content::FakeSelectFileDialogFactory>(
std::vector<base::FilePath>{file_path}, &dialog_params_));
}
void SetCancelFileSelectionFactory() {
ui::SelectFileDialog::SetFactory(
std::make_unique<content::CancellingSelectFileDialogFactory>(
&dialog_params_));
}
ui::SelectFileDialog::Type GetRequestedDialogType() {
return dialog_params_.type;
}
scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner() {
return task_environment_.GetMainThreadTaskRunner();
}
private:
test::ScopedFakeAshProxy ash_proxy_;
content::SelectFileDialogParams dialog_params_;
std::unique_ptr<FileChooser> file_chooser_;
base::test::SingleThreadTaskEnvironment task_environment_;
};
#if defined(LEAK_SANITIZER)
// TODO(crbug.com/40916564): Fix LeakSanitizer failure.
#define MAYBE_DISABLED(name) DISABLED_##name
#else
#define MAYBE_DISABLED(name) name
#endif
TEST_F(FileChooserChromeOsTest, MAYBE_DISABLED(SingleFileSelection)) {
base::test::TestFuture<FileChooser::Result> result_future;
SetResultFileSelectionFactory(kTestFilePath);
CreateAndShowFileChooser(result_future.GetCallback());
ASSERT_TRUE(result_future.Get().is_success());
EXPECT_EQ(result_future.Get().success(), kTestFilePath);
}
TEST_F(FileChooserChromeOsTest, MAYBE_DISABLED(FileCancelationAllowed)) {
base::test::TestFuture<FileChooser::Result> result_future;
SetCancelFileSelectionFactory();
CreateAndShowFileChooser(result_future.GetCallback());
ASSERT_TRUE(result_future.Get().is_error());
EXPECT_EQ(result_future.Get().error().type(),
protocol::FileTransfer_Error_Type_CANCELED);
}
TEST_F(FileChooserChromeOsTest, MAYBE_DISABLED(OnlyAllowSingleFileSelection)) {
base::test::TestFuture<FileChooser::Result> result_future;
SetResultFileSelectionFactory(kTestFilePath);
CreateAndShowFileChooser(result_future.GetCallback());
ASSERT_TRUE(result_future.Wait());
EXPECT_EQ(GetRequestedDialogType(),
ui::SelectFileDialog::Type::SELECT_OPEN_FILE);
}
} // namespace remoting
|
