Move more stuff down into ash.

http://crbug.com/108457
TEST=none
TBR=sky
Review URL: http://codereview.chromium.org/9030007

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

1 files changed, 122 insertions, 0 deletions

diff --git a/ash/launcher/launcher_model_unittest.cc b/ash/launcher/launcher_model_unittest.cc
new file mode 100644
index 0000000..3c50b1e
--- /dev/null
+++ b/ash/launcher/launcher_model_unittest.cc
@@ -0,0 +1,122 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "ash/launcher/launcher_model.h"
+
+#include "ash/launcher/launcher_model_observer.h"
+#include "base/stringprintf.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace aura_shell {
+
+namespace {
+
+// LauncherModelObserver implementation that tracks what message are invoked.
+class TestLauncherModelObserver : public LauncherModelObserver {
+ public:
+  TestLauncherModelObserver()
+      : added_count_(0),
+        removed_count_(0),
+        images_changed_count_(0),
+        moved_count_(0) {
+  }
+
+  // Returns a string description of the changes that have occurred since this
+  // was last invoked. Resets state to initial state.
+  std::string StateStringAndClear() {
+    std::string result;
+    AddToResult("added=%d", added_count_, &result);
+    AddToResult("removed=%d", removed_count_, &result);
+    AddToResult("images_changed=%d", images_changed_count_, &result);
+    AddToResult("moved=%d", moved_count_, &result);
+    added_count_ = removed_count_ = images_changed_count_ = moved_count_ = 0;
+    return result;
+  }
+
+  // LauncherModelObserver overrides:
+  virtual void LauncherItemAdded(int index) OVERRIDE {
+    added_count_++;
+  }
+  virtual void LauncherItemRemoved(int index) OVERRIDE {
+    removed_count_++;
+  }
+  virtual void LauncherItemImagesChanged(int index) OVERRIDE {
+    images_changed_count_++;
+  }
+  virtual void LauncherItemMoved(int start_index, int target_index) OVERRIDE {
+    moved_count_++;
+  }
+  virtual void LauncherItemImagesWillChange(int index) OVERRIDE {
+  }
+
+ private:
+  void AddToResult(const std::string& format, int count, std::string* result) {
+    if (!count)
+      return;
+    if (!result->empty())
+      *result += " ";
+    *result += base::StringPrintf(format.c_str(), count);
+  }
+
+  int added_count_;
+  int removed_count_;
+  int images_changed_count_;
+  int moved_count_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestLauncherModelObserver);
+};
+
+}  // namespace
+
+TEST(LauncherModel, BasicAssertions) {
+  TestLauncherModelObserver observer;
+  LauncherModel model;
+  // Add an item.
+  model.AddObserver(&observer);
+  LauncherItem item;
+  model.Add(0, item);
+  EXPECT_EQ(1, model.item_count());
+  EXPECT_EQ("added=1",
+            observer.StateStringAndClear());
+
+  // Change a tabbed image.
+  model.SetTabbedImages(0, LauncherTabbedImages());
+  EXPECT_EQ("images_changed=1",
+            observer.StateStringAndClear());
+
+  // Remove the item.
+  model.RemoveItemAt(0);
+  EXPECT_EQ(0, model.item_count());
+  EXPECT_EQ("removed=1", observer.StateStringAndClear());
+
+  // Add an app item.
+  item.type = TYPE_APP;
+  item.user_data = reinterpret_cast<void*>(1);
+  model.Add(0, item);
+  observer.StateStringAndClear();
+
+  // Change an app image.
+  model.SetAppImage(0, SkBitmap());
+  EXPECT_EQ("images_changed=1", observer.StateStringAndClear());
+
+  // Add another item.
+  item.type = TYPE_APP;
+  item.user_data = reinterpret_cast<void*>(2);
+  model.Add(1, item);
+  observer.StateStringAndClear();
+
+  // Move the second item to be first.
+  model.Move(1, 0);
+  EXPECT_EQ("moved=1", observer.StateStringAndClear());
+  EXPECT_EQ(reinterpret_cast<void*>(2), model.items()[0].user_data);
+  EXPECT_EQ(reinterpret_cast<void*>(1), model.items()[1].user_data);
+
+  // Move the first item to the second item.
+  model.Move(0, 1);
+  EXPECT_EQ("moved=1", observer.StateStringAndClear());
+  EXPECT_EQ(reinterpret_cast<void*>(1), model.items()[0].user_data);
+  EXPECT_EQ(reinterpret_cast<void*>(2), model.items()[1].user_data);
+}
+
+}  // namespace aura_shell