chrome/browser/ui/tab_contents/tab_contents_iterator_unittest.cc

   1 // Copyright (c) 2012 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/tab_contents/tab_contents_iterator.h"
   6
   7 #include "base/prefs/pref_registry_simple.h"
   8 #include "base/prefs/testing_pref_service.h"
   9 #include "chrome/browser/browser_process.h"
  10 #include "chrome/browser/browser_shutdown.h"
  11 #include "chrome/browser/lifetime/application_lifetime.h"
  12 #include "chrome/browser/profiles/profile_manager.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "chrome/browser/ui/browser_commands.h"
  15 #include "chrome/browser/ui/browser_list.h"
  16 #include "chrome/browser/ui/host_desktop.h"
  17 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  18 #include "chrome/common/pref_names.h"
  19 #include "chrome/test/base/browser_with_test_window_test.h"
  20 #include "chrome/test/base/test_browser_window.h"
  21 #include "chrome/test/base/testing_browser_process.h"
  22
  23 typedef BrowserWithTestWindowTest BrowserListTest;
  24
  25 namespace {
  26
  27 // Helper function to iterate and count all the tabs.
  28 size_t CountAllTabs() {
  29   size_t count = 0;
  30   for (TabContentsIterator iterator; !iterator.done(); iterator.Next())
  31     ++count;
  32   return count;
  33 }
  34
  35 }  // namespace
  36
  37 TEST_F(BrowserListTest, TabContentsIteratorVerifyCount) {
  38   // Make sure we have 1 window to start with.
  39   EXPECT_EQ(1U,
  40             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
  41
  42   EXPECT_EQ(0U, CountAllTabs());
  43
  44   // Create more browsers/windows.
  45   Browser::CreateParams native_params(profile(),
  46                                       chrome::HOST_DESKTOP_TYPE_NATIVE);
  47   scoped_ptr<Browser> browser2(
  48       chrome::CreateBrowserWithTestWindowForParams(&native_params));
  49   // Create browser 3 and 4 on the Ash desktop (the TabContentsIterator
  50   // shouldn't see the difference).
  51   Browser::CreateParams ash_params(profile(), chrome::HOST_DESKTOP_TYPE_ASH);
  52   scoped_ptr<Browser> browser3(
  53       chrome::CreateBrowserWithTestWindowForParams(&ash_params));
  54   scoped_ptr<Browser> browser4(
  55       chrome::CreateBrowserWithTestWindowForParams(&ash_params));
  56
  57   // Sanity checks.
  58 #if defined(OS_CHROMEOS)
  59   // The ash desktop is the native desktop on Chrome OS.
  60   EXPECT_EQ(4U,
  61             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
  62 #else
  63   EXPECT_EQ(2U,
  64             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
  65   EXPECT_EQ(2U,
  66             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_ASH)->size());
  67 #endif
  68   EXPECT_EQ(0, browser()->tab_strip_model()->count());
  69   EXPECT_EQ(0, browser2->tab_strip_model()->count());
  70   EXPECT_EQ(0, browser3->tab_strip_model()->count());
  71   EXPECT_EQ(0, browser4->tab_strip_model()->count());
  72
  73   EXPECT_EQ(0U, CountAllTabs());
  74
  75   // Add some tabs.
  76   for (size_t i = 0; i < 3; ++i)
  77     chrome::NewTab(browser2.get());
  78   chrome::NewTab(browser3.get());
  79
  80   EXPECT_EQ(4U, CountAllTabs());
  81
  82   // Close some tabs.
  83   browser2->tab_strip_model()->CloseAllTabs();
  84
  85   EXPECT_EQ(1U, CountAllTabs());
  86
  87   // Add lots of tabs.
  88   for (size_t i = 0; i < 41; ++i)
  89     chrome::NewTab(browser());
  90
  91   EXPECT_EQ(42U, CountAllTabs());
  92   // Close all remaining tabs to keep all the destructors happy.
  93   browser3->tab_strip_model()->CloseAllTabs();
  94 }
  95
  96 TEST_F(BrowserListTest, TabContentsIteratorVerifyBrowser) {
  97   // Make sure we have 1 window to start with.
  98   EXPECT_EQ(1U,
  99             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
 100
 101   // Create more browsers/windows.
 102   Browser::CreateParams native_params(profile(),
 103                                       chrome::HOST_DESKTOP_TYPE_NATIVE);
 104   scoped_ptr<Browser> browser2(
 105       chrome::CreateBrowserWithTestWindowForParams(&native_params));
 106   // Create browser 3 on the Ash desktop (the TabContentsIterator shouldn't see
 107   // the difference).
 108   Browser::CreateParams ash_params(profile(), chrome::HOST_DESKTOP_TYPE_ASH);
 109   scoped_ptr<Browser> browser3(
 110       chrome::CreateBrowserWithTestWindowForParams(&ash_params));
 111
 112   // Sanity checks.
 113 #if defined(OS_CHROMEOS)
 114   // The ash desktop is the native desktop on Chrome OS.
 115   EXPECT_EQ(3U,
 116             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
 117 #else
 118   EXPECT_EQ(2U,
 119             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE)->size());
 120   EXPECT_EQ(1U,
 121             BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_ASH)->size());
 122 #endif
 123   EXPECT_EQ(0, browser()->tab_strip_model()->count());
 124   EXPECT_EQ(0, browser2->tab_strip_model()->count());
 125   EXPECT_EQ(0, browser3->tab_strip_model()->count());
 126
 127   EXPECT_EQ(0U, CountAllTabs());
 128
 129   // Add some tabs.
 130   for (size_t i = 0; i < 3; ++i)
 131     chrome::NewTab(browser2.get());
 132   for (size_t i = 0; i < 2; ++i)
 133     chrome::NewTab(browser3.get());
 134
 135   size_t count = 0;
 136   for (TabContentsIterator iterator; !iterator.done(); iterator.Next(),
 137                                                        ++count) {
 138     if (count < 3)
 139       EXPECT_EQ(browser2.get(), iterator.browser());
 140     else if (count < 5)
 141       EXPECT_EQ(browser3.get(), iterator.browser());
 142     else
 143       ADD_FAILURE();
 144   }
 145
 146   // Close some tabs.
 147   browser2->tab_strip_model()->CloseAllTabs();
 148   browser3->tab_strip_model()->CloseWebContentsAt(1, TabStripModel::CLOSE_NONE);
 149
 150   count = 0;
 151   for (TabContentsIterator iterator; !iterator.done(); iterator.Next(),
 152                                                        ++count) {
 153     if (count == 0)
 154       EXPECT_EQ(browser3.get(), iterator.browser());
 155     else
 156       ADD_FAILURE();
 157   }
 158
 159   // Now make it one tab per browser.
 160   chrome::NewTab(browser());
 161   chrome::NewTab(browser2.get());
 162
 163   count = 0;
 164   for (TabContentsIterator iterator; !iterator.done(); iterator.Next(),
 165                                                        ++count) {
 166     if (count == 0)
 167       EXPECT_EQ(browser(), iterator.browser());
 168     else if (count == 1)
 169       EXPECT_EQ(browser2.get(), iterator.browser());
 170     else if (count == 2)
 171       EXPECT_EQ(browser3.get(), iterator.browser());
 172     else
 173       ADD_FAILURE();
 174   }
 175
 176   // Close all remaining tabs to keep all the destructors happy.
 177   browser2->tab_strip_model()->CloseAllTabs();
 178   browser3->tab_strip_model()->CloseAllTabs();
 179 }
 180
 181 #if defined(OS_CHROMEOS)
 182 // Calling AttemptRestart on ChromeOS will exit the test.
 183 #define MAYBE_AttemptRestart DISABLED_AttemptRestart
 184 #else
 185 #define MAYBE_AttemptRestart AttemptRestart
 186 #endif
 187
 188 TEST_F(BrowserListTest, MAYBE_AttemptRestart) {
 189   ASSERT_TRUE(g_browser_process);
 190   TestingPrefServiceSimple testing_pref_service;
 191   testing_pref_service.registry()->RegisterBooleanPref(
 192       prefs::kWasRestarted, false);
 193   testing_pref_service.registry()->RegisterBooleanPref(
 194       prefs::kRestartLastSessionOnShutdown,
 195       false);
 196   testing_pref_service.registry()->RegisterListPref(
 197       prefs::kProfilesLastActive);
 198
 199   TestingBrowserProcess* testing_browser_process =
 200       TestingBrowserProcess::GetGlobal();
 201   testing_browser_process->SetLocalState(&testing_pref_service);
 202   ASSERT_TRUE(g_browser_process->local_state());
 203   ProfileManager* profile_manager = new ProfileManager(base::FilePath());
 204   testing_browser_process->SetProfileManager(profile_manager);
 205
 206   chrome::AttemptRestart();
 207   // Cancel the effects of us calling chrome::AttemptRestart. Otherwise tests
 208   // ran after this one will fail.
 209   browser_shutdown::SetTryingToQuit(false);
 210
 211   EXPECT_TRUE(testing_pref_service.GetBoolean(prefs::kWasRestarted));
 212   testing_browser_process->SetLocalState(NULL);
 213 }