| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 "chrome/browser/ui/startup/session_crashed_infobar_delegate.h" | 5 #include "chrome/browser/ui/startup/session_crashed_infobar_delegate.h" |
| 6 | 6 |
| 7 #include "base/prefs/pref_registry_simple.h" | 7 #include "base/prefs/pref_registry_simple.h" |
| 8 #include "base/prefs/testing_pref_service.h" | 8 #include "base/prefs/testing_pref_service.h" |
| 9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
| 10 #include "chrome/browser/infobars/infobar_service.h" | 10 #include "chrome/browser/infobars/infobar_service.h" |
| 11 #include "chrome/browser/prefs/browser_prefs.h" | 11 #include "chrome/browser/prefs/browser_prefs.h" |
| 12 #include "chrome/browser/sessions/session_service_factory.h" | 12 #include "chrome/browser/sessions/session_service_factory.h" |
| 13 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 13 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| 14 #include "chrome/common/pref_names.h" | 14 #include "chrome/common/pref_names.h" |
| 15 #include "chrome/common/url_constants.h" | 15 #include "chrome/common/url_constants.h" |
| 16 #include "chrome/test/base/browser_with_test_window_test.h" | 16 #include "chrome/test/base/browser_with_test_window_test.h" |
| 17 #include "chrome/test/base/scoped_testing_local_state.h" | 17 #include "chrome/test/base/scoped_testing_local_state.h" |
| 18 #include "chrome/test/base/testing_browser_process.h" | 18 #include "chrome/test/base/testing_browser_process.h" |
| 19 #include "components/infobars/core/infobar.h" | 19 #include "components/infobars/core/infobar.h" |
| 20 | 20 |
| 21 #if defined(OS_CHROMEOS) | |
| 22 #include "base/command_line.h" | |
| 23 #include "chrome/common/chrome_switches.h" | |
| 24 #endif | |
| 25 | |
| 26 #if defined(OS_CHROMEOS) | |
| 27 // TODO(nkostylev): Cleanup this code once multi-profiles are enabled by | |
| 28 // default on CrOS. http://crbug.com/351655 | |
| 29 class SessionCrashedInfoBarDelegateUnitTest : | |
| 30 public BrowserWithTestWindowTest, | |
| 31 public testing::WithParamInterface<bool> { | |
| 32 #else | |
| 33 class SessionCrashedInfoBarDelegateUnitTest : public BrowserWithTestWindowTest { | 21 class SessionCrashedInfoBarDelegateUnitTest : public BrowserWithTestWindowTest { |
| 34 #endif | |
| 35 public: | 22 public: |
| 36 virtual void SetUp() OVERRIDE { | 23 virtual void SetUp() OVERRIDE { |
| 37 static_cast<TestingBrowserProcess*>(g_browser_process) | 24 static_cast<TestingBrowserProcess*>(g_browser_process) |
| 38 ->SetLocalState(&pref_service); | 25 ->SetLocalState(&pref_service); |
| 39 chrome::RegisterLocalState(pref_service.registry()); | 26 chrome::RegisterLocalState(pref_service.registry()); |
| 40 | 27 |
| 41 #if defined(OS_CHROMEOS) | |
| 42 if (GetParam()) | |
| 43 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kMultiProfiles); | |
| 44 #endif | |
| 45 | |
| 46 // This needs to be called after the local state is set, because it will | 28 // This needs to be called after the local state is set, because it will |
| 47 // create a browser which will try to read from the local state. | 29 // create a browser which will try to read from the local state. |
| 48 BrowserWithTestWindowTest::SetUp(); | 30 BrowserWithTestWindowTest::SetUp(); |
| 49 } | 31 } |
| 50 | 32 |
| 51 virtual void TearDown() OVERRIDE { | 33 virtual void TearDown() OVERRIDE { |
| 52 static_cast<TestingBrowserProcess*>(g_browser_process)->SetLocalState(NULL); | 34 static_cast<TestingBrowserProcess*>(g_browser_process)->SetLocalState(NULL); |
| 53 BrowserWithTestWindowTest::TearDown(); | 35 BrowserWithTestWindowTest::TearDown(); |
| 54 } | 36 } |
| 55 | 37 |
| 56 private: | 38 private: |
| 57 TestingPrefServiceSimple pref_service; | 39 TestingPrefServiceSimple pref_service; |
| 58 }; | 40 }; |
| 59 | 41 |
| 60 #if defined(OS_CHROMEOS) | |
| 61 TEST_P(SessionCrashedInfoBarDelegateUnitTest, DetachingTabWithCrashedInfoBar) { | |
| 62 #else | |
| 63 TEST_F(SessionCrashedInfoBarDelegateUnitTest, DetachingTabWithCrashedInfoBar) { | 42 TEST_F(SessionCrashedInfoBarDelegateUnitTest, DetachingTabWithCrashedInfoBar) { |
| 64 #endif | |
| 65 SessionServiceFactory::SetForTestProfile( | 43 SessionServiceFactory::SetForTestProfile( |
| 66 browser()->profile(), | 44 browser()->profile(), |
| 67 static_cast<SessionService*>( | 45 static_cast<SessionService*>( |
| 68 SessionServiceFactory::GetInstance()->BuildServiceInstanceFor( | 46 SessionServiceFactory::GetInstance()->BuildServiceInstanceFor( |
| 69 browser()->profile()))); | 47 browser()->profile()))); |
| 70 | 48 |
| 71 // Create a browser which we can close during the test. | 49 // Create a browser which we can close during the test. |
| 72 Browser::CreateParams params(browser()->profile(), | 50 Browser::CreateParams params(browser()->profile(), |
| 73 browser()->host_desktop_type()); | 51 browser()->host_desktop_type()); |
| 74 scoped_ptr<Browser> first_browser( | 52 scoped_ptr<Browser> first_browser( |
| (...skipping 30 matching lines...) Expand all Loading... |
| 105 infobar_service = | 83 infobar_service = |
| 106 InfoBarService::FromWebContents(tab_strip->GetWebContentsAt(0)); | 84 InfoBarService::FromWebContents(tab_strip->GetWebContentsAt(0)); |
| 107 EXPECT_EQ(1U, infobar_service->infobar_count()); | 85 EXPECT_EQ(1U, infobar_service->infobar_count()); |
| 108 | 86 |
| 109 // This used to crash. | 87 // This used to crash. |
| 110 infobar->Accept(); | 88 infobar->Accept(); |
| 111 | 89 |
| 112 // Ramp down the test. | 90 // Ramp down the test. |
| 113 tab_strip->CloseWebContentsAt(0, TabStripModel::CLOSE_NONE); | 91 tab_strip->CloseWebContentsAt(0, TabStripModel::CLOSE_NONE); |
| 114 } | 92 } |
| 115 | |
| 116 #if defined(OS_CHROMEOS) | |
| 117 INSTANTIATE_TEST_CASE_P(SessionCrashedInfoBarDelegateUnitTestInstantiation, | |
| 118 SessionCrashedInfoBarDelegateUnitTest, | |
| 119 testing::Bool()); | |
| 120 #endif | |
| OLD | NEW |