1 files changed, 135 insertions, 0 deletions
diff --git a/chrome/browser/extensions/extension_icon_manager_unittest.cc b/chrome/browser/extensions/extension_icon_manager_unittest.cc
new file mode 100644
index 0000000..f1d6dca
--- /dev/null
+++ b/chrome/browser/extensions/extension_icon_manager_unittest.cc
@@ -0,0 +1,135 @@
+// Copyright (c) 2010 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 "base/message_loop.h"
+#include "base/path_service.h"
+#include "base/values.h"
+#include "chrome/browser/chrome_thread.h"
+#include "chrome/browser/extensions/extension_icon_manager.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/common/extensions/extension_resource.h"
+#include "chrome/common/json_value_serializer.h"
+#include "gfx/skia_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+// Our test class that takes care of managing the necessary threads for loading
+// extension icons, and waiting for those loads to happen.
+class ExtensionIconManagerTest : public testing::Test {
+ public:
+  ExtensionIconManagerTest() :
+      unwaited_image_loads_(0),
+      waiting_(false),
+      ui_thread_(ChromeThread::UI, &ui_loop_),
+      file_thread_(ChromeThread::FILE),
+      io_thread_(ChromeThread::IO) {}
+
+  virtual ~ExtensionIconManagerTest() {}
+
+  void ImageLoadObserved() {
+    unwaited_image_loads_++;
+    if (waiting_) {
+      MessageLoop::current()->Quit();
+    }
+  }
+
+  void WaitForImageLoad() {
+    if (unwaited_image_loads_ == 0) {
+      waiting_ = true;
+      MessageLoop::current()->Run();
+      waiting_ = false;
+    }
+    ASSERT_GT(unwaited_image_loads_, 0);
+    unwaited_image_loads_--;
+  }
+
+ private:
+  virtual void SetUp() {
+    file_thread_.Start();
+    io_thread_.Start();
+  }
+
+  // The number of observed image loads that have not been waited for.
+  int unwaited_image_loads_;
+
+  // Whether we are currently waiting for an image load.
+  bool waiting_;
+
+  MessageLoop ui_loop_;
+  ChromeThread ui_thread_;
+  ChromeThread file_thread_;
+  ChromeThread io_thread_;
+
+  DISALLOW_COPY_AND_ASSIGN(ExtensionIconManagerTest);
+};
+
+// This is a specialization of ExtensionIconManager, with a special override to
+// call back to the test when an icon has completed loading.
+class TestIconManager : public ExtensionIconManager {
+ public:
+  explicit TestIconManager(ExtensionIconManagerTest* test) : test_(test) {}
+  virtual ~TestIconManager() {}
+
+  // Implements the ImageLoadingTracker::Observer interface, and calls through
+  // to the base class' implementation. Then it lets the test know that an
+  // image load was observed.
+  virtual void OnImageLoaded(SkBitmap* image, ExtensionResource resource,
+                             int index) {
+    ExtensionIconManager::OnImageLoaded(image, resource, index);
+    test_->ImageLoadObserved();
+  }
+
+ private:
+  ExtensionIconManagerTest* test_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestIconManager);
+};
+
+// Returns the default icon that ExtensionIconManager gives when an extension
+// doesn't have an icon.
+SkBitmap GetDefaultIcon() {
+  std::string dummy_id;
+  EXPECT_TRUE(Extension::GenerateId(std::string("whatever"), &dummy_id));
+  ExtensionIconManager manager;
+  return manager.GetIcon(dummy_id);
+}
+
+// Tests loading an icon for an extension, removing it, then re-loading it.
+TEST_F(ExtensionIconManagerTest, LoadRemoveLoad) {
+  SkBitmap default_icon = GetDefaultIcon();
+
+  FilePath test_dir;
+  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_dir));
+  FilePath manifest_path = test_dir.AppendASCII(
+      "extensions/image_loading_tracker/app.json");
+
+  JSONFileValueSerializer serializer(manifest_path);
+  scoped_ptr<DictionaryValue> manifest(
+      static_cast<DictionaryValue*>(serializer.Deserialize(NULL, NULL)));
+  ASSERT_TRUE(manifest.get() != NULL);
+
+  Extension extension(manifest_path.DirName());
+  ASSERT_TRUE(extension.InitFromValue(*manifest.get(),
+                                      false /* require_key */,
+                                      NULL /* errors */));
+  TestIconManager icon_manager(this);
+
+  // Load the icon and grab the bitmap.
+  icon_manager.LoadIcon(&extension);
+  WaitForImageLoad();
+  SkBitmap first_icon = icon_manager.GetIcon(extension.id());
+  EXPECT_FALSE(gfx::BitmapsAreEqual(first_icon, default_icon));
+
+  // Remove the icon from the manager.
+  icon_manager.RemoveIcon(extension.id());
+
+  // Now re-load the icon - we should get the same result bitmap (and not the
+  // default icon).
+  icon_manager.LoadIcon(&extension);
+  WaitForImageLoad();
+  SkBitmap second_icon = icon_manager.GetIcon(extension.id());
+  EXPECT_FALSE(gfx::BitmapsAreEqual(second_icon, default_icon));
+
+  EXPECT_TRUE(gfx::BitmapsAreEqual(first_icon, second_icon));
+}