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
129
130
131
132
133
|
// Copyright 2017 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/webui/print_preview/pdf_printer_handler.h"
#include <windows.h> // Must be in front of other Windows header files.
#include <commdlg.h>
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/memory/ref_counted.h"
#include "base/memory/ref_counted_memory.h"
#include "base/run_loop.h"
#include "chrome/browser/platform_util.h"
#include "chrome/browser/ui/browser_commands.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "ui/shell_dialogs/select_file_dialog_win.h"
#include "ui/shell_dialogs/select_file_policy.h"
using content::WebContents;
namespace printing {
namespace {
void ExecuteCancelledSelectFileDialog(
ui::SelectFileDialog::Type type,
const std::u16string& title,
const base::FilePath& default_path,
const std::vector<ui::FileFilterSpec>& filter,
int file_type_index,
const std::wstring& default_extension,
HWND owner,
ui::OnSelectFileExecutedCallback on_select_file_executed_callback) {
// Send an empty result to simulate a cancelled dialog.
std::move(on_select_file_executed_callback).Run({}, 0);
}
class FakePdfPrinterHandler : public PdfPrinterHandler {
public:
FakePdfPrinterHandler(Profile* profile,
content::WebContents* contents,
PrintPreviewStickySettings* sticky_settings)
: PdfPrinterHandler(profile, contents, sticky_settings),
save_failed_(false) {}
void FileSelected(const ui::SelectedFileInfo& file, int index) override {
// Since we always cancel the dialog as soon as it is initialized, this
// should never be called.
NOTREACHED();
}
void FileSelectionCanceled() override {
save_failed_ = true;
run_loop_.Quit();
}
void StartPrintToPdf(const std::u16string& job_title) {
StartPrint(job_title, /*settings=*/base::Value::Dict(),
/*print_data=*/nullptr, base::DoNothing());
run_loop_.Run();
}
bool save_failed() const { return save_failed_; }
private:
// Simplified version of select file to avoid checking preferences and sticky
// settings in the test
void SelectFile(const base::FilePath& default_filename,
content::WebContents* /* initiator */,
bool prompt_user) override {
ui::SelectFileDialog::FileTypeInfo file_type_info;
file_type_info.extensions.resize(1);
file_type_info.extensions[0].push_back(FILE_PATH_LITERAL("pdf"));
select_file_dialog_ = ui::CreateWinSelectFileDialog(
this, nullptr /*policy already checked*/,
base::BindRepeating(&ExecuteCancelledSelectFileDialog));
select_file_dialog_->SelectFile(
ui::SelectFileDialog::SELECT_SAVEAS_FILE, std::u16string(),
default_filename, &file_type_info, 0, base::FilePath::StringType(),
platform_util::GetTopLevel(preview_web_contents_->GetNativeView()));
}
bool save_failed_;
base::RunLoop run_loop_;
};
} // namespace
class PdfPrinterHandlerWinTest : public BrowserWithTestWindowTest {
public:
PdfPrinterHandlerWinTest() = default;
PdfPrinterHandlerWinTest(const PdfPrinterHandlerWinTest&) = delete;
PdfPrinterHandlerWinTest& operator=(const PdfPrinterHandlerWinTest&) = delete;
~PdfPrinterHandlerWinTest() override = default;
void SetUp() override {
BrowserWithTestWindowTest::SetUp();
// Create a new tab
chrome::NewTab(browser());
AddTab(browser(), GURL("chrome://print"));
// Create the PDF printer
pdf_printer_ = std::make_unique<FakePdfPrinterHandler>(
profile(), browser()->tab_strip_model()->GetWebContentsAt(0), nullptr);
}
protected:
std::unique_ptr<FakePdfPrinterHandler> pdf_printer_;
};
TEST_F(PdfPrinterHandlerWinTest, TestSaveAsPdf) {
pdf_printer_->StartPrintToPdf(u"111111111111111111111.html");
EXPECT_TRUE(pdf_printer_->save_failed());
}
TEST_F(PdfPrinterHandlerWinTest, TestSaveAsPdfLongFileName) {
pdf_printer_->StartPrintToPdf(
u"11111111111111111111111111111111111111111111111111111111111111111111111"
u"11111111111111111111111111111111111111111111111111111111111111111111111"
u"11111111111111111111111111111111111111111111111111111111111111111111111"
u"11111111111111111111111111111111111111111111111111111111111111111111111"
u"1111111111111111111111111111111111111111111111111.html");
EXPECT_TRUE(pdf_printer_->save_failed());
}
} // namespace printing
|
