1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/window_user_data.h"
#include <memory>
#include "ash/public/cpp/autotest_private_api_utils.h"
#include "ash/test/ash_test_base.h"
#include "base/memory/raw_ptr.h"
#include "ui/aura/window.h"
#include "ui/compositor/layer_type.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(const Data&) = delete;
Data& operator=(const Data&) = delete;
~Data() { *delete_setter_ = true; }
private:
raw_ptr<bool, ExperimentalAsh> delete_setter_;
};
} // namespace
using WindowUserDataTest = AshTestBase;
// Verifies clear() deletes the data associated with a window.
TEST_F(WindowUserDataTest, ClearDestroys) {
WindowUserData<Data> user_data;
std::unique_ptr<aura::Window> window =
std::make_unique<aura::Window>(nullptr);
window->Init(ui::LAYER_NOT_DRAWN);
bool data_deleted = false;
user_data.Set(window.get(), std::make_unique<Data>(&data_deleted));
EXPECT_FALSE(data_deleted);
user_data.clear();
EXPECT_TRUE(data_deleted);
}
// Verifies Set() called with an existing window replaces the existing data.
TEST_F(WindowUserDataTest, ReplaceDestroys) {
WindowUserData<Data> user_data;
std::unique_ptr<aura::Window> window =
std::make_unique<aura::Window>(nullptr);
window->Init(ui::LAYER_NOT_DRAWN);
bool data1_deleted = false;
user_data.Set(window.get(), std::make_unique<Data>(&data1_deleted));
EXPECT_FALSE(data1_deleted);
bool data2_deleted = false;
user_data.Set(window.get(), std::make_unique<Data>(&data2_deleted));
EXPECT_TRUE(data1_deleted);
EXPECT_FALSE(data2_deleted);
ASSERT_EQ(1u, user_data.GetWindows().size());
EXPECT_EQ(window.get(), *user_data.GetWindows().begin());
window.reset();
EXPECT_TRUE(data2_deleted);
EXPECT_TRUE(user_data.GetWindows().empty());
}
// Verifies Set() with null deletes existing data.
TEST_F(WindowUserDataTest, NullClears) {
WindowUserData<Data> user_data;
std::unique_ptr<aura::Window> window =
std::make_unique<aura::Window>(nullptr);
window->Init(ui::LAYER_NOT_DRAWN);
bool data1_deleted = false;
user_data.Set(window.get(), std::make_unique<Data>(&data1_deleted));
EXPECT_FALSE(data1_deleted);
user_data.Set(window.get(), nullptr);
EXPECT_TRUE(data1_deleted);
EXPECT_TRUE(user_data.GetWindows().empty());
}
} // namespace ash
|
