xref: /xnu-11417.121.6/tests/intrusive_shared_ptr_src/reset.cpp (revision a1e26a70f38d1d7daa7b49b258e2f8538ad81650)
1*a1e26a70SApple OSS Distributions //
2*a1e26a70SApple OSS Distributions // Tests for
3*a1e26a70SApple OSS Distributions //  void reset() noexcept;
4*a1e26a70SApple OSS Distributions //
5*a1e26a70SApple OSS Distributions 
6*a1e26a70SApple OSS Distributions #include <libkern/c++/intrusive_shared_ptr.h>
7*a1e26a70SApple OSS Distributions #include <darwintest.h>
8*a1e26a70SApple OSS Distributions #include "test_policy.h"
9*a1e26a70SApple OSS Distributions 
10*a1e26a70SApple OSS Distributions struct T {
11*a1e26a70SApple OSS Distributions 	int i;
12*a1e26a70SApple OSS Distributions };
13*a1e26a70SApple OSS Distributions 
14*a1e26a70SApple OSS Distributions template <typename T>
15*a1e26a70SApple OSS Distributions static void
tests()16*a1e26a70SApple OSS Distributions tests()
17*a1e26a70SApple OSS Distributions {
18*a1e26a70SApple OSS Distributions 	T obj{3};
19*a1e26a70SApple OSS Distributions 
20*a1e26a70SApple OSS Distributions 	// reset() on a non-null shared pointer
21*a1e26a70SApple OSS Distributions 	{
22*a1e26a70SApple OSS Distributions 		tracked_shared_ptr<T> ptr(&obj, libkern::retain);
23*a1e26a70SApple OSS Distributions 		tracking_policy::reset();
24*a1e26a70SApple OSS Distributions 		ptr.reset();
25*a1e26a70SApple OSS Distributions 		CHECK(tracking_policy::releases == 1);
26*a1e26a70SApple OSS Distributions 		CHECK(tracking_policy::retains == 0);
27*a1e26a70SApple OSS Distributions 		CHECK(ptr.get() == nullptr);
28*a1e26a70SApple OSS Distributions 	}
29*a1e26a70SApple OSS Distributions 
30*a1e26a70SApple OSS Distributions 	// reset() on a null shared pointer
31*a1e26a70SApple OSS Distributions 	{
32*a1e26a70SApple OSS Distributions 		tracked_shared_ptr<T> ptr = nullptr;
33*a1e26a70SApple OSS Distributions 		tracking_policy::reset();
34*a1e26a70SApple OSS Distributions 		ptr.reset();
35*a1e26a70SApple OSS Distributions 		CHECK(tracking_policy::releases == 0);
36*a1e26a70SApple OSS Distributions 		CHECK(tracking_policy::retains == 0);
37*a1e26a70SApple OSS Distributions 		CHECK(ptr.get() == nullptr);
38*a1e26a70SApple OSS Distributions 	}
39*a1e26a70SApple OSS Distributions 
40*a1e26a70SApple OSS Distributions 	// reset() as a self-reference
41*a1e26a70SApple OSS Distributions 	{
42*a1e26a70SApple OSS Distributions 		tracked_shared_ptr<T> ptr(&obj, libkern::retain);
43*a1e26a70SApple OSS Distributions 		tracked_shared_ptr<T> ptr2(&obj, libkern::retain);
44*a1e26a70SApple OSS Distributions 		CHECK(!ptr.reset());
45*a1e26a70SApple OSS Distributions 
46*a1e26a70SApple OSS Distributions 		CHECK(&ptr.reset() == &ptr);
47*a1e26a70SApple OSS Distributions 
48*a1e26a70SApple OSS Distributions 		// check short-circuiting
49*a1e26a70SApple OSS Distributions 		bool ok =  (ptr.reset() && !ptr2.reset());
50*a1e26a70SApple OSS Distributions 		CHECK(ptr2.get() != nullptr);
51*a1e26a70SApple OSS Distributions 	}
52*a1e26a70SApple OSS Distributions }
53*a1e26a70SApple OSS Distributions 
54*a1e26a70SApple OSS Distributions T_DECL(reset, "intrusive_shared_ptr.reset", T_META_TAG_VM_PREFERRED) {
55*a1e26a70SApple OSS Distributions 	tests<T>();
56*a1e26a70SApple OSS Distributions 	tests<T const>();
57*a1e26a70SApple OSS Distributions }
58