1 files changed, 142 insertions, 0 deletions
diff --git a/core/fxcrt/unowned_ptr_unittest.cpp b/core/fxcrt/unowned_ptr_unittest.cpp
new file mode 100644
index 0000000000..3fc5308e46
--- /dev/null
+++ b/core/fxcrt/unowned_ptr_unittest.cpp
@@ -0,0 +1,142 @@
+// Copyright 2017 PDFium 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 "core/fxcrt/unowned_ptr.h"
+
+#include <utility>
+#include <vector>
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace fxcrt {
+namespace {
+
+class Clink {
+ public:
+  UnownedPtr<Clink> next_ = nullptr;
+};
+
+void DeleteDangling() {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  delete ptr1;
+  delete ptr2;
+}
+
+void AssignDangling() {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  delete ptr1;
+  ptr2->next_ = nullptr;
+  delete ptr2;
+}
+
+void ReleaseDangling() {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  delete ptr1;
+  ptr2->next_.Release();
+  delete ptr2;
+}
+
+}  // namespace
+
+TEST(UnownedPtr, PtrOk) {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  delete ptr2;
+  delete ptr1;
+}
+
+TEST(UnownedPtr, PtrNotOk) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+  EXPECT_DEATH(DeleteDangling(), "");
+#else
+  DeleteDangling();
+#endif
+}
+
+TEST(UnownedPtr, AssignOk) {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  ptr2->next_ = nullptr;
+  delete ptr2;
+  delete ptr1;
+}
+
+TEST(UnownedPtr, AssignNotOk) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+  EXPECT_DEATH(AssignDangling(), "");
+#else
+  AssignDangling();
+#endif
+}
+
+TEST(UnownedPtr, ReleaseOk) {
+  Clink* ptr1 = new Clink();
+  Clink* ptr2 = new Clink();
+  ptr2->next_ = ptr1;
+  ptr2->next_.Release();
+  delete ptr1;
+  delete ptr2;
+}
+
+TEST(UnownedPtr, ReleaseNotOk) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+  EXPECT_DEATH(ReleaseDangling(), "");
+#else
+  ReleaseDangling();
+#endif
+}
+
+TEST(UnownedPtr, OperatorEQ) {
+  int foo;
+  UnownedPtr<int> ptr1;
+  EXPECT_TRUE(ptr1 == ptr1);
+
+  UnownedPtr<int> ptr2;
+  EXPECT_TRUE(ptr1 == ptr2);
+
+  UnownedPtr<int> ptr3(&foo);
+  EXPECT_TRUE(&foo == ptr3);
+  EXPECT_TRUE(ptr3 == &foo);
+  EXPECT_FALSE(ptr1 == ptr3);
+
+  ptr1 = &foo;
+  EXPECT_TRUE(ptr1 == ptr3);
+}
+
+TEST(UnownedPtr, OperatorNE) {
+  int foo;
+  UnownedPtr<int> ptr1;
+  EXPECT_FALSE(ptr1 != ptr1);
+
+  UnownedPtr<int> ptr2;
+  EXPECT_FALSE(ptr1 != ptr2);
+
+  UnownedPtr<int> ptr3(&foo);
+  EXPECT_FALSE(&foo != ptr3);
+  EXPECT_FALSE(ptr3 != &foo);
+  EXPECT_TRUE(ptr1 != ptr3);
+
+  ptr1 = &foo;
+  EXPECT_FALSE(ptr1 != ptr3);
+}
+
+TEST(UnownedPtr, OperatorLT) {
+  int foos[2];
+  UnownedPtr<int> ptr1(&foos[0]);
+  UnownedPtr<int> ptr2(&foos[1]);
+
+  EXPECT_FALSE(ptr1 < ptr1);
+  EXPECT_TRUE(ptr1 < ptr2);
+  EXPECT_FALSE(ptr2 < ptr1);
+}
+
+}  // namespace fxcrt