Second try for

https://chromiumcodereview.appspot.com/9689047/

with fix to allow for special conditions under test
BUG=115650
TEST=LauncherUpdaterTest.*
TBR=sky

Review URL: https://chromiumcodereview.appspot.com/9702072

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@126917 0039d316-1c4b-4281-b951-d872f2087c98

3 files changed, 54 insertions, 12 deletions

diff --git a/ash/launcher/launcher.cc b/ash/launcher/launcher.cc
index 516c036..3c1543f 100644
--- a/ash/launcher/launcher.cc
+++ b/ash/launcher/launcher.cc
@@ -157,4 +157,8 @@ gfx::Rect Launcher::GetScreenBoundsOfItemIconForWindow(aura::Window* window) {
                    bounds.height());
 }
 
+internal::LauncherView* Launcher::GetLauncherViewForTest() {
+  return static_cast<internal::LauncherView*>(
+      widget_->GetContentsView()->child_at(0));
+}
 }  // namespace ash
diff --git a/ash/launcher/launcher.h b/ash/launcher/launcher.h
index 4005137..6825771 100644
--- a/ash/launcher/launcher.h
+++ b/ash/launcher/launcher.h
@@ -47,6 +47,8 @@ class ASH_EXPORT Launcher {
   // Returns the screen bounds of the item for the specified window. If there is
   // no item for the specified window an empty rect is returned.
   gfx::Rect GetScreenBoundsOfItemIconForWindow(aura::Window* window);
+  // Only to be called for testing. Retrieves the LauncherView.
+  internal::LauncherView* GetLauncherViewForTest();
 
   LauncherDelegate* delegate() { return delegate_.get(); }
 
@@ -68,6 +70,7 @@ class ASH_EXPORT Launcher {
   // Contents view of the widget. Houses the LauncherView.
   DelegateView* delegate_view_;
 
+  // LauncherView used to display icons.
   internal::LauncherView* launcher_view_;
 
   scoped_ptr<LauncherDelegate> delegate_;
diff --git a/ash/launcher/launcher_unittest.cc b/ash/launcher/launcher_unittest.cc
index 86b2af5..dbb0c12 100644
--- a/ash/launcher/launcher_unittest.cc
+++ b/ash/launcher/launcher_unittest.cc
@@ -22,10 +22,7 @@ namespace ash {
 // LauncherView.
 TEST_F(LauncherTest, SetStatusWidth) {
   Launcher* launcher = Shell::GetInstance()->launcher();
-  ASSERT_TRUE(launcher);
-  views::View* launcher_view = launcher->widget()->GetContentsView();
-  ASSERT_EQ(1, launcher_view->child_count());
-  launcher_view = launcher_view->child_at(0);
+  LauncherView* launcher_view = launcher->GetLauncherViewForTest();
 
   int total_width = launcher->widget()->GetWindowScreenBounds().width();
   ASSERT_GT(total_width, 0);
@@ -33,13 +30,12 @@ TEST_F(LauncherTest, SetStatusWidth) {
   EXPECT_EQ(total_width - total_width / 2, launcher_view->width());
 }
 
+// Confirm that launching an app gets the appropriate state reflected in
+// its button.
 TEST_F(LauncherTest, LaunchApp) {
   Launcher* launcher = Shell::GetInstance()->launcher();
   ASSERT_TRUE(launcher);
-  views::View* contents_view = launcher->widget()->GetContentsView();
-  ASSERT_EQ(1, contents_view->child_count());
-  LauncherView* launcher_view =
-      static_cast<LauncherView*>(contents_view->child_at(0));
+  LauncherView* launcher_view = launcher->GetLauncherViewForTest();
   LauncherView::TestAPI test(launcher_view);
   LauncherModel* model = launcher->model();
 
@@ -72,13 +68,12 @@ TEST_F(LauncherTest, LaunchApp) {
   ASSERT_EQ(--button_count, test.GetButtonCount());
 }
 
+// Confirm that launching a browser gets the appropriate state reflected in
+// its button.
 TEST_F(LauncherTest, OpenBrowser) {
   Launcher* launcher = Shell::GetInstance()->launcher();
   ASSERT_TRUE(launcher);
-  views::View* contents_view = launcher->widget()->GetContentsView();
-  ASSERT_EQ(1, contents_view->child_count());
-  LauncherView* launcher_view =
-      static_cast<LauncherView*>(contents_view->child_at(0));
+  LauncherView* launcher_view = launcher->GetLauncherViewForTest();
   LauncherView::TestAPI test(launcher_view);
   LauncherModel* model = launcher->model();
 
@@ -100,4 +95,44 @@ TEST_F(LauncherTest, OpenBrowser) {
   ASSERT_EQ(--button_count, test.GetButtonCount());
 }
 
+// Confirm that opening two different browsers and changing their activation
+// causes the appropriate state changes in the launcher buttons.
+TEST_F(LauncherTest, OpenTwoBrowsers) {
+  Launcher* launcher = Shell::GetInstance()->launcher();
+  ASSERT_TRUE(launcher);
+  LauncherView* launcher_view = launcher->GetLauncherViewForTest();
+  LauncherView::TestAPI test(launcher_view);
+  LauncherModel* model = launcher->model();
+
+  // Initially we have the app list and chrome icon.
+  int button_count = test.GetButtonCount();
+  int item_count = model->item_count();
+  int button1 = button_count, button2 = button_count + 1;
+  int item1 = item_count;
+
+  // Add active tab.
+  {
+    LauncherItem item;
+    item.type = TYPE_TABBED;
+    item.status = STATUS_ACTIVE;
+    model->Add(item_count, item);
+  }
+  ASSERT_EQ(++button_count, test.GetButtonCount());
+  EXPECT_EQ(LauncherButton::STATE_ACTIVE, test.GetButton(button1)->state());
+
+  // Add new active tab and deactivate other.
+  {
+    LauncherItem item;
+    item.type = TYPE_TABBED;
+    model->Add(item_count, item);
+    LauncherItem last_item = model->items()[item1];
+    last_item.status = STATUS_RUNNING;
+    model->Set(item1, last_item);
+  }
+
+  ASSERT_EQ(++button_count, test.GetButtonCount());
+  EXPECT_EQ(LauncherButton::STATE_RUNNING, test.GetButton(button1)->state());
+  EXPECT_EQ(LauncherButton::STATE_ACTIVE, test.GetButton(button2)->state());
+}
+
 }  // namespace ash