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 3 // found in the LICENSE file. 4 5 #include "chrome/browser/ui/startup/session_crashed_infobar_delegate.h" 6 7 #include "base/prefs/pref_registry_simple.h" 8 #include "base/prefs/testing_pref_service.h" 9 #include "base/run_loop.h" 10 #include "chrome/browser/infobars/infobar_service.h" 11 #include "chrome/browser/prefs/browser_prefs.h" 12 #include "chrome/browser/sessions/session_service_factory.h" 13 #include "chrome/browser/ui/tabs/tab_strip_model.h" 14 #include "chrome/common/pref_names.h" 15 #include "chrome/common/url_constants.h" 16 #include "chrome/test/base/browser_with_test_window_test.h" 17 #include "chrome/test/base/scoped_testing_local_state.h" 18 #include "chrome/test/base/testing_browser_process.h" 19 #include "components/infobars/core/infobar.h" 20 21 class SessionCrashedInfoBarDelegateUnitTest : public BrowserWithTestWindowTest { 22 public: 23 virtual void SetUp() OVERRIDE { 24 static_cast<TestingBrowserProcess*>(g_browser_process) 25 ->SetLocalState(&pref_service); 26 chrome::RegisterLocalState(pref_service.registry()); 27 28 // This needs to be called after the local state is set, because it will 29 // create a browser which will try to read from the local state. 30 BrowserWithTestWindowTest::SetUp(); 31 } 32 33 virtual void TearDown() OVERRIDE { 34 static_cast<TestingBrowserProcess*>(g_browser_process)->SetLocalState(NULL); 35 BrowserWithTestWindowTest::TearDown(); 36 } 37 38 private: 39 TestingPrefServiceSimple pref_service; 40 }; 41 42 TEST_F(SessionCrashedInfoBarDelegateUnitTest, DetachingTabWithCrashedInfoBar) { 43 SessionServiceFactory::SetForTestProfile( 44 browser()->profile(), 45 static_cast<SessionService*>( 46 SessionServiceFactory::GetInstance()->BuildServiceInstanceFor( 47 browser()->profile()))); 48 49 // Create a browser which we can close during the test. 50 Browser::CreateParams params(browser()->profile(), 51 browser()->host_desktop_type()); 52 scoped_ptr<Browser> first_browser( 53 chrome::CreateBrowserWithTestWindowForParams(¶ms)); 54 AddTab(first_browser.get(), GURL(chrome::kChromeUINewTabURL)); 55 56 // Attach the crashed infobar to it. 57 SessionCrashedInfoBarDelegate::Create(first_browser.get()); 58 59 TabStripModel* tab_strip = first_browser->tab_strip_model(); 60 ASSERT_EQ(1, tab_strip->count()); 61 content::WebContents* web_contents = tab_strip->GetWebContentsAt(0); 62 InfoBarService* infobar_service = 63 InfoBarService::FromWebContents(web_contents); 64 EXPECT_EQ(1U, infobar_service->infobar_count()); 65 ConfirmInfoBarDelegate* infobar = 66 infobar_service->infobar_at(0)->delegate()->AsConfirmInfoBarDelegate(); 67 ASSERT_TRUE(infobar); 68 69 // Open another browser. 70 scoped_ptr<Browser> opened_browser( 71 chrome::CreateBrowserWithTestWindowForParams(¶ms)); 72 73 // Move the tab which is destroying the crash info bar to the new browser. 74 tab_strip->DetachWebContentsAt(0); 75 tab_strip = opened_browser->tab_strip_model(); 76 tab_strip->AppendWebContents(web_contents, true); 77 78 // Close the original browser. 79 first_browser->window()->Close(); 80 first_browser.reset(); 81 82 ASSERT_EQ(1, tab_strip->count()); 83 infobar_service = 84 InfoBarService::FromWebContents(tab_strip->GetWebContentsAt(0)); 85 EXPECT_EQ(1U, infobar_service->infobar_count()); 86 87 // This used to crash. 88 infobar->Accept(); 89 90 // Ramp down the test. 91 tab_strip->CloseWebContentsAt(0, TabStripModel::CLOSE_NONE); 92 } 93