blob: 4cf3b742c1e0c31cb55182f5bf255ade4bf0e25d [file] [log] [blame]
// Copyright (c) 2012 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 "components/printing/test/print_mock_render_thread.h"
#include <vector>
#include "base/values.h"
#include "components/printing/test/mock_printer.h"
#include "ipc/ipc_sync_message.h"
#include "printing/page_range.h"
#include "printing/print_job_constants.h"
#include "testing/gtest/include/gtest/gtest.h"
#if defined(OS_CHROMEOS)
#include <fcntl.h>
#include "base/files/file_util.h"
#endif
#if defined(ENABLE_PRINTING)
#include "components/printing/common/print_messages.h"
#endif
PrintMockRenderThread::PrintMockRenderThread()
#if defined(ENABLE_PRINTING)
: printer_(new MockPrinter),
print_dialog_user_response_(true),
print_preview_cancel_page_number_(-1),
print_preview_pages_remaining_(0)
#endif
{
}
PrintMockRenderThread::~PrintMockRenderThread() {
}
scoped_refptr<base::MessageLoopProxy>
PrintMockRenderThread::GetIOMessageLoopProxy() {
return io_message_loop_proxy_;
}
void PrintMockRenderThread::set_io_message_loop_proxy(
const scoped_refptr<base::MessageLoopProxy>& proxy) {
io_message_loop_proxy_ = proxy;
}
bool PrintMockRenderThread::OnMessageReceived(const IPC::Message& msg) {
if (content::MockRenderThread::OnMessageReceived(msg))
return true;
// Some messages we do special handling.
bool handled = true;
IPC_BEGIN_MESSAGE_MAP(PrintMockRenderThread, msg)
#if defined(ENABLE_PRINTING)
IPC_MESSAGE_HANDLER(PrintHostMsg_GetDefaultPrintSettings,
OnGetDefaultPrintSettings)
IPC_MESSAGE_HANDLER(PrintHostMsg_ScriptedPrint, OnScriptedPrint)
IPC_MESSAGE_HANDLER(PrintHostMsg_UpdatePrintSettings, OnUpdatePrintSettings)
IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPrintedPagesCount,
OnDidGetPrintedPagesCount)
IPC_MESSAGE_HANDLER(PrintHostMsg_DidPrintPage, OnDidPrintPage)
IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount,
OnDidGetPreviewPageCount)
IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage, OnDidPreviewPage)
IPC_MESSAGE_HANDLER(PrintHostMsg_CheckForCancel, OnCheckForCancel)
#if defined(OS_WIN)
IPC_MESSAGE_HANDLER(PrintHostMsg_DuplicateSection, OnDuplicateSection)
#endif
#if defined(OS_CHROMEOS)
IPC_MESSAGE_HANDLER(PrintHostMsg_AllocateTempFileForPrinting,
OnAllocateTempFileForPrinting)
IPC_MESSAGE_HANDLER(PrintHostMsg_TempFileForPrintingWritten,
OnTempFileForPrintingWritten)
#endif // defined(OS_CHROMEOS)
#endif // defined(ENABLE_PRINTING)
IPC_MESSAGE_UNHANDLED(handled = false)
IPC_END_MESSAGE_MAP()
return handled;
}
#if defined(ENABLE_PRINTING)
#if defined(OS_CHROMEOS)
void PrintMockRenderThread::OnAllocateTempFileForPrinting(
int render_view_id,
base::FileDescriptor* renderer_fd,
int* browser_fd) {
renderer_fd->fd = *browser_fd = -1;
renderer_fd->auto_close = false;
base::FilePath path;
if (base::CreateTemporaryFile(&path)) {
int fd = open(path.value().c_str(), O_WRONLY);
DCHECK_GE(fd, 0);
renderer_fd->fd = *browser_fd = fd;
}
}
void PrintMockRenderThread::OnTempFileForPrintingWritten(int render_view_id,
int browser_fd) {
close(browser_fd);
}
#endif // defined(OS_CHROMEOS)
void PrintMockRenderThread::OnGetDefaultPrintSettings(
PrintMsg_Print_Params* params) {
printer_->GetDefaultPrintSettings(params);
}
void PrintMockRenderThread::OnScriptedPrint(
const PrintHostMsg_ScriptedPrint_Params& params,
PrintMsg_PrintPages_Params* settings) {
if (print_dialog_user_response_) {
printer_->ScriptedPrint(params.cookie, params.expected_pages_count,
params.has_selection, settings);
}
}
void PrintMockRenderThread::OnDidGetPrintedPagesCount(int cookie,
int number_pages) {
printer_->SetPrintedPagesCount(cookie, number_pages);
}
void PrintMockRenderThread::OnDidPrintPage(
const PrintHostMsg_DidPrintPage_Params& params) {
printer_->PrintPage(params);
}
void PrintMockRenderThread::OnDidGetPreviewPageCount(
const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
print_preview_pages_remaining_ = params.page_count;
}
void PrintMockRenderThread::OnDidPreviewPage(
const PrintHostMsg_DidPreviewPage_Params& params) {
DCHECK_GE(params.page_number, printing::FIRST_PAGE_INDEX);
print_preview_pages_remaining_--;
}
void PrintMockRenderThread::OnCheckForCancel(int32 preview_ui_id,
int preview_request_id,
bool* cancel) {
*cancel =
(print_preview_pages_remaining_ == print_preview_cancel_page_number_);
}
void PrintMockRenderThread::OnUpdatePrintSettings(
int document_cookie,
const base::DictionaryValue& job_settings,
PrintMsg_PrintPages_Params* params,
bool* canceled) {
if (canceled)
*canceled = false;
// Check and make sure the required settings are all there.
// We don't actually care about the values.
std::string dummy_string;
int margins_type = 0;
if (!job_settings.GetBoolean(printing::kSettingLandscape, NULL) ||
!job_settings.GetBoolean(printing::kSettingCollate, NULL) ||
!job_settings.GetInteger(printing::kSettingColor, NULL) ||
!job_settings.GetBoolean(printing::kSettingPrintToPDF, NULL) ||
!job_settings.GetBoolean(printing::kIsFirstRequest, NULL) ||
!job_settings.GetString(printing::kSettingDeviceName, &dummy_string) ||
!job_settings.GetInteger(printing::kSettingDuplexMode, NULL) ||
!job_settings.GetInteger(printing::kSettingCopies, NULL) ||
!job_settings.GetInteger(printing::kPreviewUIID, NULL) ||
!job_settings.GetInteger(printing::kPreviewRequestID, NULL) ||
!job_settings.GetInteger(printing::kSettingMarginsType, &margins_type)) {
return;
}
// Just return the default settings.
const base::ListValue* page_range_array;
printing::PageRanges new_ranges;
if (job_settings.GetList(printing::kSettingPageRange, &page_range_array)) {
for (size_t index = 0; index < page_range_array->GetSize(); ++index) {
const base::DictionaryValue* dict;
if (!page_range_array->GetDictionary(index, &dict))
continue;
printing::PageRange range;
if (!dict->GetInteger(printing::kSettingPageRangeFrom, &range.from) ||
!dict->GetInteger(printing::kSettingPageRangeTo, &range.to)) {
continue;
}
// Page numbers are 1-based in the dictionary.
// Page numbers are 0-based for the printing context.
range.from--;
range.to--;
new_ranges.push_back(range);
}
}
std::vector<int> pages(printing::PageRange::GetPages(new_ranges));
printer_->UpdateSettings(document_cookie, params, pages, margins_type);
job_settings.GetBoolean(printing::kSettingShouldPrintSelectionOnly,
&params->params.selection_only);
job_settings.GetBoolean(printing::kSettingShouldPrintBackgrounds,
&params->params.should_print_backgrounds);
}
MockPrinter* PrintMockRenderThread::printer() {
return printer_.get();
}
void PrintMockRenderThread::set_print_dialog_user_response(bool response) {
print_dialog_user_response_ = response;
}
void PrintMockRenderThread::set_print_preview_cancel_page_number(int page) {
print_preview_cancel_page_number_ = page;
}
int PrintMockRenderThread::print_preview_pages_remaining() const {
return print_preview_pages_remaining_;
}
#endif // defined(ENABLE_PRINTING)