xnu-7195.50.7.100.1.tar.gz

[apple/xnu.git] / tests / intrusive_shared_ptr_src / ctor.ptr.retain.cpp

diff --git a/tests/intrusive_shared_ptr_src/ctor.ptr.retain.cpp b/tests/intrusive_shared_ptr_src/ctor.ptr.retain.cpp
new file mode 100644 (file)
index 0000000..b666378
--- /dev/null
+++ b/tests/intrusive_shared_ptr_src/ctor.ptr.retain.cpp
@@ -0,0 +1,36 @@
+//
+// Tests for
+//  explicit intrusive_shared_ptr(pointer p, retain_t);
+//
+
+#include <libkern/c++/intrusive_shared_ptr.h>
+#include <darwintest.h>
+#include <darwintest_utils.h>
+#include "test_policy.h"
+
+struct T { int i; };
+
+template <typename T, typename TQual>
+static void
+tests()
+{
+       T obj{0};
+
+       {
+               test_policy::retain_count = 0;
+               libkern::intrusive_shared_ptr<TQual, test_policy> ptr(&obj, libkern::retain);
+               CHECK(ptr.get() == &obj);
+               CHECK(test_policy::retain_count == 1);
+       }
+       {
+               test_policy::retain_count = 0;
+               libkern::intrusive_shared_ptr<TQual, test_policy> ptr{&obj, libkern::retain};
+               CHECK(ptr.get() == &obj);
+               CHECK(test_policy::retain_count == 1);
+       }
+}
+
+T_DECL(ctor_ptr_retain, "intrusive_shared_ptr.ctor.ptr.retain") {
+       tests<T, T>();
+       tests<T, T const>();
+}