| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <vector> | 5 #include <vector> |
| 6 | 6 |
| 7 #include "base/memory/ref_counted_memory.h" | 7 #include "base/memory/ref_counted_memory.h" |
| 8 #include "chrome/browser/prefs/pref_service.h" | 8 #include "chrome/browser/prefs/pref_service.h" |
| 9 #include "chrome/browser/printing/print_preview_tab_controller.h" | 9 #include "chrome/browser/printing/print_preview_tab_controller.h" |
| 10 #include "chrome/browser/printing/print_preview_test.h" |
| 10 #include "chrome/browser/printing/print_view_manager.h" | 11 #include "chrome/browser/printing/print_view_manager.h" |
| 11 #include "chrome/browser/ui/browser_commands.h" | 12 #include "chrome/browser/ui/browser_commands.h" |
| 12 #include "chrome/browser/ui/browser_tabstrip.h" | 13 #include "chrome/browser/ui/browser_tabstrip.h" |
| 13 #include "chrome/browser/ui/constrained_window_tab_helper.h" | 14 #include "chrome/browser/ui/constrained_window_tab_helper.h" |
| 14 #include "chrome/browser/ui/tab_contents/tab_contents.h" | 15 #include "chrome/browser/ui/tab_contents/tab_contents.h" |
| 15 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h" | 16 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h" |
| 16 #include "chrome/common/pref_names.h" | 17 #include "chrome/common/pref_names.h" |
| 17 #include "chrome/test/base/browser_with_test_window_test.h" | 18 #include "chrome/test/base/browser_with_test_window_test.h" |
| 19 #include "content/public/browser/plugin_service.h" |
| 18 #include "content/public/browser/site_instance.h" | 20 #include "content/public/browser/site_instance.h" |
| 19 #include "content/public/browser/web_contents.h" | 21 #include "content/public/browser/web_contents.h" |
| 20 #include "content/public/test/web_contents_tester.h" | 22 #include "content/public/test/web_contents_tester.h" |
| 21 #include "printing/print_job_constants.h" | 23 #include "printing/print_job_constants.h" |
| 24 #include "webkit/plugins/npapi/mock_plugin_list.h" |
| 22 | 25 |
| 23 using content::WebContents; | 26 using content::WebContents; |
| 24 using content::WebContentsTester; | 27 using content::WebContentsTester; |
| 25 | 28 |
| 26 namespace { | 29 namespace { |
| 27 | 30 |
| 28 base::RefCountedBytes* CreateTestData() { | 31 base::RefCountedBytes* CreateTestData() { |
| 29 const unsigned char blob1[] = | 32 const unsigned char blob1[] = |
| 30 "12346102356120394751634516591348710478123649165419234519234512349134"; | 33 "12346102356120394751634516591348710478123649165419234519234512349134"; |
| 31 std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1)); | 34 std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1)); |
| 32 return new base::RefCountedBytes(preview_data); | 35 return new base::RefCountedBytes(preview_data); |
| 33 } | 36 } |
| 34 | 37 |
| 35 size_t GetConstrainedWindowCount(TabContents* tab) { | 38 size_t GetConstrainedWindowCount(TabContents* tab) { |
| 36 ConstrainedWindowTabHelper* constrained_window_tab_helper = | 39 ConstrainedWindowTabHelper* constrained_window_tab_helper = |
| 37 ConstrainedWindowTabHelper::FromWebContents(tab->web_contents()); | 40 ConstrainedWindowTabHelper::FromWebContents(tab->web_contents()); |
| 38 return constrained_window_tab_helper->constrained_window_count(); | 41 return constrained_window_tab_helper->constrained_window_count(); |
| 39 } | 42 } |
| 40 | 43 |
| 41 } // namespace | 44 } // namespace |
| 42 | 45 |
| 43 class PrintPreviewUIUnitTest : public BrowserWithTestWindowTest { | 46 class PrintPreviewUIUnitTest : public PrintPreviewTest { |
| 44 public: | 47 public: |
| 45 PrintPreviewUIUnitTest() {} | 48 PrintPreviewUIUnitTest(); |
| 46 virtual ~PrintPreviewUIUnitTest() {} | 49 virtual ~PrintPreviewUIUnitTest(); |
| 47 | 50 |
| 48 protected: | 51 protected: |
| 49 virtual void SetUp() OVERRIDE { | 52 virtual void SetUp() OVERRIDE; |
| 50 BrowserWithTestWindowTest::SetUp(); | |
| 51 | 53 |
| 52 profile()->GetPrefs()->SetBoolean(prefs::kPrintPreviewDisabled, false); | 54 DISALLOW_COPY_AND_ASSIGN(PrintPreviewUIUnitTest); |
| 55 }; |
| 53 | 56 |
| 54 chrome::NewTab(browser()); | 57 PrintPreviewUIUnitTest::PrintPreviewUIUnitTest() {} |
| 55 } | 58 PrintPreviewUIUnitTest::~PrintPreviewUIUnitTest() {} |
| 56 }; | 59 |
| 60 void PrintPreviewUIUnitTest::SetUp() { |
| 61 PrintPreviewTest::SetUp(); |
| 62 |
| 63 chrome::NewTab(browser()); |
| 64 } |
| 57 | 65 |
| 58 // Create/Get a preview tab for initiator tab. | 66 // Create/Get a preview tab for initiator tab. |
| 59 TEST_F(PrintPreviewUIUnitTest, PrintPreviewData) { | 67 TEST_F(PrintPreviewUIUnitTest, PrintPreviewData) { |
| 60 TabContents* initiator_tab = chrome::GetActiveTabContents(browser()); | 68 TabContents* initiator_tab = chrome::GetActiveTabContents(browser()); |
| 61 ASSERT_TRUE(initiator_tab); | 69 ASSERT_TRUE(initiator_tab); |
| 62 EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); | 70 EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); |
| 63 | 71 |
| 64 printing::PrintPreviewTabController* controller = | 72 printing::PrintPreviewTabController* controller = |
| 65 printing::PrintPreviewTabController::GetInstance(); | 73 printing::PrintPreviewTabController::GetInstance(); |
| 66 ASSERT_TRUE(controller); | 74 ASSERT_TRUE(controller); |
| (...skipping 193 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 260 PrintPreviewUI* preview_ui = static_cast<PrintPreviewUI*>( | 268 PrintPreviewUI* preview_ui = static_cast<PrintPreviewUI*>( |
| 261 preview_tab->web_contents()->GetWebUI()->GetController()); | 269 preview_tab->web_contents()->GetWebUI()->GetController()); |
| 262 ASSERT_TRUE(preview_ui != NULL); | 270 ASSERT_TRUE(preview_ui != NULL); |
| 263 | 271 |
| 264 preview_ui->OnPrintPreviewTabClosed(); | 272 preview_ui->OnPrintPreviewTabClosed(); |
| 265 | 273 |
| 266 EXPECT_EQ(2, browser()->tab_count()); | 274 EXPECT_EQ(2, browser()->tab_count()); |
| 267 EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); | 275 EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); |
| 268 EXPECT_EQ(1, initiator_tester->GetNumberOfFocusCalls()); | 276 EXPECT_EQ(1, initiator_tester->GetNumberOfFocusCalls()); |
| 269 } | 277 } |
| OLD | NEW |