Reland r265807: "Moves the notification icon out of th..."

chrome/browser/ui/views/status_icons/status_tray_state_changer_interactive_uitest_win.cc

Index: chrome/browser/ui/views/status_icons/status_tray_state_changer_interactive_uitest_win.cc
diff --git a/chrome/browser/ui/views/status_icons/status_tray_state_changer_interactive_uitest_win.cc b/chrome/browser/ui/views/status_icons/status_tray_state_changer_interactive_uitest_win.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b4119796822f6658825b5dc9f7925acf0b60658d
--- /dev/null
+++ b/chrome/browser/ui/views/status_icons/status_tray_state_changer_interactive_uitest_win.cc
@@ -0,0 +1,162 @@
+// Copyright 2014 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 "chrome/browser/ui/views/status_icons/status_tray_state_changer_win.h"
+
+#include "base/memory/scoped_ptr.h"
+#include "base/win/scoped_com_initializer.h"
+#include "base/win/windows_version.h"
+#include "chrome/browser/status_icons/status_icon.h"
+#include "chrome/browser/ui/views/status_icons/status_icon_win.h"
+#include "chrome/browser/ui/views/status_icons/status_tray_win.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/gfx/image/image_skia.h"
+
+class StatusTrayStateChangerWinTest : public testing::Test {
+ public:
+  StatusTrayStateChangerWinTest() {}
+
+  void SetUp() OVERRIDE {
+    testing::Test::SetUp();
+    com_.reset(new base::win::ScopedCOMInitializer());
+    status_tray_.reset(new StatusTrayWin());
+    SkBitmap bitmap;
+
+    // Put a real bitmap into "bitmap".  2x2 bitmap of green 32 bit pixels.
+    bitmap.setConfig(SkBitmap::kARGB_8888_Config, 16, 16);
+    bitmap.allocPixels();
+    bitmap.eraseColor(SK_ColorGREEN);
+    status_icon_win_ = (StatusIconWin*)status_tray_->CreateStatusIcon(
+        StatusTray::OTHER_ICON,
+        gfx::ImageSkia::CreateFrom1xBitmap(bitmap),
+        base::string16());
+    tray_watcher_ = new StatusTrayStateChangerWin(status_icon_win_->icon_id(),
+                                                  status_icon_win_->window());
+  }
+
+  void TearDown() OVERRIDE {
+    tray_watcher_ = NULL;
+    status_tray_.reset();
+    com_.reset();
+    testing::Test::TearDown();
+  }
+
+ protected:
+  HWND icon_window() { return status_icon_win_->window(); }
+  UINT icon_id() { return status_icon_win_->icon_id(); }
+
+  scoped_ptr<NOTIFYITEM> SetupAndGetCurrentNotifyItem() {
+    EXPECT_TRUE(CallCreateTrayNotify());
+
+    EXPECT_TRUE(IsInterfaceKnown());
+
+    scoped_ptr<NOTIFYITEM> notify_item = GetNotifyItem();
+    EXPECT_TRUE(notify_item.get() != NULL);
+    DCHECK_EQ(notify_item->hwnd, icon_window());
+    DCHECK_EQ(notify_item->id, icon_id());
+
+    return notify_item.Pass();
+  }
+
+  bool CallCreateTrayNotify() { return tray_watcher_->CreateTrayNotify(); }
+
+  bool IsInterfaceKnown() {
+    return StatusTrayStateChangerWin::INTERFACE_VERSION_UNKNOWN !=
+           tray_watcher_->interface_version_;
+  }
+
+  void SendNotifyItemUpdate(scoped_ptr<NOTIFYITEM> notify_item) {
+    tray_watcher_->SendNotifyItemUpdate(notify_item.Pass());
+  }
+
+  scoped_ptr<NOTIFYITEM> GetNotifyItem() {
+    return tray_watcher_->RegisterCallback();
+  }
+
+  scoped_ptr<base::win::ScopedCOMInitializer> com_;
+  scoped_ptr<StatusTrayWin> status_tray_;
+  scoped_refptr<StatusTrayStateChangerWin> tray_watcher_;
+
+  StatusIconWin* status_icon_win_;
+
+  DISALLOW_COPY_AND_ASSIGN(StatusTrayStateChangerWinTest);
+};
+
+TEST_F(StatusTrayStateChangerWinTest, Setup) {
+  // This tests the code path that will read the NOTIFYITEM data structure for
+  // use in future tests.
+  scoped_ptr<NOTIFYITEM> notify_item = SetupAndGetCurrentNotifyItem();
+  EXPECT_FALSE(notify_item.get() == NULL);
+}
+
+TEST_F(StatusTrayStateChangerWinTest, ComApiTest) {
+
+  // Setup code to read the current preference.
+  scoped_ptr<NOTIFYITEM> notify_item = SetupAndGetCurrentNotifyItem();
+  ASSERT_TRUE(notify_item.get() != NULL);
+
+  // Store the current pref.
+  DWORD current_preference = notify_item->preference;
+
+  // Ensure that running our code will do something.
+  if (notify_item->preference != PREFERENCE_SHOW_WHEN_ACTIVE) {
+    scoped_ptr<NOTIFYITEM> notify_item_copy(new NOTIFYITEM(*notify_item));
+    notify_item_copy->preference = PREFERENCE_SHOW_WHEN_ACTIVE;
+    SendNotifyItemUpdate(notify_item_copy.Pass());
+  }
+
+  // Run the interesting code.
+  tray_watcher_->EnsureTrayIconVisible();
+
+  EXPECT_EQ(PREFERENCE_SHOW_ALWAYS, GetNotifyItem()->preference);
+  SendNotifyItemUpdate(notify_item.Pass());
+  notify_item = GetNotifyItem();
+
+  EXPECT_EQ(notify_item->preference, current_preference);
+};
+
+TEST_F(StatusTrayStateChangerWinTest, TraySizeApiTest) {
+
+  // The tray does not auto-hide icons immediately on Vista so this test does
+  // not detect a size change.
+  if (base::win::GetVersion() <= base::win::VERSION_VISTA)
+    return;
+
+  // Used to reset operating system state afterwards.
+  scoped_ptr<NOTIFYITEM> notify_item = SetupAndGetCurrentNotifyItem();
+  // We can't actually run this test if we're already showing the icon.
+  if (notify_item->preference == PREFERENCE_SHOW_ALWAYS)
+    return;
+
+  // This test can only run if the tray window structure conforms to what I've
+  // seen in Win7 and Win8.
+  HWND shell_tray_hwnd = ::FindWindow(L"Shell_TrayWnd", NULL);
+  if (shell_tray_hwnd == NULL)
+    return;
+
+  HWND tray_notify_hwnd =
+      ::FindWindowEx(shell_tray_hwnd, NULL, L"TrayNotifyWnd", NULL);
+  ASSERT_TRUE(tray_notify_hwnd != NULL);
+
+  RECT original_tray_notify_rect;
+  ::GetWindowRect(tray_notify_hwnd, &original_tray_notify_rect);
+
+  LONG width = original_tray_notify_rect.right - original_tray_notify_rect.left;
+  ASSERT_GT(width, 0);
+
+  tray_watcher_->EnsureTrayIconVisible();
+
+  RECT new_tray_notify_rect;
+  ::GetWindowRect(tray_notify_hwnd, &new_tray_notify_rect);
+
+  LONG new_width = new_tray_notify_rect.right - new_tray_notify_rect.left;
+
+  EXPECT_GT(new_width, width);
+
+  SendNotifyItemUpdate(notify_item.Pass());
+  ::GetWindowRect(tray_notify_hwnd, &new_tray_notify_rect);
+  new_width = new_tray_notify_rect.right - new_tray_notify_rect.left;
+  EXPECT_EQ(width, new_width);
+}