Created min size for print preview dialog and modified to allow the

chrome/browser/ui/webui/print_preview/print_preview_ui_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stdint.h>
 
 #include <vector>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted_memory.h"
@@ skipping 23 matching lines @@
   std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1));
   return new base::RefCountedBytes(preview_data);
 }
 
 bool IsShowingWebContentsModalDialog(WebContents* tab) {
   WebContentsModalDialogManager* web_contents_modal_dialog_manager =
       WebContentsModalDialogManager::FromWebContents(tab);
   return web_contents_modal_dialog_manager->IsDialogActive();
 }
 
+#if defined(USE_AURA)
+class PrintPreviewDialogControllerTest:
+  public printing::PrintPreviewDialogController {
+
+ public:
+  explicit PrintPreviewDialogControllerTest(gfx::NativeView context):
+    context_(context) {}
+
+  // Need to override the function to get the delegate so that it uses the
+  // correct window.
+  ConstrainedWebDialogDelegate* GetWebDialogDelegate(
+      WebContents* initiator) override {
+    ConstrainedWebDialogDelegate* dialog =
+      ShowTopLevelConstrainedWebDialogInWindow(
+           initiator->GetBrowserContext(),
+           printing::GetPrintPreviewDialogDelegate(initiator),
+           initiator, context_);
+    return dialog;
+  }
+
+ protected:
+  ~PrintPreviewDialogControllerTest() override {}
+
+ private:
+  gfx::NativeWindow context_;
+
+};
+#endif
 }  // namespace
 
 class PrintPreviewUIUnitTest : public PrintPreviewTest {
  public:
   PrintPreviewUIUnitTest();
   ~PrintPreviewUIUnitTest() override;
 
  protected:
   void SetUp() override;
 
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUIUnitTest);
 };
 
 PrintPreviewUIUnitTest::PrintPreviewUIUnitTest() {}
 PrintPreviewUIUnitTest::~PrintPreviewUIUnitTest() {}
 
 void PrintPreviewUIUnitTest::SetUp() {
   PrintPreviewTest::SetUp();
 
   chrome::NewTab(browser());
 }
 
 // Create/Get a preview tab for initiator.
 TEST_F(PrintPreviewUIUnitTest, PrintPreviewData) {
   WebContents* initiator = browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_TRUE(initiator);
   EXPECT_FALSE(IsShowingWebContentsModalDialog(initiator));
 
+#if defined(USE_AURA)
+  printing::PrintPreviewDialogController* controller =
+      new PrintPreviewDialogControllerTest(GetContext());
+#else
   printing::PrintPreviewDialogController* controller =
       printing::PrintPreviewDialogController::GetInstance();
+#endif
   ASSERT_TRUE(controller);
 
   printing::PrintViewManager* print_view_manager =
       printing::PrintViewManager::FromWebContents(initiator);
   print_view_manager->PrintPreviewNow(false);
   WebContents* preview_dialog = controller->GetOrCreatePreviewDialog(initiator);
 
   EXPECT_NE(initiator, preview_dialog);
   EXPECT_EQ(1, browser()->tab_strip_model()->count());
   EXPECT_TRUE(IsShowingWebContentsModalDialog(initiator));
@@ skipping 24 matching lines @@
   preview_ui->SetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX,
                                           dummy_data.get());
 
   // Clear the preview data.
   preview_ui->ClearAllPreviewData();
 
   preview_ui->GetPrintPreviewDataForIndex(
       printing::COMPLETE_PREVIEW_DOCUMENT_INDEX,
       &data);
   EXPECT_EQ(NULL, data.get());
+
+  initiator->Close();
 }
 
 // Set and get the individual draft pages.
 TEST_F(PrintPreviewUIUnitTest, PrintPreviewDraftPages) {
   WebContents* initiator = browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_TRUE(initiator);
 
+#if defined(USE_AURA)
+  printing::PrintPreviewDialogController* controller =
+      new PrintPreviewDialogControllerTest(GetContext());
+#else
   printing::PrintPreviewDialogController* controller =
       printing::PrintPreviewDialogController::GetInstance();
+#endif
   ASSERT_TRUE(controller);
 
   printing::PrintViewManager* print_view_manager =
       printing::PrintViewManager::FromWebContents(initiator);
   print_view_manager->PrintPreviewNow(false);
   WebContents* preview_dialog = controller->GetOrCreatePreviewDialog(initiator);
 
   EXPECT_NE(initiator, preview_dialog);
   EXPECT_EQ(1, browser()->tab_strip_model()->count());
   EXPECT_TRUE(IsShowingWebContentsModalDialog(initiator));
@@ skipping 31 matching lines @@
                                           dummy_data.get());
   preview_ui->GetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX + 1,
                                           &data);
   EXPECT_EQ(dummy_data->size(), data->size());
   EXPECT_EQ(dummy_data.get(), data.get());
 
   // Clear the preview data.
   preview_ui->ClearAllPreviewData();
   preview_ui->GetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX, &data);
   EXPECT_EQ(NULL, data.get());
+
+  initiator->Close();
 }
 
 // Test the browser-side print preview cancellation functionality.
 TEST_F(PrintPreviewUIUnitTest, GetCurrentPrintPreviewStatus) {
   WebContents* initiator = browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_TRUE(initiator);
 
+#if defined(USE_AURA)
+  printing::PrintPreviewDialogController* controller =
+      new PrintPreviewDialogControllerTest(GetContext());
+#else
   printing::PrintPreviewDialogController* controller =
       printing::PrintPreviewDialogController::GetInstance();
+#endif
   ASSERT_TRUE(controller);
 
   printing::PrintViewManager* print_view_manager =
       printing::PrintViewManager::FromWebContents(initiator);
   print_view_manager->PrintPreviewNow(false);
   WebContents* preview_dialog = controller->GetOrCreatePreviewDialog(initiator);
 
   EXPECT_NE(initiator, preview_dialog);
   EXPECT_EQ(1, browser()->tab_strip_model()->count());
   EXPECT_TRUE(IsShowingWebContentsModalDialog(initiator));
@@ skipping 28 matching lines @@
   preview_ui->OnPrintPreviewRequest(kSecondRequestId);
   cancel = false;
   preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kFirstRequestId,
                                            &cancel);
   EXPECT_TRUE(cancel);
 
   cancel = true;
   preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kSecondRequestId,
                                            &cancel);
   EXPECT_FALSE(cancel);
+
+  initiator->Close();
 }