xnu-7195.50.7.100.1.tar.gz

[apple/xnu.git] / tests / intrusive_shared_ptr_src / cast.const.cpp

diff --git a/tests/intrusive_shared_ptr_src/cast.const.cpp b/tests/intrusive_shared_ptr_src/cast.const.cpp
new file mode 100644 (file)
index 0000000..461ef13
--- /dev/null
+++ b/tests/intrusive_shared_ptr_src/cast.const.cpp
@@ -0,0 +1,67 @@
+//
+// Tests for
+//  template<typename To, typename From, typename R>
+//  intrusive_shared_ptr<To, R> const_pointer_cast(intrusive_shared_ptr<From, R> const& ptr) noexcept;
+//
+//  template<typename To, typename From, typename R>
+//  intrusive_shared_ptr<To, R> const_pointer_cast(intrusive_shared_ptr<From, R>&& ptr) noexcept
+//
+
+#include <libkern/c++/intrusive_shared_ptr.h>
+#include <utility>
+#include <darwintest.h>
+#include "test_policy.h"
+
+struct T { int i; };
+
+template <typename Stored, typename From, typename To>
+static void
+tests()
+{
+       Stored obj{3};
+
+       {
+               tracked_shared_ptr<From> const from(&obj, libkern::no_retain);
+               tracking_policy::reset();
+               tracked_shared_ptr<To> to = libkern::const_pointer_cast<To>(from);
+               CHECK(tracking_policy::retains == 1);
+               CHECK(tracking_policy::releases == 0);
+               CHECK(to.get() == const_cast<To*>(&obj));
+               CHECK(from.get() == &obj);
+       }
+       {
+               tracked_shared_ptr<From> from(&obj, libkern::no_retain);
+               tracking_policy::reset();
+               tracked_shared_ptr<To> to = libkern::const_pointer_cast<To>(std::move(from));
+               CHECK(tracking_policy::retains == 0);
+               CHECK(tracking_policy::releases == 0);
+               CHECK(to.get() == const_cast<To*>(&obj));
+               CHECK(from.get() == nullptr);
+       }
+
+       // Test `const_pointer_cast`ing a null pointer
+       {
+               tracked_shared_ptr<From> const from = nullptr;
+               tracking_policy::reset();
+               tracked_shared_ptr<To> to = libkern::const_pointer_cast<To>(from);
+               CHECK(tracking_policy::retains == 0);
+               CHECK(tracking_policy::releases == 0);
+               CHECK(to.get() == nullptr);
+               CHECK(from.get() == nullptr);
+       }
+       {
+               tracked_shared_ptr<From> from = nullptr;
+               tracking_policy::reset();
+               tracked_shared_ptr<To> to = libkern::const_pointer_cast<To>(std::move(from));
+               CHECK(tracking_policy::retains == 0);
+               CHECK(tracking_policy::releases == 0);
+               CHECK(to.get() == nullptr);
+               CHECK(from.get() == nullptr);
+       }
+}
+
+T_DECL(cast_const, "intrusive_shared_ptr.cast.const") {
+       tests</*stored*/ T, /*from*/ T, /*to*/ T>();
+       tests</*stored*/ T, /*from*/ T, /*to*/ T const>();
+       tests</*stored*/ T, /*from*/ T const, /*to*/ T>();
+}