1 // SPDX-License-Identifier: GPL-2.0
2 #include <stdarg.h>
3 #include <stdio.h>
4 #include <perf/threadmap.h>
5 #include <internal/tests.h>
6 #include "tests.h"
7
libperf_print(enum libperf_print_level level,const char * fmt,va_list ap)8 static int libperf_print(enum libperf_print_level level,
9 const char *fmt, va_list ap)
10 {
11 return vfprintf(stderr, fmt, ap);
12 }
13
test_threadmap_array(int nr,pid_t * array)14 static int test_threadmap_array(int nr, pid_t *array)
15 {
16 struct perf_thread_map *threads;
17 int i;
18
19 threads = perf_thread_map__new_array(nr, array);
20 __T("Failed to allocate new thread map", threads);
21
22 __T("Unexpected number of threads", perf_thread_map__nr(threads) == nr);
23
24 for (i = 0; i < nr; i++) {
25 __T("Unexpected initial value of thread",
26 perf_thread_map__pid(threads, i) == (array ? array[i] : -1));
27 }
28
29 for (i = 1; i < nr; i++)
30 perf_thread_map__set_pid(threads, i, i * 100);
31
32 __T("Unexpected value of thread 0",
33 perf_thread_map__pid(threads, 0) == (array ? array[0] : -1));
34
35 for (i = 1; i < nr; i++) {
36 __T("Unexpected thread value",
37 perf_thread_map__pid(threads, i) == i * 100);
38 }
39
40 perf_thread_map__put(threads);
41
42 return 0;
43 }
44
45 #define THREADS_NR 10
test_threadmap(int argc,char ** argv)46 int test_threadmap(int argc, char **argv)
47 {
48 struct perf_thread_map *threads;
49 pid_t thr_array[THREADS_NR];
50 int i;
51
52 __T_START;
53
54 libperf_init(libperf_print);
55
56 threads = perf_thread_map__new_dummy();
57 if (!threads)
58 return -1;
59
60 perf_thread_map__get(threads);
61 perf_thread_map__put(threads);
62 perf_thread_map__put(threads);
63
64 test_threadmap_array(THREADS_NR, NULL);
65
66 for (i = 0; i < THREADS_NR; i++)
67 thr_array[i] = i + 100;
68
69 test_threadmap_array(THREADS_NR, thr_array);
70
71 __T_END;
72 return tests_failed == 0 ? 0 : -1;
73 }
74