Index: printing/printing_context_win_unittest.cc |
diff --git a/printing/printing_context_win_unittest.cc b/printing/printing_context_win_unittest.cc |
index 5917b65c928e4d64761b4aaa0b5e357209f1971d..2251e64a59572eb5cebcfd8e496892203eae867e 100644 |
--- a/printing/printing_context_win_unittest.cc |
+++ b/printing/printing_context_win_unittest.cc |
@@ -10,6 +10,7 @@ |
#include "base/bind.h" |
#include "base/bind_helpers.h" |
#include "base/memory/scoped_ptr.h" |
+#include "printing/backend/printing_info_win.h" |
#include "printing/printing_test.h" |
#include "printing/printing_context.h" |
#include "printing/printing_context_win.h" |
@@ -50,18 +51,15 @@ HRESULT WINAPI PrintDlgExMock(LPPRINTDLGEX lppd) { |
return E_FAIL; |
scoped_array<uint8> buffer; |
- DEVMODE* dev_mode = NULL; |
- PRINTER_INFO_2* info_2 = NULL; |
+ const DEVMODE* dev_mode = NULL; |
HRESULT result = S_OK; |
lppd->hDC = NULL; |
lppd->hDevMode = NULL; |
lppd->hDevNames = NULL; |
- printing::PrintingContextWin::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; |
+ printing::PrinterInfo2 info_2; |
+ if (info_2.Init(printer)) { |
+ dev_mode = info_2.get()->pDevMode; |
} |
if (!dev_mode) { |
result = E_FAIL; |
@@ -89,11 +87,12 @@ HRESULT WINAPI PrintDlgExMock(LPPRINTDLGEX lppd) { |
GlobalUnlock(lppd->hDevMode); |
dev_mode_ptr = NULL; |
- 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 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; |
+ port_size; |
lppd->hDevNames = GlobalAlloc(GHND, dev_names_size); |
if (!lppd->hDevNames) { |
result = E_FAIL; |
@@ -108,13 +107,13 @@ HRESULT WINAPI PrintDlgExMock(LPPRINTDLGEX lppd) { |
dev_names->wDefault = 1; |
dev_names->wDriverOffset = sizeof(DEVNAMES); |
memcpy(reinterpret_cast<uint8*>(dev_names_ptr) + dev_names->wDriverOffset, |
- info_2->pDriverName, driver_size); |
+ info_2.get()->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); |
+ info_2.get()->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); |
+ info_2.get()->pPortName, port_size); |
GlobalUnlock(lppd->hDevNames); |
dev_names_ptr = NULL; |