[go: nahoru, domu]

blob: 5055c66b63123d310f410cdd5f1b7936bc07ba02 [file] [log] [blame]
// 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