summaryrefslogtreecommitdiffstats
path: root/printing/printing_context_win_unittest.cc
diff options
context:
space:
mode:
authorvitalybuka <vitalybuka@chromium.org>2014-09-09 12:53:33 -0700
committerCommit bot <commit-bot@chromium.org>2014-09-09 19:59:05 +0000
commitf9d0c0ca61adcc78669177583c701607d70cdb0f (patch)
tree87512338a2aa0516805c1b942e199cd0fdcf1bcb /printing/printing_context_win_unittest.cc
parent75844fdb8b70aa957fc8bc6bba019c2026fde28f (diff)
downloadchromium_src-f9d0c0ca61adcc78669177583c701607d70cdb0f.zip
chromium_src-f9d0c0ca61adcc78669177583c701607d70cdb0f.tar.gz
chromium_src-f9d0c0ca61adcc78669177583c701607d70cdb0f.tar.bz2
Restored printing windows context for builds without preview.
This is code placed into separate file that was removed in https://codereview.chromium.org/478183005 Code is still used by CEF. BUG=374321 NOTRY=true Review URL: https://codereview.chromium.org/557663002 Cr-Commit-Position: refs/heads/master@{#293992}
Diffstat (limited to 'printing/printing_context_win_unittest.cc')
-rw-r--r--printing/printing_context_win_unittest.cc141
1 files changed, 140 insertions, 1 deletions
diff --git a/printing/printing_context_win_unittest.cc b/printing/printing_context_win_unittest.cc
index 5147249..eb6de5ca 100644
--- a/printing/printing_context_win_unittest.cc
+++ b/printing/printing_context_win_unittest.cc
@@ -4,8 +4,13 @@
#include "printing/printing_context_win.h"
-#include "printing/printing_test.h"
+#include "base/bind.h"
+#include "base/message_loop/message_loop.h"
+#include "printing/backend/printing_info_win.h"
+#include "printing/backend/win_helper.h"
#include "printing/print_settings.h"
+#include "printing/printing_context_system_dialog_win.h"
+#include "printing/printing_test.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace printing {
@@ -29,6 +34,140 @@ class PrintingContextTest : public PrintingTest<testing::Test>,
PrintingContext::Result result_;
};
+class MockPrintingContextWin : public PrintingContextSytemDialogWin {
+ public:
+ MockPrintingContextWin(Delegate* delegate)
+ : PrintingContextSytemDialogWin(delegate) {}
+
+ protected:
+ // This is a fake PrintDlgEx implementation that sets the right fields in
+ // |lppd| to trigger a bug in older revisions of PrintingContext.
+ HRESULT ShowPrintDialog(PRINTDLGEX* lppd) OVERRIDE {
+ // 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[0].nFromPage = 1;
+ lppd->lpPageRanges[0].nToPage = 5;
+
+ base::string16 printer_name = PrintingContextTest::GetDefaultPrinter();
+ ScopedPrinterHandle printer;
+ if (!printer.OpenPrinter(printer_name.c_str()))
+ return E_FAIL;
+
+ scoped_ptr<uint8[]> buffer;
+ const DEVMODE* dev_mode = NULL;
+ HRESULT result = S_OK;
+ lppd->hDC = NULL;
+ lppd->hDevMode = NULL;
+ lppd->hDevNames = NULL;
+
+ PrinterInfo2 info_2;
+ if (info_2.Init(printer)) {
+ dev_mode = info_2.get()->pDevMode;
+ }
+ if (!dev_mode) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+
+ lppd->hDC = CreateDC(L"WINSPOOL", printer_name.c_str(), NULL, dev_mode);
+ if (!lppd->hDC) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+
+ size_t dev_mode_size = dev_mode->dmSize + dev_mode->dmDriverExtra;
+ lppd->hDevMode = GlobalAlloc(GMEM_MOVEABLE, dev_mode_size);
+ if (!lppd->hDevMode) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+ void* dev_mode_ptr = GlobalLock(lppd->hDevMode);
+ if (!dev_mode_ptr) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+ memcpy(dev_mode_ptr, dev_mode, dev_mode_size);
+ GlobalUnlock(lppd->hDevMode);
+ dev_mode_ptr = NULL;
+
+ size_t driver_size =
+ 2 + sizeof(wchar_t) * lstrlen(info_2.get()->pDriverName);
+ size_t printer_size =
+ 2 + sizeof(wchar_t) * lstrlen(info_2.get()->pPrinterName);
+ size_t port_size = 2 + sizeof(wchar_t) * lstrlen(info_2.get()->pPortName);
+ size_t dev_names_size =
+ sizeof(DEVNAMES) + driver_size + printer_size + port_size;
+ lppd->hDevNames = GlobalAlloc(GHND, dev_names_size);
+ if (!lppd->hDevNames) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+ void* dev_names_ptr = GlobalLock(lppd->hDevNames);
+ if (!dev_names_ptr) {
+ result = E_FAIL;
+ goto Cleanup;
+ }
+ DEVNAMES* dev_names = reinterpret_cast<DEVNAMES*>(dev_names_ptr);
+ dev_names->wDefault = 1;
+ dev_names->wDriverOffset = sizeof(DEVNAMES) / sizeof(wchar_t);
+ memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wDriverOffset,
+ info_2.get()->pDriverName,
+ driver_size);
+ dev_names->wDeviceOffset =
+ dev_names->wDriverOffset + driver_size / sizeof(wchar_t);
+ memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wDeviceOffset,
+ info_2.get()->pPrinterName,
+ printer_size);
+ dev_names->wOutputOffset =
+ dev_names->wDeviceOffset + printer_size / sizeof(wchar_t);
+ memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wOutputOffset,
+ info_2.get()->pPortName,
+ port_size);
+ GlobalUnlock(lppd->hDevNames);
+ dev_names_ptr = NULL;
+
+ Cleanup:
+ // Note: This section does proper deallocation/free of DC/global handles. We
+ // did not use ScopedHGlobal or ScopedHandle because they did not
+ // perform what we need. Goto's are used based on Windows programming
+ // idiom, to avoid deeply nested if's, and try-catch-finally is not
+ // allowed in Chromium.
+ if (FAILED(result)) {
+ if (lppd->hDC) {
+ DeleteDC(lppd->hDC);
+ }
+ if (lppd->hDevMode) {
+ GlobalFree(lppd->hDevMode);
+ }
+ if (lppd->hDevNames) {
+ GlobalFree(lppd->hDevNames);
+ }
+ }
+ return result;
+ }
+};
+
+TEST_F(PrintingContextTest, PrintAll) {
+ base::MessageLoop message_loop;
+ if (IsTestCaseDisabled())
+ return;
+
+ MockPrintingContextWin context(this);
+ context.AskUserForSettings(
+ 123,
+ false,
+ base::Bind(&PrintingContextTest::PrintSettingsCallback,
+ base::Unretained(this)));
+ EXPECT_EQ(PrintingContext::OK, result());
+ PrintSettings settings = context.settings();
+ EXPECT_EQ(settings.ranges().size(), 0);
+}
+
TEST_F(PrintingContextTest, Base) {
if (IsTestCaseDisabled())
return;