190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// Copyright (c) 2013 The Chromium Authors. All rights reserved. 290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// found in the LICENSE file. 490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 5ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "chrome/browser/ui/startup/session_crashed_infobar_delegate.h" 6ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch 790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/prefs/pref_registry_simple.h" 890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/prefs/testing_pref_service.h" 990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/run_loop.h" 1090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/browser/infobars/infobar_service.h" 1190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/browser/sessions/session_service_factory.h" 1290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/browser/ui/tabs/tab_strip_model.h" 1390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/common/pref_names.h" 1490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/common/url_constants.h" 1590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/test/base/browser_with_test_window_test.h" 1690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/test/base/scoped_testing_local_state.h" 1790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/test/base/testing_browser_process.h" 1890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 1990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)class SessionCrashedInfoBarDelegateUnitTest : public BrowserWithTestWindowTest { 2090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) public: 2190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void SetUp() OVERRIDE { 2290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) pref_service.registry() 2390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ->RegisterBooleanPref(prefs::kAllowFileSelectionDialogs, true); 2490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) pref_service.registry()->RegisterBooleanPref(prefs::kWasRestarted, false); 2590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) static_cast<TestingBrowserProcess*>(g_browser_process) 2690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ->SetLocalState(&pref_service); 2790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // This needs to be called after the local state is set, because it will 2890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // create a browser which will try to read from the local state. 2990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) BrowserWithTestWindowTest::SetUp(); 3090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 3190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void TearDown() OVERRIDE { 3390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) static_cast<TestingBrowserProcess*>(g_browser_process)->SetLocalState(NULL); 3490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) BrowserWithTestWindowTest::TearDown(); 3590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 3690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) private: 3890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TestingPrefServiceSimple pref_service; 3990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)}; 4090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 4190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)TEST_F(SessionCrashedInfoBarDelegateUnitTest, DetachingTabWithCrashedInfoBar) { 4290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) SessionServiceFactory::SetForTestProfile( 4390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) browser()->profile(), 4490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) static_cast<SessionService*>( 4590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) SessionServiceFactory::GetInstance()->BuildServiceInstanceFor( 4690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) browser()->profile()))); 4790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 4890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Create a browser which we can close during the test. 4990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) Browser::CreateParams params(browser()->profile(), 5090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) browser()->host_desktop_type()); 5190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) scoped_ptr<Browser> first_browser( 5290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) chrome::CreateBrowserWithTestWindowForParams(¶ms)); 5390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) AddTab(first_browser.get(), GURL(chrome::kChromeUINewTabURL)); 5490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 5590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Attach the crashed infobar to it. 5690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) SessionCrashedInfoBarDelegate::Create(first_browser.get()); 5790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 5890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TabStripModel* tab_strip = first_browser->tab_strip_model(); 5990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_EQ(1, tab_strip->count()); 6090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) content::WebContents* web_contents = tab_strip->GetWebContentsAt(0); 6190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) InfoBarService* infobar_service = 6290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) InfoBarService::FromWebContents(web_contents); 6390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(1U, infobar_service->infobar_count()); 64ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch scoped_ptr<ConfirmInfoBarDelegate> infobar(InfoBarService::FromWebContents( 65ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch web_contents)->infobar_at(0)->AsConfirmInfoBarDelegate()); 66ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch ASSERT_TRUE(infobar); 6790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 6890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Open another browser. 6990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) scoped_ptr<Browser> opened_browser( 7090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) chrome::CreateBrowserWithTestWindowForParams(¶ms)); 7190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 7290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Move the tab which is destroying the crash info bar to the new browser. 7390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) tab_strip->DetachWebContentsAt(0); 7490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) tab_strip = opened_browser->tab_strip_model(); 7590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) tab_strip->AppendWebContents(web_contents, true); 7690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 7790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Close the original browser. 7890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) first_browser->window()->Close(); 7990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) first_browser.reset(); 8090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 8190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_EQ(1, tab_strip->count()); 8290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) web_contents = tab_strip->GetWebContentsAt(0); 8390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) infobar_service = InfoBarService::FromWebContents(web_contents); 8490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(1U, infobar_service->infobar_count()); 8590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 8690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // This used to crash. 87ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch infobar->Accept(); 8890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 8990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Ramp down the test. 9090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) tab_strip->CloseWebContentsAt(0, TabStripModel::CLOSE_NONE); 9190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 92