Open Menu / tests / intrusive_shared_ptr_src / test_policy.h
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#ifndef TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H
#define TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H

#include <libkern/c++/intrusive_shared_ptr.h>
#include <darwintest_utils.h>

struct test_policy {
	static inline int retain_count = 0;

	template <typename T>
	static void
	retain(T&)
	{
		++retain_count;
	}
	template <typename T>
	static void
	release(T&)
	{
		--retain_count;
	}
};

struct tracking_policy {
	static inline int retains = 0;
	static inline int releases = 0;
	static inline int refcount = 0;
	static inline bool hit_zero = false;

	static void
	reset()
	{
		retains = 0;
		releases = 0;
		refcount = 0;
		hit_zero = false;
	}

	template <typename T>
	static void
	retain(T&)
	{
		++retains;
		++refcount;
	}
	template <typename T>
	static void
	release(T&)
	{
		++releases;
		--refcount;
		if (refcount == 0) {
			hit_zero = true;
		}
	}
};

template <int>
struct dummy_policy {
	template <typename T>
	static void
	retain(T&)
	{
	}
	template <typename T>
	static void
	release(T&)
	{
	}
};

template <typename T>
using tracked_shared_ptr = libkern::intrusive_shared_ptr<T, tracking_policy>;

template <typename T>
using test_shared_ptr = libkern::intrusive_shared_ptr<T, test_policy>;

#define CHECK(...) T_ASSERT_TRUE((__VA_ARGS__), # __VA_ARGS__)

#endif // !TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H