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