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