session_crashed_infobar_delegate_unittest.cc revision cedac228d2dd51db4b79ea1e72c7f249408ee061 
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
21class 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
42TEST_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