Win32: Print the right number of pages.

BUG=48392
TEST=included.
Review URL: http://codereview.chromium.org/3113012

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@56063 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 85 insertions, 1 deletions

diff --git a/printing/printing_context_win_unittest.cc b/printing/printing_context_win_unittest.cc
index 9e19e3b..c2e377b 100644
--- a/printing/printing_context_win_unittest.cc
+++ b/printing/printing_context_win_unittest.cc
@@ -12,13 +12,88 @@
 class PrintingContextTest : public PrintingTest<testing::Test> {
 };
 
+// This is a fake PrintDlgEx implementation that sets the right fields in
+// |lppd| to trigger a bug in older revisions of PrintingContext.
+HRESULT WINAPI PrintDlgExMock(LPPRINTDLGEX lppd) {
+  // The interesting bits:
+  // Pretend the user hit print
+  lppd->dwResultAction = PD_RESULT_PRINT;
+
+  // Pretend the page range is 1-5, but since lppd->Flags does not have
+  // PD_SELECTION set, this really shouldn't matter.
+  lppd->nPageRanges = 1;
+  lppd->lpPageRanges = new PRINTPAGERANGE[1];
+  lppd->lpPageRanges[0].nFromPage = 1;
+  lppd->lpPageRanges[0].nToPage = 5;
+
+  // Painful paperwork.
+  std::wstring printer_name = PrintingContextTest::GetDefaultPrinter();
+  HANDLE printer;
+  if (!OpenPrinter(const_cast<wchar_t*>(printer_name.c_str()), &printer, NULL))
+    return E_FAIL;
+
+  scoped_array<uint8> buffer;
+  DEVMODE* dev_mode = NULL;
+  PRINTER_INFO_2* info_2 = NULL;
+
+  printing::PrintingContext::GetPrinterHelper(printer, 2, &buffer);
+  if (buffer.get()) {
+    info_2 = reinterpret_cast<PRINTER_INFO_2*>(buffer.get());
+    if (info_2->pDevMode != NULL)
+      dev_mode = info_2->pDevMode;
+  }
+  if (!dev_mode)
+    return E_FAIL;
+
+  if (!printing::PrintingContext::AllocateContext(printer_name, dev_mode,
+      &lppd->hDC)) {
+    return E_FAIL;
+  }
+
+  size_t dev_mode_size = dev_mode->dmSize + dev_mode->dmDriverExtra;
+  lppd->hDevMode = GlobalAlloc(GMEM_MOVEABLE, dev_mode_size);
+  if (!lppd->hDevMode)
+    return E_FAIL;
+  void* dev_mode_ptr = GlobalLock(lppd->hDevMode);
+  if (!dev_mode_ptr)
+    return E_FAIL;
+  memcpy(dev_mode_ptr, dev_mode, dev_mode_size);
+  GlobalUnlock(lppd->hDevMode);
+
+  size_t driver_size = 2 + sizeof(wchar_t) * lstrlen(info_2->pDriverName);
+  size_t printer_size = 2 + sizeof(wchar_t) * lstrlen(info_2->pPrinterName);
+  size_t port_size = 2 + sizeof(wchar_t) * lstrlen(info_2->pPortName);
+  size_t dev_names_size = sizeof(DEVNAMES) + driver_size + printer_size +
+      port_size;
+  lppd->hDevNames = GlobalAlloc(GHND, dev_names_size);
+  if (!lppd->hDevNames)
+    return E_FAIL;
+  void* dev_names_ptr = GlobalLock(lppd->hDevNames);
+  if (!dev_names_ptr)
+    return E_FAIL;
+  DEVNAMES* dev_names = reinterpret_cast<DEVNAMES*>(dev_names_ptr);
+  dev_names->wDefault = 1;
+  dev_names->wDriverOffset = sizeof(DEVNAMES);
+  memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wDriverOffset,
+      info_2->pDriverName, driver_size);
+  dev_names->wDeviceOffset = dev_names->wDriverOffset + driver_size;
+  memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wDeviceOffset,
+      info_2->pPrinterName, printer_size);
+  dev_names->wOutputOffset = dev_names->wDeviceOffset + printer_size;
+  memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wOutputOffset,
+      info_2->pPortName, port_size);
+  GlobalUnlock(lppd->hDevNames);
+
+  return S_OK;
+}
+
 TEST_F(PrintingContextTest, Base) {
   printing::PrintSettings settings;
 
   settings.set_device_name(GetDefaultPrinter());
   // Initialize it.
   printing::PrintingContext context;
-  EXPECT_EQ(context.InitWithSettings(settings), printing::PrintingContext::OK);
+  EXPECT_EQ(printing::PrintingContext::OK, context.InitWithSettings(settings));
 
   // The print may lie to use and may not support world transformation.
   // Verify right now.
@@ -26,3 +101,12 @@ TEST_F(PrintingContextTest, Base) {
   EXPECT_TRUE(SetWorldTransform(context.context(), &random_matrix));
   EXPECT_TRUE(ModifyWorldTransform(context.context(), NULL, MWT_IDENTITY));
 }
+
+TEST_F(PrintingContextTest, PrintAll) {
+  printing::PrintingContext context;
+  context.SetPrintDialog(&PrintDlgExMock);
+  ASSERT_EQ(printing::PrintingContext::OK,
+      context.AskUserForSettings(NULL, 123, false));
+  printing::PrintSettings settings = context.settings();
+  EXPECT_EQ(settings.ranges.size(), 0);
+}
\ No newline at end of file