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/browser_iterator.h"
6
7 #include "chrome/browser/ui/browser.h"
8 #include "chrome/browser/ui/browser_list.h"
9 #include "chrome/browser/ui/host_desktop.h"
10 #include "chrome/test/base/browser_with_test_window_test.h"
11
12 typedef BrowserWithTestWindowTest BrowserIteratorTest;
13
14 namespace {
15
16 // BrowserWithTestWindowTest's default window is on the native desktop by
17 // default. Force it to be on the Ash desktop to be able to test the iterator
18 // in a situation with no browsers on the native desktop.
19 class BrowserIteratorTestWithInitialWindowInAsh
20 : public BrowserWithTestWindowTest {
21 public:
BrowserIteratorTestWithInitialWindowInAsh()22 BrowserIteratorTestWithInitialWindowInAsh()
23 : BrowserWithTestWindowTest(Browser::TYPE_TABBED,
24 chrome::HOST_DESKTOP_TYPE_ASH,
25 false) {
26 }
27 };
28
29 // Helper function to iterate and count all the browsers.
CountAllBrowsers()30 size_t CountAllBrowsers() {
31 size_t count = 0;
32 for (chrome::BrowserIterator iterator; !iterator.done(); iterator.Next())
33 ++count;
34 return count;
35 }
36
37 }
38
TEST_F(BrowserIteratorTest,BrowsersOnMultipleDesktops)39 TEST_F(BrowserIteratorTest, BrowsersOnMultipleDesktops) {
40 Browser::CreateParams native_params(profile(),
41 chrome::HOST_DESKTOP_TYPE_NATIVE);
42 Browser::CreateParams ash_params(profile(), chrome::HOST_DESKTOP_TYPE_ASH);
43 // Note, browser 1 is on the native desktop.
44 scoped_ptr<Browser> browser2(
45 chrome::CreateBrowserWithTestWindowForParams(&native_params));
46 scoped_ptr<Browser> browser3(
47 chrome::CreateBrowserWithTestWindowForParams(&native_params));
48 scoped_ptr<Browser> browser4(
49 chrome::CreateBrowserWithTestWindowForParams(&ash_params));
50 scoped_ptr<Browser> browser5(
51 chrome::CreateBrowserWithTestWindowForParams(&ash_params));
52
53 // Sanity checks.
54 BrowserList* native_list =
55 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE);
56 #if defined(OS_CHROMEOS)
57 // On Chrome OS, the native list and the ash list are the same.
58 EXPECT_EQ(5U, native_list->size());
59 #else
60 BrowserList* ash_list =
61 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_ASH);
62 EXPECT_EQ(3U, native_list->size());
63 EXPECT_EQ(2U, ash_list->size());
64 #endif
65
66 // Make sure the iterator finds all 5 browsers regardless of which desktop
67 // they are on.
68 EXPECT_EQ(5U, CountAllBrowsers());
69 }
70
TEST_F(BrowserIteratorTest,NoBrowsersOnAshDesktop)71 TEST_F(BrowserIteratorTest, NoBrowsersOnAshDesktop) {
72 Browser::CreateParams native_params(profile(),
73 chrome::HOST_DESKTOP_TYPE_NATIVE);
74 // Note, browser 1 is on the native desktop.
75 scoped_ptr<Browser> browser2(
76 chrome::CreateBrowserWithTestWindowForParams(&native_params));
77 scoped_ptr<Browser> browser3(
78 chrome::CreateBrowserWithTestWindowForParams(&native_params));
79
80 // Sanity checks.
81 BrowserList* native_list =
82 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE);
83 EXPECT_EQ(3U, native_list->size());
84 #if !defined(OS_CHROMEOS)
85 // On non-Chrome OS platforms the Ash list is independent from the native
86 // list, double-check that it's empty.
87 BrowserList* ash_list =
88 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_ASH);
89 EXPECT_EQ(0U, ash_list->size());
90 #endif
91
92 // Make sure the iterator finds all 3 browsers on the native desktop and
93 // doesn't trip over the empty Ash desktop list.
94 EXPECT_EQ(3U, CountAllBrowsers());
95 }
96
TEST_F(BrowserIteratorTestWithInitialWindowInAsh,NoBrowsersOnNativeDesktop)97 TEST_F(BrowserIteratorTestWithInitialWindowInAsh, NoBrowsersOnNativeDesktop) {
98 Browser::CreateParams ash_params(profile(), chrome::HOST_DESKTOP_TYPE_ASH);
99 // Note, browser 1 is on the ash desktop.
100 scoped_ptr<Browser> browser2(
101 chrome::CreateBrowserWithTestWindowForParams(&ash_params));
102 scoped_ptr<Browser> browser3(
103 chrome::CreateBrowserWithTestWindowForParams(&ash_params));
104
105 BrowserList* ash_list =
106 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_ASH);
107 EXPECT_EQ(3U, ash_list->size());
108 #if !defined(OS_CHROMEOS)
109 // On non-Chrome OS platforms the native list is independent from the Ash
110 // list; double-check that it's empty.
111 BrowserList* native_list =
112 BrowserList::GetInstance(chrome::HOST_DESKTOP_TYPE_NATIVE);
113 EXPECT_EQ(0U, native_list->size());
114 #endif
115
116 // Make sure the iterator finds all 3 browsers on the ash desktop and
117 // doesn't trip over the empty native desktop list.
118 EXPECT_EQ(3U, CountAllBrowsers());
119 }
120
121 // Verify that the iterator still behaves if there are no browsers at all.
TEST(BrowserIteratorTestWithNoTestWindow,NoBrowser)122 TEST(BrowserIteratorTestWithNoTestWindow, NoBrowser) {
123 EXPECT_EQ(0U, CountAllBrowsers());
124 }
125