Chromium Code Reviews| Index: chrome/browser/printing/print_preview_pdf_generated_browsertest.cc |
| diff --git a/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc b/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..1e4fea62b1c8fc9bdc46d08d09c11290cab022dc |
| --- /dev/null |
| +++ b/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc |
| @@ -0,0 +1,715 @@ |
| +// Copyright 2014 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 <algorithm> |
| +#include <fstream> |
| +#include <iostream> |
| +#include <iterator> |
| +#include <limits> |
| +#include <string> |
| +#include <utility> |
| +#include <vector> |
| + |
| +#include "base/file_util.h" |
| +#include "base/files/file.h" |
| +#include "base/files/file_path.h" |
| +#include "base/files/scoped_temp_dir.h" |
| +#include "base/logging.h" |
| +#include "base/md5.h" |
| +#include "base/memory/scoped_ptr.h" |
| +#include "base/path_service.h" |
| +#include "base/run_loop.h" |
| +#include "base/scoped_native_library.h" |
| +#include "base/strings/string_split.h" |
| +#include "base/strings/utf_string_conversions.h" |
| +#include "chrome/browser/printing/print_preview_dialog_controller.h" |
| +#include "chrome/browser/ui/browser.h" |
| +#include "chrome/browser/ui/browser_commands.h" |
| +#include "chrome/browser/ui/tabs/tab_strip_model.h" |
| +#include "chrome/browser/ui/webui/print_preview/print_preview_ui.h" |
| +#include "chrome/common/chrome_paths.h" |
| +#include "chrome/common/print_messages.h" |
| +#include "chrome/test/base/in_process_browser_test.h" |
| +#include "chrome/test/base/ui_test_utils.h" |
| +#include "content/public/browser/web_contents.h" |
| +#include "content/public/browser/web_ui_message_handler.h" |
| +#include "content/public/test/browser_test_utils.h" |
| +#include "ipc/ipc_message_macros.h" |
| +#include "net/base/filename_util.h" |
| +#include "printing/pdf_render_settings.h" |
| +#include "printing/units.h" |
| +#include "ui/gfx/codec/png_codec.h" |
| +#include "ui/gfx/geometry/rect.h" |
| +#include "url/gurl.h" |
| + |
| +#if defined(OS_WIN) |
| +#include <fcntl.h> |
| +#include <io.h> |
| +#endif |
| + |
| +using content::WebContents; |
| +using content::WebContentsObserver; |
| + |
| +namespace printing { |
| + |
| +// Number of color channels in a BGRA bitmap. |
| +const int kColorChannels = 4; |
| +const int kDpi = 300; |
| + |
| +// Every state is used when the document is a non-PDF source. When the source is |
| +// a PDF, kWaitingToSendSaveAsPDF, kWaitingToSendPageNumbers, and |
| +// kWaitingForFinalMessage are the only states used. |
| +enum State { |
| + // Waiting for the first message so the program can select Save as PDF |
| + kWaitingToSendSaveAsPdf = 0, |
| + // Waiting for the second message so the test can set the layout |
| + kWaitingToSendLayoutSettings = 1, |
| + // Waiting for the third message so the test can set the page numbers |
| + kWaitingToSendPageNumbers = 2, |
| + // Waiting for the forth message so the test can set the headers checkbox |
| + kWaitingToSendHeadersAndFooters = 3, |
| + // Waiting for the fifth message so the test can set the background checkbox |
| + kWaitingToSendBackgroundColorsAndImages = 4, |
| + // Waiting for the sixth message so the test can set the margins combobox |
| + kWaitingToSendMargins = 5, |
| + // Waiting for the final message so the program can save to PDF. |
| + kWaitingForFinalMessage = 6, |
| +}; |
| + |
| +// Settings for print preview. It reflects the current options provided by |
| +// print preview. If more options are added, more states should be added and |
| +// there should be more settings added to this struct. |
| +struct PrintPreviewSettings { |
| + PrintPreviewSettings(bool is_portrait, |
| + std::string page_numbers, |
| + bool headers_and_footers, |
| + bool background_colors_and_images, |
| + MarginType margins, |
| + bool source_is_pdf) |
| + : is_portrait(is_portrait), |
| + page_numbers(page_numbers), |
| + headers_and_footers(headers_and_footers), |
| + background_colors_and_images(background_colors_and_images), |
| + margins(margins), |
| + source_is_pdf(source_is_pdf) {} |
| + |
| + bool is_portrait; |
| + std::string page_numbers; |
| + bool headers_and_footers; |
| + bool background_colors_and_images; |
| + MarginType margins; |
| + bool source_is_pdf; |
| +}; |
| + |
| +// Observes the print preview webpage. Once it observes the PreviewPageCount |
| +// message, will send a sequence of commands to the print preview dialog and |
| +// change the settings of the preview dialog. |
| +class PrintPreviewObserver : public WebContentsObserver { |
| + public: |
| + PrintPreviewObserver(Browser* browser, WebContents* dialog) |
| + : WebContentsObserver(dialog), |
| + browser_(browser), |
| + state_(kWaitingToSendSaveAsPdf), |
| + failed_setting_("None") {} |
| + |
| + virtual ~PrintPreviewObserver() {} |
| + |
| + // Sets closure for the observer so that it can end the loop. |
| + void set_quit_closure(const base::Closure &closure) { |
| + quit_closure_ = closure; |
| + } |
| + |
| + // Actually stops the message loop so that the test can proceed. |
| + void EndLoop() { |
| + base::MessageLoop::current()->PostTask(FROM_HERE, quit_closure_); |
| + } |
| + |
| + bool OnMessageReceived(const IPC::Message& message) OVERRIDE { |
| + IPC_BEGIN_MESSAGE_MAP(PrintPreviewObserver, message) |
| + IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount, |
| + OnDidGetPreviewPageCount) |
| + IPC_END_MESSAGE_MAP(); |
| + return false; |
| + } |
| + |
| + // Gets the web contents for the print preview dialog so that the UI and |
| + // other elements can be accessed. |
| + WebContents* GetDialog() { |
| + WebContents* tab = browser_->tab_strip_model()->GetActiveWebContents(); |
| + PrintPreviewDialogController* dialog_controller = |
| + PrintPreviewDialogController::GetInstance(); |
| + return dialog_controller->GetPrintPreviewForContents(tab); |
| + } |
| + |
| + // Gets the PrintPreviewUI so that certain elements can be accessed. |
| + PrintPreviewUI* GetUI() { |
| + return static_cast<PrintPreviewUI*>( |
| + GetDialog()->GetWebUI()->GetController()); |
| + } |
| + |
| + // Calls native_layer.onManipulateSettingsForTest() and sends a dictionary |
| + // value containing the type of setting and the value to set that settings |
| + // to. |
| + void ManipulatePreviewSettings() { |
| + base::DictionaryValue script_argument; |
| + |
| + if (state_ == kWaitingToSendSaveAsPdf) { |
| + script_argument.SetBoolean("selectSaveAsPdfDestination", true); |
| + state_ = settings_->source_is_pdf ? |
| + kWaitingToSendPageNumbers : kWaitingToSendLayoutSettings; |
| + failed_setting_ = "Save as PDF"; |
| + } else if (state_ == kWaitingToSendLayoutSettings) { |
| + script_argument.SetBoolean("layoutSettings.portrait", |
| + settings_->is_portrait); |
| + state_ = kWaitingToSendPageNumbers; |
| + failed_setting_ = "Layout Settings"; |
| + } else if (state_ == kWaitingToSendPageNumbers) { |
| + script_argument.SetString("pageRange", settings_->page_numbers); |
| + state_ = settings_->source_is_pdf ? |
| + kWaitingForFinalMessage : kWaitingToSendHeadersAndFooters; |
| + failed_setting_ = "Page Range"; |
| + } else if (state_ == kWaitingToSendHeadersAndFooters) { |
| + script_argument.SetBoolean("headersAndFooters", |
| + settings_->headers_and_footers); |
| + state_ = kWaitingToSendBackgroundColorsAndImages; |
| + failed_setting_ = "Headers and Footers"; |
| + } else if (state_ == kWaitingToSendBackgroundColorsAndImages) { |
| + script_argument.SetBoolean("backgroundColorsAndImages", |
| + settings_->background_colors_and_images); |
| + state_ = kWaitingToSendMargins; |
| + failed_setting_ = "Background Colors and Images"; |
| + } else if (state_ == kWaitingToSendMargins) { |
| + script_argument.SetInteger("margins", settings_->margins); |
| + state_ = kWaitingForFinalMessage; |
| + failed_setting_ = "Margins"; |
| + } else if (state_ == kWaitingForFinalMessage) { |
| + EndLoop(); |
| + return; |
| + } |
| + |
| + ASSERT_FALSE(script_argument.empty()); |
| + GetUI()->web_ui()->CallJavascriptFunction( |
| + "onManipulateSettingsForTest", script_argument); |
| + } |
| + |
| + // Saves the print preview settings to be sent to the print preview dialog. |
| + void SetPrintPreviewSettings(const PrintPreviewSettings& settings) { |
| + settings_.reset(new PrintPreviewSettings(settings)); |
| + } |
| + |
| + // Returns the setting that could not be set in the preview dialog. |
| + const std::string& GetFailedSetting() const { |
| + return failed_setting_; |
| + } |
| + |
| + private: |
| + // Listens for messages from the print preview dialog. Specifically, it |
| + // listens for 'UILoadedForTest' and 'UIFailedLoadingForTest.' |
| + class UIDoneLoadingMessageHandler : public content::WebUIMessageHandler { |
| + public: |
| + explicit UIDoneLoadingMessageHandler(PrintPreviewObserver* observer) |
| + : observer_(observer) {} |
| + |
| + virtual ~UIDoneLoadingMessageHandler() {} |
| + |
| + // When a setting has been set succesfully, this is called and the observer |
| + // is told to send the next setting to be set. |
| + void HandleDone(const base::ListValue* /* args */) { |
| + observer_->ManipulatePreviewSettings(); |
| + } |
| + |
| + // Ends the test because a setting was not set successfully. Called when |
| + // this class hears 'UIFailedLoadingForTest.' |
| + void HandleFailure(const base::ListValue* /* args */) { |
| + FAIL() << "Failed to set: " << observer_->GetFailedSetting(); |
| + } |
| + |
| + // Allows this class to listen for the 'UILoadedForTest' and |
| + // 'UIFailedLoadingForTest' messages. These messages are sent by the print |
| + // preview dialog. 'UILoadedForTest' is sent when a setting has been |
| + // successfully set and its effects have been finalized. |
| + // 'UIFailedLoadingForTest' is sent when the setting could not be set. This |
| + // causes the browser test to fail. |
| + void RegisterMessages() OVERRIDE { |
| + // |observer_| is being checked here because it can't be done in the |
| + // constructor and all the the other functions get called many times, |
| + // causing unnecessary checks. |
| + ASSERT_TRUE(observer_); |
|
Lei Zhang
2014/07/18 02:11:01
This is a weird place to put it. I'd just omit the
ivandavid
2014/07/18 02:30:26
Done.
|
| + web_ui()->RegisterMessageCallback( |
| + "UILoadedForTest", |
| + base::Bind(&UIDoneLoadingMessageHandler::HandleDone, |
| + base::Unretained(this))); |
| + |
| + web_ui()->RegisterMessageCallback( |
| + "UIFailedLoadingForTest", |
| + base::Bind(&UIDoneLoadingMessageHandler::HandleFailure, |
| + base::Unretained(this))); |
| + } |
| + |
| + private: |
| + PrintPreviewObserver* const observer_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(UIDoneLoadingMessageHandler); |
| + }; |
| + |
| + // Called when the observer gets the IPC message stating that the page count |
| + // is ready. |
| + void OnDidGetPreviewPageCount( |
| + const PrintHostMsg_DidGetPreviewPageCount_Params ¶ms) { |
| + WebContents* web_contents = GetDialog(); |
| + ASSERT_TRUE(web_contents); |
| + Observe(web_contents); |
| + |
| + PrintPreviewUI* ui = GetUI(); |
| + ASSERT_TRUE(ui); |
| + ASSERT_TRUE(ui->web_ui()); |
| + |
| + // The |ui->web_ui()| owns the message handler. |
| + ui->web_ui()->AddMessageHandler(new UIDoneLoadingMessageHandler(this)); |
| + ui->web_ui()->CallJavascriptFunction("onEnableManipulateSettingsForTest"); |
| + } |
| + |
| + void DidCloneToNewWebContents(WebContents* old_web_contents, |
| + WebContents* new_web_contents) OVERRIDE { |
| + Observe(new_web_contents); |
| + } |
| + |
| + void WebContentsDestroyed() OVERRIDE { |
| + EndLoop(); |
| + } |
| + |
| + Browser* browser_; |
| + base::Closure quit_closure_; |
| + scoped_ptr<PrintPreviewSettings> settings_; |
| + |
| + // State of the observer. The state indicates what message to send |
| + // next. The state advances whenever the message handler calls |
| + // ManipulatePreviewSettings() on the observer. |
| + State state_; |
| + std::string failed_setting_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(PrintPreviewObserver); |
| +}; |
| + |
| +class PrintPreviewPdfGeneratedBrowserTest : public InProcessBrowserTest { |
| + public: |
| + PrintPreviewPdfGeneratedBrowserTest() {} |
| + virtual ~PrintPreviewPdfGeneratedBrowserTest() {} |
| + |
| + // Navigates to the given web page, then initiates print preview and waits |
| + // for all the settings to be set. |
| + void NavigateAndPreview(const base::FilePath::StringType& file_name, |
| + const PrintPreviewSettings& settings) { |
| + print_preview_observer_->SetPrintPreviewSettings(settings); |
| + base::FilePath path(file_name); |
| + GURL gurl = net::FilePathToFileURL(path); |
| + |
| + ui_test_utils::NavigateToURL(browser(), gurl); |
| + |
| + base::RunLoop loop; |
| + print_preview_observer_->set_quit_closure(loop.QuitClosure()); |
| + chrome::Print(browser()); |
| + loop.Run(); |
| + } |
| + |
| + // Prints the web page to a PDF. NavigateAndPreview must be called first. |
| + void Print() { |
| + ASSERT_FALSE(pdf_file_save_path_.empty()); |
| + |
| + base::RunLoop loop; |
| + print_preview_observer_->set_quit_closure(loop.QuitClosure()); |
| + print_preview_observer_->GetUI()->SetSelectedFileForTesting( |
| + pdf_file_save_path_); |
| + loop.Run(); |
| + |
| + // Checks to see if the file exists and is readable. If the file doesn't |
| + // exist, the test will fail. If it exists, but isn't readable, the test |
| + // will keep polling until the file becomes readable. This is due to a |
| + // problem on Windows where the file exists, but isn't readable, causing |
| + // other ASSERT statements in this test to fail. |
| + // TODO(ivandavid): Come up with a better way to do this. |
| + ASSERT_TRUE(base::PathExists(pdf_file_save_path_)); |
| + while (true) { |
| + base::File pdf_file( |
| + pdf_file_save_path_, base::File::FLAG_OPEN | base::File::FLAG_READ); |
| + if (pdf_file.IsValid()) |
| + break; |
| + } |
| + } |
| + |
| + // Initializes function pointers from the PDF library. Called once when the |
| + // test starts. The library is closed when the browser test ends. |
| + void InitPdfFunctions() { |
| + base::FilePath pdf_module_path; |
| + |
| + ASSERT_TRUE(PathService::Get(chrome::FILE_PDF_PLUGIN, &pdf_module_path)); |
| + ASSERT_TRUE(base::PathExists(pdf_module_path)); |
| + pdf_lib_.Reset(base::LoadNativeLibrary(pdf_module_path, NULL)); |
| + |
| + ASSERT_TRUE(pdf_lib_.is_valid()); |
| + pdf_to_bitmap_func_ = |
| + reinterpret_cast<PDFPageToBitmapProc>( |
| + pdf_lib_.GetFunctionPointer("RenderPDFPageToBitmap")); |
| + |
| + pdf_doc_info_func_ = |
| + reinterpret_cast<GetPDFDocInfoProc>( |
| + pdf_lib_.GetFunctionPointer("GetPDFDocInfo")); |
| + |
| + pdf_page_size_func_ = |
| + reinterpret_cast<GetPDFPageSizeByIndexProc>( |
| + pdf_lib_.GetFunctionPointer("GetPDFPageSizeByIndex")); |
| + |
| + ASSERT_TRUE(pdf_to_bitmap_func_); |
| + ASSERT_TRUE(pdf_doc_info_func_); |
| + ASSERT_TRUE(pdf_page_size_func_); |
| + } |
| + |
| + // Converts the PDF to a PNG file so that the layout test can do an image |
| + // diff on this image and a reference image. |
| + void PdfToPng() { |
| + int num_pages; |
| + double max_width_in_points = 0; |
| + std::vector<uint8_t> bitmap_data; |
| + double total_height_in_pixels = 0; |
| + std::string pdf_data; |
| + |
| + ASSERT_TRUE(base::ReadFileToString(pdf_file_save_path_, &pdf_data)); |
| + ASSERT_TRUE(pdf_doc_info_func_(pdf_data.data(), |
| + pdf_data.size(), |
| + &num_pages, |
| + &max_width_in_points)); |
| + |
| + ASSERT_GT(num_pages, 0); |
| + double max_width_in_pixels = |
| + ConvertPointsToPixelDouble(max_width_in_points); |
| + |
| + for (int i = 0; i < num_pages; ++i) { |
| + double width_in_points, height_in_points; |
| + ASSERT_TRUE(pdf_page_size_func_(pdf_data.data(), |
| + pdf_data.size(), |
| + i, |
| + &width_in_points, |
| + &height_in_points)); |
| + |
| + double width_in_pixels = ConvertPointsToPixelDouble(width_in_points); |
| + double height_in_pixels = ConvertPointsToPixelDouble(height_in_points); |
| + |
| + // The image will be rotated if |width_in_pixels| is greater than |
| + // |height_in_pixels|. This is because the page will be rotated to fit |
| + // within a piece of paper. Therefore, |width_in_pixels| and |
| + // |height_in_pixels| have to be swapped or else they won't reflect the |
| + // dimensions of the rotated page. |
| + if (width_in_pixels > height_in_pixels) |
| + std::swap(width_in_pixels, height_in_pixels); |
| + |
| + total_height_in_pixels += height_in_pixels; |
| + gfx::Rect rect(width_in_pixels, height_in_pixels); |
| + PdfRenderSettings settings(rect, kDpi, true); |
| + |
| + int int_max = std::numeric_limits<int>::max(); |
| + if (settings.area().width() > int_max / kColorChannels || |
| + settings.area().height() > int_max / (kColorChannels * |
| + settings.area().width())) { |
| + FAIL() << "The dimensions of the image are too large." |
| + << "Decrease the DPI or the dimensions of the image."; |
| + } |
| + |
| + std::vector<uint8_t> page_bitmap_data( |
| + kColorChannels * settings.area().size().GetArea()); |
| + |
| + ASSERT_TRUE(pdf_to_bitmap_func_(pdf_data.data(), |
| + pdf_data.size(), |
| + i, |
| + page_bitmap_data.data(), |
| + settings.area().size().width(), |
| + settings.area().size().height(), |
| + settings.dpi(), |
| + settings.dpi(), |
| + true)); |
| + FillPng(&page_bitmap_data, |
| + width_in_pixels, |
| + max_width_in_pixels, |
| + settings.area().size().height()); |
| + bitmap_data.insert(bitmap_data.end(), |
| + page_bitmap_data.begin(), |
| + page_bitmap_data.end()); |
| + } |
| + |
| + CreatePng(bitmap_data, max_width_in_pixels, total_height_in_pixels); |
| + } |
| + |
| + // Fills out a bitmap with whitespace so that the image will correctly fit |
| + // within a PNG that is wider than the bitmap itself. |
| + void FillPng(std::vector<uint8_t>* bitmap, |
| + int current_width, |
| + int desired_width, |
| + int height) { |
| + ASSERT_TRUE(bitmap); |
| + ASSERT_GT(height, 0); |
| + ASSERT_LE(current_width, desired_width); |
| + |
| + if (current_width == desired_width) |
| + return; |
| + |
| + int current_width_in_bytes = current_width * kColorChannels; |
| + int desired_width_in_bytes = desired_width * kColorChannels; |
| + const uint8_t kColorByte = 255; |
|
Lei Zhang
2014/07/18 02:11:01
Can you add a comment to mention this works becaus
ivandavid
2014/07/18 02:30:26
Done.
|
| + std::vector<uint8_t> filled_bitmap( |
| + desired_width * kColorChannels * height, kColorByte); |
| + std::vector<uint8_t>::iterator filled_bitmap_it = filled_bitmap.begin(); |
| + std::vector<uint8_t>::iterator bitmap_it = bitmap->begin(); |
| + |
| + for (int i = 0; i < height; ++i) { |
| + std::copy( |
| + bitmap_it, bitmap_it + current_width_in_bytes, filled_bitmap_it); |
| + std::advance(bitmap_it, current_width_in_bytes); |
| + std::advance(filled_bitmap_it, desired_width_in_bytes); |
| + } |
| + |
| + bitmap->assign(filled_bitmap.begin(), filled_bitmap.end()); |
| + } |
| + |
| + // Sends the PNG image to the layout test framework for comparison. |
| + void SendPng() { |
| + // Send image header and |hash_| to the layout test framework. |
| + std::cout << "Content-Type: image/png\n"; |
| + std::cout << "ActualHash: " << base::MD5DigestToBase16(hash_) << "\n"; |
| + std::cout << "Content-Length: " << png_output_.size() << "\n"; |
| + |
| + std::copy(png_output_.begin(), |
| + png_output_.end(), |
| + std::ostream_iterator<unsigned char>(std::cout, "")); |
| + |
| + std::cout << "#EOF\n"; |
| + std::cout.flush(); |
| + std::cerr << "#EOF\n"; |
| + std::cerr.flush(); |
| + } |
| + |
| + // Duplicates the tab that was created when the browser opened. This is done |
| + // so that the observer can listen to the duplicated tab as soon as possible |
| + // and start listening for messages related to print preview. |
| + void DuplicateTab() { |
| + WebContents* tab = |
| + browser()->tab_strip_model()->GetActiveWebContents(); |
| + ASSERT_TRUE(tab); |
| + |
| + print_preview_observer_.reset(new PrintPreviewObserver(browser(), tab)); |
| + chrome::DuplicateTab(browser()); |
| + |
| + WebContents* initiator = |
| + browser()->tab_strip_model()->GetActiveWebContents(); |
| + ASSERT_TRUE(initiator); |
| + ASSERT_NE(tab, initiator); |
| + } |
| + |
| + // Resets the test so that another web page can be printed. It also deletes |
| + // the duplicated tab as it isn't needed anymore. |
| + void Reset() { |
| + png_output_.clear(); |
| + ASSERT_EQ(2, browser()->tab_strip_model()->count()); |
| + chrome::CloseTab(browser()); |
| + ASSERT_EQ(1, browser()->tab_strip_model()->count()); |
| + } |
| + |
| + // Creates a temporary directory to store a text file that will be used for |
| + // stdin to accept input from the layout test framework. A path for the PDF |
| + // file is also created. The directory and files within it are automatically |
| + // cleaned up once the test ends. |
| + void SetupStdinAndSavePath() { |
| + // Sets the filemode to binary because it will force |std::cout| to send LF |
| + // rather than CRLF. Sending CRLF will cause an error message for the |
| + // layout tests. |
| +#if defined(OS_WIN) |
| + _setmode(_fileno(stdout), _O_BINARY); |
| + _setmode(_fileno(stderr), _O_BINARY); |
| +#endif |
| + // Sends a message to the layout test framework indicating indicating |
| + // that the browser test has completed setting itself up. The layout |
| + // test will then expect the file path for stdin. |
| + base::FilePath stdin_path; |
| + std::cout << "#READY\n"; |
| + std::cout.flush(); |
| + |
| + ASSERT_TRUE(tmp_dir_.CreateUniqueTempDir()); |
| + ASSERT_TRUE(base::CreateTemporaryFileInDir(tmp_dir_.path(), &stdin_path)); |
| + |
| + // Redirects |std::cin| to the file |stdin_path|. |in| is not freed because |
| + // if it goes out of scope, |std::cin.rdbuf| will be freed. Causing an |
| + // error. |
| + std::ifstream* in = new std::ifstream(stdin_path.value().c_str()); |
| + ASSERT_TRUE(in->is_open()); |
| + std::cin.rdbuf(in->rdbuf()); |
| + |
| + pdf_file_save_path_ = |
| + tmp_dir_.path().Append(FILE_PATH_LITERAL("dummy.pdf")); |
| + |
| + // Send the file path to the layout test framework so that it can |
| + // communicate with this browser test. |
| + std::cout << "StdinPath: " << stdin_path.value() << "\n"; |
| + std::cout << "#EOF\n"; |
| + std::cout.flush(); |
| + } |
| + |
| + private: |
| + // Generates a png from bitmap data and stores it in |png_output_|. |
| + void CreatePng(const std::vector<uint8_t>& bitmap_data, |
| + int width, |
| + int height) { |
| + base::MD5Sum(static_cast<const void*>(bitmap_data.data()), |
| + bitmap_data.size(), |
| + &hash_); |
| + gfx::Rect png_rect(width, height); |
| + |
| + // tEXtchecksum looks funny, but that's what the layout test framework |
| + // expects. |
| + std::string comment_title("tEXtchecksum\x00"); |
| + gfx::PNGCodec::Comment hash_comment(comment_title, |
| + base::MD5DigestToBase16(hash_)); |
| + std::vector<gfx::PNGCodec::Comment> comments; |
| + comments.push_back(hash_comment); |
| + ASSERT_TRUE(gfx::PNGCodec::Encode(bitmap_data.data(), |
| + gfx::PNGCodec::FORMAT_BGRA, |
| + png_rect.size(), |
| + png_rect.size().width() * kColorChannels, |
| + false, |
| + comments, |
| + &png_output_)); |
| + } |
| + |
| + scoped_ptr<PrintPreviewObserver> print_preview_observer_; |
| + base::FilePath pdf_file_save_path_; |
| + |
| + // These typedefs are function pointers to pdflib functions that give |
| + // information about the PDF as a whole and about specific pages. |
| + |
| + // Converts the PDF to a bitmap. |
| + typedef bool (*PDFPageToBitmapProc)(const void* pdf_buffer, |
| + int pdf_buffer_size, |
| + int page_number, |
| + void* bitmap_buffer, |
| + int bitmap_width, |
| + int bitmap_height, |
| + int dpi_x, |
| + int dpi_y, |
| + bool autorotate); |
| + |
| + // Gets the page count and maximum page width of the PDF in points. |
| + typedef bool (*GetPDFDocInfoProc)(const void* pdf_buffer, |
| + int buffer_size, |
| + int* pages_count, |
| + double* max_page_width); |
| + |
| + // Gets the dimensions of a specific page within a PDF. |
| + typedef bool (*GetPDFPageSizeByIndexProc)(const void* pdf_buffer, |
| + int buffer_size, |
| + int index, |
| + double* width, |
| + double* height); |
| + |
| + // Instantiations of the function pointers described above. |
| + PDFPageToBitmapProc pdf_to_bitmap_func_; |
| + GetPDFDocInfoProc pdf_doc_info_func_; |
| + GetPDFPageSizeByIndexProc pdf_page_size_func_; |
| + |
| + // Used to open up the pdf plugin, which contains the functions above. |
| + base::ScopedNativeLibrary pdf_lib_; |
| + |
| + // Vector for storing the PNG to be sent to the layout test framework. |
| + // TODO(ivandavid): Eventually change this to uint32_t and make everything |
| + // work with that. It might be a bit tricky to fix everything to work with |
| + // uint32_t, but not too tricky. |
| + std::vector<unsigned char> png_output_; |
| + |
| + // Image hash of the bitmap that is turned into a PNG. The hash is put into |
| + // the PNG as a comment, as it is needed by the layout test framework. |
| + base::MD5Digest hash_; |
| + |
| + // Temporary directory for storing the pdf and the file for stdin. |
| + base::ScopedTempDir tmp_dir_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(PrintPreviewPdfGeneratedBrowserTest); |
| +}; |
| + |
| +// This test acts as a driver for the layout test framework. |
| +IN_PROC_BROWSER_TEST_F(PrintPreviewPdfGeneratedBrowserTest, |
| + MANUAL_LayoutTestDriver) { |
| + // What this code is supposed to do: |
| + // - Setup communication with the layout test framework |
| + // - Print webpage to a pdf |
| + // - Convert pdf to a png |
| + // - Send png to layout test framework, where it doesn an image diff |
| + // on the image sent by this test and a reference image. |
| + // |
| + // Throughout this code, there will be |std::cout| statements. The layout test |
| + // framework uses stdout to get data from the browser test and uses stdin |
| + // to send data to the browser test. Writing "EOF\n" to |std::cout| indicates |
| + // that whatever block of data that the test was expecting has been completely |
| + // sent. Sometimes EOF is printed to stderr because the test will expect it |
| + // from stderr in addition to stdout for certain blocks of data. |
| + InitPdfFunctions(); |
| + SetupStdinAndSavePath(); |
| + |
| + // There is no way to determine how many tests are to be run ahead of time |
| + // without undesirable changes to the layout test framework. However, that is |
| + // ok since whenever all the tests have been run, the layout test framework |
| + // calls SIGKILL on this process, ending the test. This will end the while |
| + // loop and cause the test to clean up after itself. For this to work, the |
| + // browsertest must be run with '--single_process' not '--single-process.' |
| + while (true) { |
| + std::string input; |
| + std::getline(std::cin, input); |
| + if (input.empty()) { |
| + while (std::cin.eof()) { |
| + std::cin.clear(); |
| + std::getline(std::cin, input); |
| + if (!input.empty()) { |
| + break; |
| + } |
| + } |
| + } |
| + |
| + base::FilePath::StringType file_extension = FILE_PATH_LITERAL(".pdf"); |
| + base::FilePath::StringType cmd; |
| +#if defined(OS_POSIX) |
| + cmd = input; |
| +#elif defined(OS_WIN) |
| + cmd = base::UTF8ToWide(input); |
| +#endif |
| + |
| + DuplicateTab(); |
| + PrintPreviewSettings settings( |
| + true, |
| + "", |
| + false, |
| + false, |
| + DEFAULT_MARGINS, |
| + cmd.find(file_extension) != base::FilePath::StringType::npos); |
| + |
| + // Splits the command sent by the layout test framework. The first command |
| + // is always the file path to use for the test. The rest isn't relevant, |
| + // so it can be ignored. The separator for the commands is an apostrophe. |
| + std::vector<base::FilePath::StringType> cmd_arguments; |
| + base::SplitString(cmd, '\'', &cmd_arguments); |
| + |
| + ASSERT_GE(cmd_arguments.size(), 1U); |
| + base::FilePath::StringType test_name(cmd_arguments[0]); |
| + NavigateAndPreview(test_name, settings); |
| + Print(); |
| + PdfToPng(); |
| + |
| + // Message to the layout test framework indicating that it should start |
| + // waiting for the image data, as there is no more text data to be read. |
| + // There actually isn't any text data at all, however because the layout |
| + // test framework requires it, a message has to be sent to stop it from |
| + // waiting for this message and start waiting for the image data. |
| + std::cout << "#EOF\n"; |
| + std::cout.flush(); |
| + |
| + SendPng(); |
| + Reset(); |
| + } |
| +} |
| + |
| +} // namespace printing |