xref: /xnu-8792.61.2/tests/safe_allocation_src/usage.for_loop.cpp (revision 42e220869062b56f8d7d0726fd4c88954f87902c) 
1*42e22086SApple OSS Distributions //
2*42e22086SApple OSS Distributions // Make sure `safe_allocation` works nicely with the range-based for-loop.
3*42e22086SApple OSS Distributions //
4*42e22086SApple OSS Distributions 
5*42e22086SApple OSS Distributions #include <libkern/c++/safe_allocation.h>
6*42e22086SApple OSS Distributions #include <darwintest.h>
7*42e22086SApple OSS Distributions #include "test_utils.h"
8*42e22086SApple OSS Distributions 
9*42e22086SApple OSS Distributions struct T {
10*42e22086SApple OSS Distributions 	int i;
11*42e22086SApple OSS Distributions };
12*42e22086SApple OSS Distributions 
13*42e22086SApple OSS Distributions template <typename T>
14*42e22086SApple OSS Distributions static void
tests()15*42e22086SApple OSS Distributions tests()
16*42e22086SApple OSS Distributions {
17*42e22086SApple OSS Distributions 	test_safe_allocation<T> array(10, libkern::allocate_memory);
18*42e22086SApple OSS Distributions 	for (T& element : array) {
19*42e22086SApple OSS Distributions 		element = T{3};
20*42e22086SApple OSS Distributions 	}
21*42e22086SApple OSS Distributions 
22*42e22086SApple OSS Distributions 	for (T const& element : array) {
23*42e22086SApple OSS Distributions 		CHECK(element.i == 3);
24*42e22086SApple OSS Distributions 	}
25*42e22086SApple OSS Distributions }
26*42e22086SApple OSS Distributions 
27*42e22086SApple OSS Distributions T_DECL(usage_for_loop, "safe_allocation.usage.for_loop") {
28*42e22086SApple OSS Distributions 	tests<T>();
29*42e22086SApple OSS Distributions }
30