Home | History | Annotate | Download | only in startup 
      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(&params));
     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(&params));
     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