Add app.window.setIcon

chrome/browser/extensions/api/app_window/app_window_apitest.cc

 // Copyright (c) 2012 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/string_number_conversions.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
 #include "chrome/browser/extensions/platform_app_browsertest_util.h"
+#include "chrome/browser/extensions/shell_window_registry.h"
+#include "chrome/browser/ui/base_window.h"
+#include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/extensions/native_app_window.h"
 #include "chrome/browser/ui/extensions/shell_window.h"
+#include "chrome/test/base/testing_profile.h"
 #include "ui/gfx/rect.h"
 
+namespace {
+
+class TestShellWindowRegistryObserver
+    : public extensions::ShellWindowRegistry::Observer {
+ public:
+  explicit TestShellWindowRegistryObserver(Profile* profile)
+      : profile_(profile),
+        icon_updates_(0) {
+    extensions::ShellWindowRegistry::Get(profile_)->AddObserver(this);
+  }
+  virtual ~TestShellWindowRegistryObserver() {
+    extensions::ShellWindowRegistry::Get(profile_)->RemoveObserver(this);
+  }
+
+  // Overridden from ShellWindowRegistry::Observer:
+  virtual void OnShellWindowAdded(ShellWindow* shell_window) {}
+  virtual void OnShellWindowIconChanged(ShellWindow* shell_window) {
+    ++icon_updates_;
+  }
+  virtual void OnShellWindowRemoved(ShellWindow* shell_window) {
+  }
+
+  int icon_updates() { return icon_updates_; }
+
+ private:
+  Profile* profile_;
+  int icon_updates_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestShellWindowRegistryObserver);
+};
+
+}  // namespace
+
 namespace extensions {
 
 IN_PROC_BROWSER_TEST_F(PlatformAppBrowserTest, WindowsApiBounds) {
   ExtensionTestMessageListener background_listener("background_ok", false);
   ExtensionTestMessageListener ready_listener("ready", true /* will_reply */);
   ExtensionTestMessageListener success_listener("success", false);
 
   LoadAndLaunchPlatformApp("windows_api_bounds");
   ASSERT_TRUE(background_listener.WaitUntilSatisfied());
   ASSERT_TRUE(ready_listener.WaitUntilSatisfied());
   ShellWindow* window = GetFirstShellWindow();
 
   gfx::Rect new_bounds(100, 200, 300, 400);
   window->GetBaseWindow()->SetBounds(new_bounds);
 
   // TODO(jeremya/asargent) figure out why in GTK the window doesn't end up
   // with exactly the bounds we set. Is it a bug in our shell window
   // implementation?  crbug.com/160252
 #ifdef TOOLKIT_GTK
   int slop = 50;
 #else
   int slop = 0;
 #endif  // !TOOLKIT_GTK
 
   ready_listener.Reply(base::IntToString(slop));
   ASSERT_TRUE(success_listener.WaitUntilSatisfied());
 }
 
+// Tests chrome.app.window.setIcon.
+IN_PROC_BROWSER_TEST_F(ExperimentalPlatformAppBrowserTest, WindowsApiSetIcon) {
+  scoped_ptr<TestShellWindowRegistryObserver> test_observer(
+      new TestShellWindowRegistryObserver(browser()->profile()));
+  ExtensionTestMessageListener listener("IconSet", false);
+  LoadAndLaunchPlatformApp("windows_api_set_icon");
+  EXPECT_EQ(0, test_observer->icon_updates());
+
+  ASSERT_TRUE(listener.WaitUntilSatisfied());
+
+  ShellWindow* shell_window = GetFirstShellWindow();
+  ASSERT_TRUE(shell_window);
+  EXPECT_NE(std::string::npos,
+            shell_window->app_icon_url().spec().find("icon.png"));
+  EXPECT_EQ(1, test_observer->icon_updates());
+}
+
 // TODO(asargent) - Fix onMinimzed event on OSX (crbug.com/162793) and figure
 // out what to do about the fact that minimize events don't work under ubuntu
 // unity (crbug.com/162794 and https://bugs.launchpad.net/unity/+bug/998073).
 #if defined(TOOLKIT_VIEWS)
 
 IN_PROC_BROWSER_TEST_F(PlatformAppBrowserTest, WindowsApiProperties) {
   EXPECT_TRUE(
       RunExtensionTest("platform_apps/windows_api_properties")) << message_;
 }
 
 #endif  // defined(TOOLKIT_VIEWS)
 
 }  // namespace extensions