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