Refactors DimWindow and moves to ash/common

ash/common/wm_window_user_data_unittest.cc

+// Copyright 2016 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/common/wm_window_user_data.h"
+
+#include <memory>
+
+#include "ash/common/wm_shell.h"
+#include "ash/common/wm_window.h"
+#include "ash/common/wm_window_user_data.h"
+#include "ash/test/ash_test_base.h"
+#include "base/memory/ptr_util.h"
+
+namespace ash {
+namespace {
+
+// Class that sets a bool* to true from the destructor. Used to track
+// destruction.
+class Data {
+ public:
+  explicit Data(bool* delete_setter) : delete_setter_(delete_setter) {}
+  ~Data() { *delete_setter_ = true; }
+
+ private:
+  bool* delete_setter_;
+
+  DISALLOW_COPY_AND_ASSIGN(Data);
+};
+
+}  // namespace
+
+using WmWindowUserDataTest = test::AshTestBase;
+
+// Verifies clear() deletes the data associated with a window.
+TEST_F(WmWindowUserDataTest, ClearDestroys) {
+  WmWindowUserData<Data> user_data;
+  WmWindow* window = WmShell::Get()->NewWindow(ui::wm::WINDOW_TYPE_UNKNOWN,
+                                               ui::LAYER_NOT_DRAWN);
+  bool data_deleted = false;
+  user_data.Set(window, base::MakeUnique<Data>(&data_deleted));
+  EXPECT_FALSE(data_deleted);
+  user_data.clear();
+  EXPECT_TRUE(data_deleted);
+  window->Destroy();
+}
+
+// Verifies Set() called with an existing window replaces the existing data.
+TEST_F(WmWindowUserDataTest, ReplaceDestroys) {
+  WmWindowUserData<Data> user_data;
+  WmWindow* window = WmShell::Get()->NewWindow(ui::wm::WINDOW_TYPE_UNKNOWN,
+                                               ui::LAYER_NOT_DRAWN);
+  bool data1_deleted = false;
+  user_data.Set(window, base::MakeUnique<Data>(&data1_deleted));
+  EXPECT_FALSE(data1_deleted);
+  bool data2_deleted = false;
+  user_data.Set(window, base::MakeUnique<Data>(&data2_deleted));
+  EXPECT_TRUE(data1_deleted);
+  EXPECT_FALSE(data2_deleted);
+  ASSERT_EQ(1u, user_data.GetWindows().size());
+  EXPECT_EQ(window, *user_data.GetWindows().begin());
+  window->Destroy();
+  EXPECT_TRUE(data2_deleted);
+  EXPECT_TRUE(user_data.GetWindows().empty());
+}
+
+// Verifies Set() with null deletes existing data.
+TEST_F(WmWindowUserDataTest, NullClears) {
+  WmWindowUserData<Data> user_data;
+  WmWindow* window = WmShell::Get()->NewWindow(ui::wm::WINDOW_TYPE_UNKNOWN,
+                                               ui::LAYER_NOT_DRAWN);
+  bool data1_deleted = false;
+  user_data.Set(window, base::MakeUnique<Data>(&data1_deleted));
+  EXPECT_FALSE(data1_deleted);
+  user_data.Set(window, nullptr);
+  EXPECT_TRUE(data1_deleted);
+  EXPECT_TRUE(user_data.GetWindows().empty());
+  window->Destroy();
+}
+
+}  // namespace ash