Index: ui/base/class_property_unittest.cc |
diff --git a/ui/base/class_property_unittest.cc b/ui/base/class_property_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..31e85efb709eb59dcb6b26d7a27fb79d877c06f7 |
--- /dev/null |
+++ b/ui/base/class_property_unittest.cc |
@@ -0,0 +1,116 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "ui/base/class_property.h" |
+ |
+#include <limits.h> |
+ |
+#include <string> |
+#include <utility> |
+#include <vector> |
+ |
+#include "base/compiler_specific.h" |
+#include "base/macros.h" |
+#include "base/memory/ptr_util.h" |
+#include "build/build_config.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+DECLARE_UI_CLASS_PROPERTY_TYPE(const char*) |
+DECLARE_UI_CLASS_PROPERTY_TYPE(int) |
+ |
+namespace { |
+ |
+class TestProperty { |
+ public: |
+ TestProperty() {} |
+ ~TestProperty() { |
+ last_deleted_ = this; |
+ } |
+ static void* last_deleted() { return last_deleted_; } |
+ |
+ private: |
+ static void* last_deleted_; |
+ DISALLOW_COPY_AND_ASSIGN(TestProperty); |
+}; |
+ |
+void* TestProperty::last_deleted_ = nullptr; |
+ |
+DEFINE_OWNED_UI_CLASS_PROPERTY_KEY(TestProperty, kOwnedKey, NULL); |
+ |
+} // namespace |
+ |
+DECLARE_UI_CLASS_PROPERTY_TYPE(TestProperty*); |
+ |
+namespace ui { |
+namespace test { |
+ |
+namespace { |
+ |
+const int kDefaultIntValue = -2; |
+const char* kDefaultStringValue = "squeamish"; |
+const char* kTestStringValue = "ossifrage"; |
+ |
+DEFINE_UI_CLASS_PROPERTY_KEY(int, kIntKey, kDefaultIntValue); |
+DEFINE_UI_CLASS_PROPERTY_KEY(const char*, kStringKey, kDefaultStringValue); |
+} |
+ |
+TEST(PropertyTest, Property) { |
+ PropertyHandler h; |
+ |
+ // Non-existent properties should return the default values. |
+ EXPECT_EQ(kDefaultIntValue, h.GetProperty(kIntKey)); |
+ EXPECT_EQ(std::string(kDefaultStringValue), h.GetProperty(kStringKey)); |
+ |
+ // A set property value should be returned again (even if it's the default |
+ // value). |
+ h.SetProperty(kIntKey, INT_MAX); |
+ EXPECT_EQ(INT_MAX, h.GetProperty(kIntKey)); |
+ h.SetProperty(kIntKey, kDefaultIntValue); |
+ EXPECT_EQ(kDefaultIntValue, h.GetProperty(kIntKey)); |
+ h.SetProperty(kIntKey, INT_MIN); |
+ EXPECT_EQ(INT_MIN, h.GetProperty(kIntKey)); |
+ |
+ h.SetProperty<const char*>(kStringKey, nullptr); |
+ EXPECT_EQ(NULL, h.GetProperty(kStringKey)); |
+ h.SetProperty(kStringKey, kDefaultStringValue); |
+ EXPECT_EQ(std::string(kDefaultStringValue), h.GetProperty(kStringKey)); |
+ h.SetProperty(kStringKey, kTestStringValue); |
+ EXPECT_EQ(std::string(kTestStringValue), h.GetProperty(kStringKey)); |
+ |
+ // ClearProperty should restore the default value. |
+ h.ClearProperty(kIntKey); |
+ EXPECT_EQ(kDefaultIntValue, h.GetProperty(kIntKey)); |
+ h.ClearProperty(kStringKey); |
+ EXPECT_EQ(std::string(kDefaultStringValue), h.GetProperty(kStringKey)); |
+} |
+ |
+TEST(PropertyTest, OwnedProperty) { |
+ std::unique_ptr<PropertyHandler> h = base::MakeUnique<PropertyHandler>(); |
+ |
+ EXPECT_EQ(NULL, h->GetProperty(kOwnedKey)); |
+ void* last_deleted = TestProperty::last_deleted(); |
+ TestProperty* p1 = new TestProperty(); |
+ h->SetProperty(kOwnedKey, p1); |
+ EXPECT_EQ(p1, h->GetProperty(kOwnedKey)); |
+ EXPECT_EQ(last_deleted, TestProperty::last_deleted()); |
+ |
+ TestProperty* p2 = new TestProperty(); |
+ h->SetProperty(kOwnedKey, p2); |
+ EXPECT_EQ(p2, h->GetProperty(kOwnedKey)); |
+ EXPECT_EQ(p1, TestProperty::last_deleted()); |
+ |
+ h->ClearProperty(kOwnedKey); |
+ EXPECT_EQ(NULL, h->GetProperty(kOwnedKey)); |
+ EXPECT_EQ(p2, TestProperty::last_deleted()); |
+ |
+ TestProperty* p3 = new TestProperty(); |
+ h->SetProperty(kOwnedKey, p3); |
+ EXPECT_EQ(p3, h->GetProperty(kOwnedKey)); |
+ EXPECT_EQ(p2, TestProperty::last_deleted()); |
+ h.reset(); |
+ EXPECT_EQ(p3, TestProperty::last_deleted()); |
+} |
+ |
+} // namespace test |
+} // namespace ui |