Add basic tests for PDFiumPrintTest::PrintPagesAsPdf().

Change-Id: I6b5262ccf903b69dbf3e0a5c937e38a0c9b2e886
Reviewed-on: https://chromium-review.googlesource.com/1244879Reviewed-by: Rebekah Potter <rbpotter@chromium.org>
Commit-Queue: Lei Zhang <thestig@chromium.org>
Cr-Commit-Position: refs/heads/master@{#594829}

pdf/pdfium/pdfium_print_unittest.cc

@@ -5,29 +5,133 @@
 #include "pdf/pdfium/pdfium_print.h"
 
 #include "base/stl_util.h"
+#include "pdf/pdfium/pdfium_engine.h"
+#include "pdf/pdfium/pdfium_engine_exports.h"
+#include "pdf/pdfium/pdfium_test_base.h"
+#include "pdf/test/test_client.h"
 #include "ppapi/c/dev/ppp_printing_dev.h"
+#include "ppapi/c/private/ppp_pdf.h"
 #include "testing/gmock/include/gmock/gmock.h"
-#include "testing/gtest/include/gtest/gtest.h"
 
 namespace chrome_pdf {
 
+using PDFiumPrintTest = PDFiumTestBase;
 using testing::ElementsAre;
 
-TEST(PDFiumPrintTest, GetPageNumbersFromPrintPageNumberRange) {
+namespace {
+
+constexpr PP_Size kUSLetterSize = {612, 792};
+constexpr PP_Rect kUSLetterRect = {{0, 0}, kUSLetterSize};
+
+struct SizeDouble {
+  double width;
+  double height;
+};
+
+using ExpectedDimensions = std::vector<SizeDouble>;
+
+void CheckPdfDimensions(const std::vector<uint8_t>& pdf_data,
+                        const ExpectedDimensions& expected_dimensions) {
+  PDFiumEngineExports exports;
+  int page_count;
+  ASSERT_TRUE(exports.GetPDFDocInfo(pdf_data, &page_count, nullptr));
+  ASSERT_GT(page_count, 0);
+  ASSERT_EQ(expected_dimensions.size(), static_cast<size_t>(page_count));
+
+  for (int i = 0; i < page_count; ++i) {
+    double width;
+    double height;
+    ASSERT_TRUE(exports.GetPDFPageSizeByIndex(pdf_data, i, &width, &height));
+    EXPECT_DOUBLE_EQ(expected_dimensions[i].width, width);
+    EXPECT_DOUBLE_EQ(expected_dimensions[i].height, height);
+  }
+}
+
+}  // namespace
+
+TEST_F(PDFiumPrintTest, GetPageNumbersFromPrintPageNumberRange) {
   std::vector<uint32_t> page_numbers;
 
   {
-    PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 2}};
+    const PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 2}};
     page_numbers = PDFiumPrint::GetPageNumbersFromPrintPageNumberRange(
         &page_ranges[0], base::size(page_ranges));
     EXPECT_THAT(page_numbers, ElementsAre(0, 1, 2));
   }
   {
-    PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 0}, {2, 2}, {4, 5}};
+    const PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 0}, {2, 2}, {4, 5}};
     page_numbers = PDFiumPrint::GetPageNumbersFromPrintPageNumberRange(
         &page_ranges[0], base::size(page_ranges));
     EXPECT_THAT(page_numbers, ElementsAre(0, 2, 4, 5));
   }
 }
 
+TEST_F(PDFiumPrintTest, Basic) {
+  SetDocumentForTest(FILE_PATH_LITERAL("hello_world2.pdf"));
+  pp::URLLoader dummy_loader;
+  TestClient client;
+  PDFiumEngine engine(&client, true);
+  ASSERT_TRUE(engine.New("https://chromium.org/dummy.pdf", ""));
+  ASSERT_TRUE(engine.HandleDocumentLoad(dummy_loader));
+
+  PDFiumPrint print(&engine);
+
+  const PP_PrintSettings_Dev print_settings = {kUSLetterRect,
+                                               kUSLetterRect,
+                                               kUSLetterSize,
+                                               72,
+                                               PP_PRINTORIENTATION_NORMAL,
+                                               PP_PRINTSCALINGOPTION_NONE,
+                                               PP_FALSE,
+                                               PP_PRINTOUTPUTFORMAT_PDF};
+  const PP_PdfPrintSettings_Dev pdf_print_settings = {1, 100};
+
+  {
+    // Print 2 pages.
+    const ExpectedDimensions kExpectedDimensions = {{612.0, 792.0},
+                                                    {612.0, 792.0}};
+    const PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 1}};
+    std::vector<uint8_t> pdf_data =
+        print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                              print_settings, pdf_print_settings,
+                              /*raster=*/false);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+
+    pdf_data = print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                                     print_settings, pdf_print_settings,
+                                     /*raster=*/true);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+  }
+  {
+    // Print 1 page.
+    const ExpectedDimensions kExpectedDimensions = {{612.0, 792.0}};
+    const PP_PrintPageNumberRange_Dev page_ranges[] = {{0, 0}};
+    std::vector<uint8_t> pdf_data =
+        print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                              print_settings, pdf_print_settings,
+                              /*raster=*/false);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+
+    pdf_data = print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                                     print_settings, pdf_print_settings,
+                                     /*raster=*/true);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+  }
+  {
+    // Print the other page.
+    const ExpectedDimensions kExpectedDimensions = {{612.0, 792.0}};
+    const PP_PrintPageNumberRange_Dev page_ranges[] = {{1, 1}};
+    std::vector<uint8_t> pdf_data =
+        print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                              print_settings, pdf_print_settings,
+                              /*raster=*/false);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+
+    pdf_data = print.PrintPagesAsPdf(&page_ranges[0], base::size(page_ranges),
+                                     print_settings, pdf_print_settings,
+                                     /*raster=*/true);
+    CheckPdfDimensions(pdf_data, kExpectedDimensions);
+  }
+}
+
 }  // namespace chrome_pdf

pdf/test/test_client.cc

@@ -27,6 +27,13 @@ pp::URLLoader TestClient::CreateURLLoader() {
   return pp::URLLoader();
 }
 
+std::vector<PDFEngine::Client::SearchStringResult> TestClient::SearchString(
+    const base::char16* string,
+    const base::char16* term,
+    bool case_sensitive) {
+  return std::vector<SearchStringResult>();
+}
+
 pp::Instance* TestClient::GetPluginInstance() {
   return nullptr;
 }

pdf/test/test_client.h

@@ -6,6 +6,7 @@
 #define PDF_TEST_TEST_CLIENT_H_
 
 #include <string>
+#include <vector>
 
 #include "pdf/pdf_engine.h"
 
@@ -22,6 +23,9 @@ class TestClient : public PDFEngine::Client {
                      const std::string& default_answer) override;
   std::string GetURL() override;
   pp::URLLoader CreateURLLoader() override;
+  std::vector<SearchStringResult> SearchString(const base::char16* string,
+                                               const base::char16* term,
+                                               bool case_sensitive) override;
   pp::Instance* GetPluginInstance() override;
   bool IsPrintPreview() override;
   uint32_t GetBackgroundColor() override;