| 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;
|
|
|
|
|