| Index: ash/common/wm_window_user_data_unittest.cc
|
| diff --git a/ash/common/wm_window_user_data_unittest.cc b/ash/common/wm_window_user_data_unittest.cc
|
| index 3c65711cf52e924f5e155ab8cd21556bb4d4c302..64fb42c655e7dc6943f1f75b2ed2d1312b400579 100644
|
| --- a/ash/common/wm_window_user_data_unittest.cc
|
| +++ b/ash/common/wm_window_user_data_unittest.cc
|
| @@ -11,6 +11,8 @@
|
| #include "ash/common/wm_window.h"
|
| #include "ash/common/wm_window_user_data.h"
|
| #include "base/memory/ptr_util.h"
|
| +#include "ui/aura/window.h"
|
| +#include "ui/compositor/layer_type.h"
|
|
|
| namespace ash {
|
| namespace {
|
| @@ -35,31 +37,35 @@ using WmWindowUserDataTest = AshTest;
|
| // 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);
|
| + std::unique_ptr<aura::Window> window(
|
| + base::MakeUnique<aura::Window>(nullptr, ui::wm::WINDOW_TYPE_UNKNOWN));
|
| + window->Init(ui::LAYER_NOT_DRAWN);
|
| bool data_deleted = false;
|
| - user_data.Set(window, base::MakeUnique<Data>(&data_deleted));
|
| + user_data.Set(WmWindow::Get(window.get()),
|
| + 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);
|
| + std::unique_ptr<aura::Window> window(
|
| + base::MakeUnique<aura::Window>(nullptr, ui::wm::WINDOW_TYPE_UNKNOWN));
|
| + window->Init(ui::LAYER_NOT_DRAWN);
|
| bool data1_deleted = false;
|
| - user_data.Set(window, base::MakeUnique<Data>(&data1_deleted));
|
| + user_data.Set(WmWindow::Get(window.get()),
|
| + base::MakeUnique<Data>(&data1_deleted));
|
| EXPECT_FALSE(data1_deleted);
|
| bool data2_deleted = false;
|
| - user_data.Set(window, base::MakeUnique<Data>(&data2_deleted));
|
| + user_data.Set(WmWindow::Get(window.get()),
|
| + 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_EQ(WmWindow::Get(window.get()), *user_data.GetWindows().begin());
|
| + window.reset();
|
| EXPECT_TRUE(data2_deleted);
|
| EXPECT_TRUE(user_data.GetWindows().empty());
|
| }
|
| @@ -67,15 +73,16 @@ TEST_F(WmWindowUserDataTest, ReplaceDestroys) {
|
| // 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);
|
| + std::unique_ptr<aura::Window> window(
|
| + base::MakeUnique<aura::Window>(nullptr, ui::wm::WINDOW_TYPE_UNKNOWN));
|
| + window->Init(ui::LAYER_NOT_DRAWN);
|
| bool data1_deleted = false;
|
| - user_data.Set(window, base::MakeUnique<Data>(&data1_deleted));
|
| + user_data.Set(WmWindow::Get(window.get()),
|
| + base::MakeUnique<Data>(&data1_deleted));
|
| EXPECT_FALSE(data1_deleted);
|
| - user_data.Set(window, nullptr);
|
| + user_data.Set(WmWindow::Get(window.get()), nullptr);
|
| EXPECT_TRUE(data1_deleted);
|
| EXPECT_TRUE(user_data.GetWindows().empty());
|
| - window->Destroy();
|
| }
|
|
|
| } // namespace ash
|
|
|