supports_user_data_unittest.cc 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. // Copyright 2014 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 "base/supports_user_data.h"
  5. #include "base/memory/ptr_util.h"
  6. #include "base/memory/raw_ptr.h"
  7. #include "testing/gtest/include/gtest/gtest.h"
  8. namespace base {
  9. namespace {
  10. struct TestSupportsUserData : public SupportsUserData {
  11. // Make ClearAllUserData public so tests can access it.
  12. using SupportsUserData::ClearAllUserData;
  13. };
  14. struct UsesItself : public SupportsUserData::Data {
  15. UsesItself(SupportsUserData* supports_user_data, const void* key)
  16. : supports_user_data_(supports_user_data),
  17. key_(key) {
  18. }
  19. ~UsesItself() override {
  20. EXPECT_EQ(nullptr, supports_user_data_->GetUserData(key_));
  21. }
  22. raw_ptr<SupportsUserData> supports_user_data_;
  23. raw_ptr<const void> key_;
  24. };
  25. TEST(SupportsUserDataTest, ClearWorksRecursively) {
  26. TestSupportsUserData supports_user_data;
  27. char key = 0;
  28. supports_user_data.SetUserData(
  29. &key, std::make_unique<UsesItself>(&supports_user_data, &key));
  30. // Destruction of supports_user_data runs the actual test.
  31. }
  32. struct TestData : public SupportsUserData::Data {};
  33. TEST(SupportsUserDataTest, Movable) {
  34. TestSupportsUserData supports_user_data_1;
  35. char key1 = 0;
  36. supports_user_data_1.SetUserData(&key1, std::make_unique<TestData>());
  37. void* data_1_ptr = supports_user_data_1.GetUserData(&key1);
  38. TestSupportsUserData supports_user_data_2;
  39. char key2 = 0;
  40. supports_user_data_2.SetUserData(&key2, std::make_unique<TestData>());
  41. supports_user_data_2 = std::move(supports_user_data_1);
  42. EXPECT_EQ(data_1_ptr, supports_user_data_2.GetUserData(&key1));
  43. EXPECT_EQ(nullptr, supports_user_data_2.GetUserData(&key2));
  44. }
  45. TEST(SupportsUserDataTest, ClearAllUserData) {
  46. TestSupportsUserData supports_user_data;
  47. char key1 = 0;
  48. supports_user_data.SetUserData(&key1, std::make_unique<TestData>());
  49. char key2 = 0;
  50. supports_user_data.SetUserData(&key2, std::make_unique<TestData>());
  51. EXPECT_TRUE(supports_user_data.GetUserData(&key1));
  52. EXPECT_TRUE(supports_user_data.GetUserData(&key2));
  53. supports_user_data.ClearAllUserData();
  54. EXPECT_FALSE(supports_user_data.GetUserData(&key1));
  55. EXPECT_FALSE(supports_user_data.GetUserData(&key2));
  56. }
  57. } // namespace
  58. } // namespace base