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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef PRINTING_TEST_PRINTING_CONTEXT_H_
#define PRINTING_TEST_PRINTING_CONTEXT_H_
#include <memory>
#include <string>
#include "base/containers/flat_map.h"
#include "base/functional/callback_forward.h"
#include "build/build_config.h"
#include "printing/mojom/print.mojom.h"
#include "printing/print_settings.h"
#include "printing/printing_context.h"
#if BUILDFLAG(IS_WIN)
#include "third_party/abseil-cpp/absl/types/optional.h"
#endif
namespace printing {
class TestPrintingContextDelegate : public PrintingContext::Delegate {
public:
TestPrintingContextDelegate();
TestPrintingContextDelegate(const TestPrintingContextDelegate&) = delete;
TestPrintingContextDelegate& operator=(const TestPrintingContextDelegate&) =
delete;
~TestPrintingContextDelegate() override;
// PrintingContext::Delegate overrides:
gfx::NativeView GetParentView() override;
std::string GetAppLocale() override;
};
// Note that TestPrintingContext uses a PrintBackend internally, so tests that
// want to avoid using a real PrintingContext will also want to use
// PrintBackend::SetPrintBackendForTesting() to avoid using a real PrintBackend.
class TestPrintingContext : public PrintingContext {
public:
using OnNewDocumentCallback = base::RepeatingCallback<void(
#if BUILDFLAG(IS_MAC)
bool destination_is_preview,
#endif
const PrintSettings&)>;
TestPrintingContext(Delegate* delegate, ProcessBehavior process_behavior);
TestPrintingContext(const TestPrintingContext&) = delete;
TestPrintingContext& operator=(const TestPrintingContext&) = delete;
~TestPrintingContext() override;
// Methods for test setup:
// Provide settings that will be used as the current settings for the
// indicated device.
void SetDeviceSettings(const std::string& device_name,
std::unique_ptr<PrintSettings> settings);
// Provide the settings which should be applied to mimic a user's choices
// during AskUserForSettings().
void SetUserSettings(const PrintSettings& settings);
// Enables tests to fail with an access-denied error.
void SetNewDocumentBlockedByPermissions() {
new_document_blocked_by_permissions_ = true;
}
#if BUILDFLAG(IS_WIN)
void SetOnRenderPageBlockedByPermissions() {
render_page_blocked_by_permissions_ = true;
}
void SetOnRenderPageFailsForPage(uint32_t page_number) {
render_page_fail_for_page_number_ = page_number;
}
#endif
void SetOnRenderDocumentBlockedByPermissions() {
render_document_blocked_by_permissions_ = true;
}
void SetDocumentDoneBlockedByPermissions() {
document_done_blocked_by_permissions_ = true;
}
// Enables tests to fail with a failed error.
void SetNewDocumentFails() { new_document_fails_ = true; }
void SetUseDefaultSettingsFails() { use_default_settings_fails_ = true; }
// Enables tests to fail with a canceled error.
void SetNewDocumentCancels() { new_document_cancels_ = true; }
void SetAskUserForSettingsCanceled() { ask_user_for_settings_cancel_ = true; }
void SetOnNewDocumentCallback(OnNewDocumentCallback callback) {
on_new_document_callback_ = std::move(callback);
}
// PrintingContext overrides:
void AskUserForSettings(int max_pages,
bool has_selection,
bool is_scripted,
PrintSettingsCallback callback) override;
mojom::ResultCode UseDefaultSettings() override;
gfx::Size GetPdfPaperSizeDeviceUnits() override;
mojom::ResultCode UpdatePrinterSettings(
const PrinterSettings& printer_settings) override;
mojom::ResultCode NewDocument(const std::u16string& document_name) override;
#if BUILDFLAG(IS_WIN)
mojom::ResultCode RenderPage(const PrintedPage& page,
const PageSetup& page_setup) override;
#endif
mojom::ResultCode PrintDocument(const MetafilePlayer& metafile,
const PrintSettings& settings,
uint32_t num_pages) override;
mojom::ResultCode DocumentDone() override;
void Cancel() override;
void ReleaseContext() override;
NativeDrawingContext context() const override;
#if BUILDFLAG(IS_WIN)
mojom::ResultCode InitWithSettingsForTest(
std::unique_ptr<PrintSettings> settings) override;
#endif
private:
mojom::ResultCode AskUserForSettingsImpl(int max_pages,
bool has_selection,
bool is_scripted);
// Simulation of platform drivers' default settings.
base::flat_map<std::string, std::unique_ptr<PrintSettings>> device_settings_;
// Settings to apply to mimic a user's choices in `AskUserForSettings()`.
absl::optional<PrintSettings> user_settings_;
// Platform implementations of `PrintingContext` apply PrintSettings to the
// respective device contexts. Once the upper printing layers call
// `TakeAndResetSettings()`, the values in `settings_` no longer reflect
// what the printer driver's device context are set for.
// Simulate this by capturing what the settings are whenever `settings_` is
// applied to a device context.
PrintSettings applied_settings_;
#if BUILDFLAG(IS_MAC)
// When printing a new document, Preview app is a special macOS destination.
// This member is used to track when this was indicated as the destination to
// use in `UpdatePrinterSettings()`.
bool destination_is_preview_ = false;
#endif
bool use_default_settings_fails_ = false;
bool ask_user_for_settings_cancel_ = false;
bool new_document_cancels_ = false;
bool new_document_fails_ = false;
bool new_document_blocked_by_permissions_ = false;
#if BUILDFLAG(IS_WIN)
bool render_page_blocked_by_permissions_ = false;
absl::optional<uint32_t> render_page_fail_for_page_number_;
#endif
bool render_document_blocked_by_permissions_ = false;
bool document_done_blocked_by_permissions_ = false;
// Called every time `NewDocument()` is called. Provides a copy of the
// effective device context settings.
OnNewDocumentCallback on_new_document_callback_;
};
} // namespace printing
#endif // PRINTING_TEST_PRINTING_CONTEXT_H_
|
