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
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
|
// Copyright (c) 2011 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/command_line.h"
#include "base/json/json_writer.h"
#include "base/values.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/printing/background_printing_manager.h"
#include "chrome/browser/printing/print_preview_tab_controller.h"
#include "chrome/browser/printing/print_view_manager.h"
#include "chrome/browser/ui/browser_list.h"
#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
#include "chrome/browser/ui/webui/print_preview_handler.h"
#include "chrome/browser/ui/webui/print_preview_ui.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "content/browser/tab_contents/tab_contents.h"
#include "printing/page_size_margins.h"
#include "printing/print_job_constants.h"
namespace {
DictionaryValue* GetCustomMarginsDictionary(
const double margin_top, const double margin_right,
const double margin_bottom, const double margin_left) {
base::DictionaryValue* custom_settings = new base::DictionaryValue();
custom_settings->SetDouble(printing::kSettingMarginTop, margin_top);
custom_settings->SetDouble(printing::kSettingMarginRight, margin_right);
custom_settings->SetDouble(printing::kSettingMarginBottom, margin_bottom);
custom_settings->SetDouble(printing::kSettingMarginLeft, margin_left);
return custom_settings;
}
} // namespace
class PrintPreviewHandlerTest : public BrowserWithTestWindowTest {
protected:
void SetUp() {
BrowserWithTestWindowTest::SetUp();
#if !defined(GOOGLE_CHROME_BUILD) || defined(OS_CHROMEOS)
CommandLine::ForCurrentProcess()->AppendSwitch(
switches::kEnablePrintPreview);
#endif
ASSERT_TRUE(browser());
BrowserList::SetLastActive(browser());
ASSERT_TRUE(BrowserList::GetLastActive());
browser()->NewTab();
EXPECT_EQ(1, browser()->tab_count());
OpenPrintPreviewTab();
}
virtual void TearDown() {
DeletePrintPreviewTab();
ClearStickySettings();
}
void OpenPrintPreviewTab() {
TabContentsWrapper* initiator_tab =
browser()->GetSelectedTabContentsWrapper();
ASSERT_TRUE(initiator_tab);
printing::PrintPreviewTabController* controller =
printing::PrintPreviewTabController::GetInstance();
ASSERT_TRUE(controller);
initiator_tab->print_view_manager()->PrintPreviewNow();
preview_tab_ = controller->GetOrCreatePreviewTab(initiator_tab);
ASSERT_TRUE(preview_tab_);
preview_ui_ =
static_cast<PrintPreviewUI*>(preview_tab_->tab_contents()->web_ui());
ASSERT_TRUE(preview_ui_);
}
void DeletePrintPreviewTab() {
printing::BackgroundPrintingManager* bg_printing_manager =
g_browser_process->background_printing_manager();
ASSERT_TRUE(bg_printing_manager->HasPrintPreviewTab(preview_tab_));
// Deleting TabContentsWrapper* to avoid warings from pref_notifier_impl.cc
// after the test ends.
delete preview_tab_;
}
void CheckCustomMargins(const double margin_top,
const double margin_right,
const double margin_bottom,
const double margin_left) {
EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_top,
margin_top);
EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_right,
margin_right);
EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_bottom,
margin_bottom);
EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_left,
margin_left);
}
void RequestPrintWithDefaultMargins() {
// Set the minimal dummy settings to make the HandlePrint() code happy.
DictionaryValue settings;
settings.SetBoolean(printing::kSettingPreviewModifiable, true);
settings.SetInteger(printing::kSettingColor, printing::COLOR);
settings.SetBoolean(printing::kSettingPrintToPDF, false);
settings.SetInteger(printing::kSettingMarginsType,
printing::DEFAULT_MARGINS);
// Put |settings| in to |args| as a JSON string.
std::string json_string;
base::JSONWriter::Write(&settings, false, &json_string);
ListValue args;
args.Append(new base::StringValue(json_string)); // |args| takes ownership.
preview_ui_->handler_->HandlePrint(&args);
}
void RequestPrintWithCustomMargins(
const double margin_top, const double margin_right,
const double margin_bottom, const double margin_left) {
// Set the minimal dummy settings to make the HandlePrint() code happy.
DictionaryValue settings;
settings.SetBoolean(printing::kSettingPreviewModifiable, true);
settings.SetInteger(printing::kSettingColor, printing::COLOR);
settings.SetBoolean(printing::kSettingPrintToPDF, false);
settings.SetInteger(printing::kSettingMarginsType,
printing::CUSTOM_MARGINS);
// Creating custom margins dictionary and nesting it in |settings|.
DictionaryValue* custom_settings = GetCustomMarginsDictionary(
margin_top, margin_right, margin_bottom, margin_left);
// |settings| takes ownership.
settings.Set(printing::kSettingMarginsCustom, custom_settings);
// Put |settings| in to |args| as a JSON string.
std::string json_string;
base::JSONWriter::Write(&settings, false, &json_string);
ListValue args;
args.Append(new base::StringValue(json_string)); // |args| takes ownership.
preview_ui_->handler_->HandlePrint(&args);
}
TabContentsWrapper* preview_tab_;
PrintPreviewUI* preview_ui_;
private:
void ClearStickySettings() {
PrintPreviewHandler::last_used_margins_type_ = printing::DEFAULT_MARGINS;
delete PrintPreviewHandler::last_used_page_size_margins_;
PrintPreviewHandler::last_used_page_size_margins_ = NULL;
}
};
// Tests that margin settings are saved correctly when printing with custom
// margins selected.
TEST_F(PrintPreviewHandlerTest, StickyMarginsCustom) {
const double kMarginTop = 25.5;
const double kMarginRight = 26.5;
const double kMarginBottom = 27.5;
const double kMarginLeft = 28.5;
RequestPrintWithCustomMargins(
kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
EXPECT_EQ(1, browser()->tab_count());
// Checking that sticky settings were saved correctly.
EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
printing::CUSTOM_MARGINS);
ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
}
// Tests that margin settings are saved correctly when printing with default
// margins selected.
TEST_F(PrintPreviewHandlerTest, StickyMarginsDefault) {
RequestPrintWithDefaultMargins();
EXPECT_EQ(1, browser()->tab_count());
// Checking that sticky settings were saved correctly.
EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
printing::DEFAULT_MARGINS);
ASSERT_FALSE(PrintPreviewHandler::last_used_page_size_margins_);
}
// Tests that margin settings are saved correctly when printing with custom
// margins selected and then again with default margins selected.
TEST_F(PrintPreviewHandlerTest, StickyMarginsCustomThenDefault) {
const double kMarginTop = 125.5;
const double kMarginRight = 126.5;
const double kMarginBottom = 127.5;
const double kMarginLeft = 128.5;
RequestPrintWithCustomMargins(
kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
EXPECT_EQ(1, browser()->tab_count());
DeletePrintPreviewTab();
EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
printing::CUSTOM_MARGINS);
ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
OpenPrintPreviewTab();
RequestPrintWithDefaultMargins();
// Checking that sticky settings were saved correctly.
EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
printing::DEFAULT_MARGINS);
ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
}
// Tests that margin settings are retrieved correctly after printing with custom
// margins.
TEST_F(PrintPreviewHandlerTest, GetLastUsedMarginSettingsCustom) {
const double kMarginTop = 125.5;
const double kMarginRight = 126.5;
const double kMarginBottom = 127.5;
const double kMarginLeft = 128.5;
RequestPrintWithCustomMargins(
kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
base::DictionaryValue initial_settings;
preview_ui_->handler_->GetLastUsedMarginSettings(&initial_settings);
int margins_type;
EXPECT_TRUE(initial_settings.GetInteger(printing::kSettingMarginsType,
&margins_type));
EXPECT_EQ(margins_type, printing::CUSTOM_MARGINS);
double margin_value;
EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginTop,
&margin_value));
EXPECT_EQ(kMarginTop, margin_value);
EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginRight,
&margin_value));
EXPECT_EQ(kMarginRight, margin_value);
EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginBottom,
&margin_value));
EXPECT_EQ(kMarginBottom, margin_value);
EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginLeft,
&margin_value));
EXPECT_EQ(kMarginLeft, margin_value);
}
// Tests that margin settings are retrieved correctly after printing with
// default margins.
TEST_F(PrintPreviewHandlerTest, GetLastUsedMarginSettingsDefault) {
RequestPrintWithDefaultMargins();
base::DictionaryValue initial_settings;
preview_ui_->handler_->GetLastUsedMarginSettings(&initial_settings);
int margins_type;
EXPECT_TRUE(initial_settings.GetInteger(printing::kSettingMarginsType,
&margins_type));
EXPECT_EQ(margins_type, printing::DEFAULT_MARGINS);
double margin_value;
EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginTop,
&margin_value));
EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginRight,
&margin_value));
EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginBottom,
&margin_value));
EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginLeft,
&margin_value));
}
|
