• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
test_skip(const char * format,...)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
test_fail(const char * format,...)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 *
test_status_string(test_status_t test_status)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
p_test_init(const char * name)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
p_test_fini(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 static test_status_t
p_test_impl(bool do_malloc_init,test_t * t,va_list ap)64 p_test_impl(bool do_malloc_init, test_t *t, va_list ap)
65 {
66 	test_status_t ret;
67 
68 	if (do_malloc_init) {
69 		/*
70 		 * Make sure initialization occurs prior to running tests.
71 		 * Tests are special because they may use internal facilities
72 		 * prior to triggering initialization as a side effect of
73 		 * calling into the public API.
74 		 */
75 		if (nallocx(1, 0) == 0) {
76 			malloc_printf("Initialization error");
77 			return (test_status_fail);
78 		}
79 	}
80 
81 	ret = test_status_pass;
82 	for (; t != NULL; t = va_arg(ap, test_t *)) {
83 		t();
84 		if (test_status > ret)
85 			ret = test_status;
86 	}
87 
88 	malloc_printf("--- %s: %u/%u, %s: %u/%u, %s: %u/%u ---\n",
89 	    test_status_string(test_status_pass),
90 	    test_counts[test_status_pass], test_count,
91 	    test_status_string(test_status_skip),
92 	    test_counts[test_status_skip], test_count,
93 	    test_status_string(test_status_fail),
94 	    test_counts[test_status_fail], test_count);
95 
96 	return (ret);
97 }
98 
99 test_status_t
p_test(test_t * t,...)100 p_test(test_t *t, ...)
101 {
102 	test_status_t ret;
103 	va_list ap;
104 
105 	ret = test_status_pass;
106 	va_start(ap, t);
107 	ret = p_test_impl(true, t, ap);
108 	va_end(ap);
109 
110 	return (ret);
111 }
112 
113 test_status_t
p_test_no_malloc_init(test_t * t,...)114 p_test_no_malloc_init(test_t *t, ...)
115 {
116 	test_status_t ret;
117 	va_list ap;
118 
119 	ret = test_status_pass;
120 	va_start(ap, t);
121 	ret = p_test_impl(false, t, ap);
122 	va_end(ap);
123 
124 	return (ret);
125 }
126 
127 void
p_test_fail(const char * prefix,const char * message)128 p_test_fail(const char *prefix, const char *message)
129 {
130 
131 	malloc_cprintf(NULL, NULL, "%s%s\n", prefix, message);
132 	test_status = test_status_fail;
133 }
134