1 #include "test/jemalloc_test.h" 2 3 static unsigned test_count = 0; 4 static test_status_t test_counts[test_status_count] = {0, 0, 0}; 5 static test_status_t test_status = test_status_pass; 6 static const char * test_name = ""; 7 8 JEMALLOC_FORMAT_PRINTF(1, 2) 9 void 10 test_skip(const char *format, ...) 11 { 12 va_list ap; 13 14 va_start(ap, format); 15 malloc_vcprintf(NULL, NULL, format, ap); 16 va_end(ap); 17 malloc_printf("\n"); 18 test_status = test_status_skip; 19 } 20 21 JEMALLOC_FORMAT_PRINTF(1, 2) 22 void 23 test_fail(const char *format, ...) 24 { 25 va_list ap; 26 27 va_start(ap, format); 28 malloc_vcprintf(NULL, NULL, format, ap); 29 va_end(ap); 30 malloc_printf("\n"); 31 test_status = test_status_fail; 32 } 33 34 static const char * 35 test_status_string(test_status_t test_status) 36 { 37 38 switch (test_status) { 39 case test_status_pass: return "pass"; 40 case test_status_skip: return "skip"; 41 case test_status_fail: return "fail"; 42 default: not_reached(); 43 } 44 } 45 46 void 47 p_test_init(const char *name) 48 { 49 50 test_count++; 51 test_status = test_status_pass; 52 test_name = name; 53 } 54 55 void 56 p_test_fini(void) 57 { 58 59 test_counts[test_status]++; 60 malloc_printf("%s: %s\n", test_name, test_status_string(test_status)); 61 } 62 63 test_status_t 64 p_test(test_t *t, ...) 65 { 66 test_status_t ret; 67 va_list ap; 68 69 /* 70 * Make sure initialization occurs prior to running tests. Tests are 71 * special because they may use internal facilities prior to triggering 72 * initialization as a side effect of calling into the public API. This 73 * is a final safety that works even if jemalloc_constructor() doesn't 74 * run, as for MSVC builds. 75 */ 76 if (nallocx(1, 0) == 0) { 77 malloc_printf("Initialization error"); 78 return (test_status_fail); 79 } 80 81 ret = test_status_pass; 82 va_start(ap, t); 83 for (; t != NULL; t = va_arg(ap, test_t *)) { 84 t(); 85 if (test_status > ret) 86 ret = test_status; 87 } 88 va_end(ap); 89 90 malloc_printf("--- %s: %u/%u, %s: %u/%u, %s: %u/%u ---\n", 91 test_status_string(test_status_pass), 92 test_counts[test_status_pass], test_count, 93 test_status_string(test_status_skip), 94 test_counts[test_status_skip], test_count, 95 test_status_string(test_status_fail), 96 test_counts[test_status_fail], test_count); 97 98 return (ret); 99 } 100 101 void 102 p_test_fail(const char *prefix, const char *message) 103 { 104 105 malloc_cprintf(NULL, NULL, "%s%s\n", prefix, message); 106 test_status = test_status_fail; 107 } 108
