xnu-7195.101.1.tar.gz

[apple/xnu.git] / tests / intrusive_shared_ptr_src / reset.cpp

//
// Tests for
//  void reset() noexcept;
//

#include <libkern/c++/intrusive_shared_ptr.h>
#include <darwintest.h>
#include "test_policy.h"

struct T {
	int i;
};

template <typename T>
static void
tests()
{
	T obj{3};

	// reset() on a non-null shared pointer
	{
		tracked_shared_ptr<T> ptr(&obj, libkern::retain);
		tracking_policy::reset();
		ptr.reset();
		CHECK(tracking_policy::releases == 1);
		CHECK(tracking_policy::retains == 0);
		CHECK(ptr.get() == nullptr);
	}

	// reset() on a null shared pointer
	{
		tracked_shared_ptr<T> ptr = nullptr;
		tracking_policy::reset();
		ptr.reset();
		CHECK(tracking_policy::releases == 0);
		CHECK(tracking_policy::retains == 0);
		CHECK(ptr.get() == nullptr);
	}

	// reset() as a self-reference
	{
		tracked_shared_ptr<T> ptr(&obj, libkern::retain);
		tracked_shared_ptr<T> ptr2(&obj, libkern::retain);
		CHECK(!ptr.reset());

		CHECK(&ptr.reset() == &ptr);

		// check short-circuiting
		bool ok =  (ptr.reset() && !ptr2.reset());
		CHECK(ptr2.get() != nullptr);
	}
}

T_DECL(reset, "intrusive_shared_ptr.reset") {
	tests<T>();
	tests<T const>();
}