| Index: chrome/browser/printing/print_preview_test.cc
|
| diff --git a/chrome/browser/printing/print_preview_test.cc b/chrome/browser/printing/print_preview_test.cc
|
| index 939a89d06e4014c6e4e1fe1ca412113a35fb64b1..00a528a9c6fcc5819a68e2e438d1c8dfeae30b5f 100644
|
| --- a/chrome/browser/printing/print_preview_test.cc
|
| +++ b/chrome/browser/printing/print_preview_test.cc
|
| @@ -4,8 +4,10 @@
|
|
|
| #include "chrome/browser/printing/print_preview_test.h"
|
|
|
| +#include "chrome/browser/printing/print_preview_dialog_controller.h"
|
| #include "chrome/test/base/dialog_test_browser_window.h"
|
| #include "content/public/browser/plugin_service.h"
|
| +#include "ui/views/window/dialog_delegate.h"
|
|
|
| PrintPreviewTest::PrintPreviewTest() {}
|
| PrintPreviewTest::~PrintPreviewTest() {}
|
| @@ -21,3 +23,27 @@ void PrintPreviewTest::SetUp() {
|
| BrowserWindow* PrintPreviewTest::CreateBrowserWindow() {
|
| return new DialogTestBrowserWindow;
|
| }
|
| +
|
| +#if defined(USE_AURA)
|
| +PrintPreviewDialogControllerTest::PrintPreviewDialogControllerTest(
|
| + gfx::NativeView context): context_(context) {}
|
| +
|
| +PrintPreviewDialogControllerTest::~PrintPreviewDialogControllerTest() {}
|
| +
|
| +// Need to override the function to get the delegate so that it uses the
|
| +// correct window.
|
| +ConstrainedWebDialogDelegate* PrintPreviewDialogControllerTest::
|
| + GetWebDialogDelegate(
|
| + content::WebContents* initiator) {
|
| + ConstrainedWebDialogDelegate* dialog =
|
| + ShowTopLevelConstrainedWebDialogInWindow(
|
| + initiator->GetBrowserContext(),
|
| + printing::GetPrintPreviewDialogDelegate(initiator),
|
| + initiator, context_);
|
| + return dialog;
|
| +}
|
| +
|
| +void PrintPreviewDialogControllerTest::SetContext(gfx::NativeView context) {
|
| + context_ = context;
|
| +}
|
| +#endif
|
|
|