window_user_data_unittest.cc 2.5 KB

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