1*bb611c8fSApple OSS Distributions // 2*bb611c8fSApple OSS Distributions // Tests for 3*bb611c8fSApple OSS Distributions // safe_allocation& operator=(std::nullptr_t); 4*bb611c8fSApple OSS Distributions // 5*bb611c8fSApple OSS Distributions 6*bb611c8fSApple OSS Distributions #include <libkern/c++/safe_allocation.h> 7*bb611c8fSApple OSS Distributions #include <darwintest.h> 8*bb611c8fSApple OSS Distributions #include "test_utils.h" 9*bb611c8fSApple OSS Distributions 10*bb611c8fSApple OSS Distributions struct T { 11*bb611c8fSApple OSS Distributions int i; 12*bb611c8fSApple OSS Distributions }; 13*bb611c8fSApple OSS Distributions 14*bb611c8fSApple OSS Distributions template <typename T> 15*bb611c8fSApple OSS Distributions static void 16*bb611c8fSApple OSS Distributions tests() 17*bb611c8fSApple OSS Distributions { 18*bb611c8fSApple OSS Distributions // Assign to a non-null allocation 19*bb611c8fSApple OSS Distributions { 20*bb611c8fSApple OSS Distributions tracked_safe_allocation<T> array(10, libkern::allocate_memory); 21*bb611c8fSApple OSS Distributions tracking_allocator::reset(); 22*bb611c8fSApple OSS Distributions 23*bb611c8fSApple OSS Distributions tracked_safe_allocation<T>& ref = (array = nullptr); 24*bb611c8fSApple OSS Distributions CHECK(&ref == &array); 25*bb611c8fSApple OSS Distributions CHECK(array.data() == nullptr); 26*bb611c8fSApple OSS Distributions CHECK(array.size() == 0); 27*bb611c8fSApple OSS Distributions CHECK(tracking_allocator::did_deallocate); 28*bb611c8fSApple OSS Distributions } 29*bb611c8fSApple OSS Distributions 30*bb611c8fSApple OSS Distributions // Assign to a null allocation 31*bb611c8fSApple OSS Distributions { 32*bb611c8fSApple OSS Distributions tracked_safe_allocation<T> array = nullptr; 33*bb611c8fSApple OSS Distributions tracking_allocator::reset(); 34*bb611c8fSApple OSS Distributions 35*bb611c8fSApple OSS Distributions tracked_safe_allocation<T>& ref = (array = nullptr); 36*bb611c8fSApple OSS Distributions CHECK(&ref == &array); 37*bb611c8fSApple OSS Distributions CHECK(array.data() == nullptr); 38*bb611c8fSApple OSS Distributions CHECK(array.size() == 0); 39*bb611c8fSApple OSS Distributions CHECK(!tracking_allocator::did_deallocate); 40*bb611c8fSApple OSS Distributions } 41*bb611c8fSApple OSS Distributions } 42*bb611c8fSApple OSS Distributions 43*bb611c8fSApple OSS Distributions T_DECL(assign_nullptr, "safe_allocation.assign.nullptr") { 44*bb611c8fSApple OSS Distributions tests<T>(); 45*bb611c8fSApple OSS Distributions tests<T const>(); 46*bb611c8fSApple OSS Distributions } 47