// SPDX-License-Identifier: GPL-2.0
#include <stdarg.h>
#include <stdio.h>
#include <perf/threadmap.h>
#include <internal/tests.h>
#include "tests.h"

static int libperf_print(enum libperf_print_level level,
    const char *fmt, va_list ap)
{
 return vfprintf(stderr, fmt, ap);
}

static int test_threadmap_array(int nr, pid_t *array)
{
 struct perf_thread_map *threads;
 int i;

 threads = perf_thread_map__new_array(nr, array);
 __T("Failed to allocate new thread map", threads);

 __T("Unexpected number of threads", perf_thread_map__nr(threads) == nr);

 for (i = 0; i < nr; i++) {
  __T("Unexpected initial value of thread",
      perf_thread_map__pid(threads, i) == (array ? array[i] : -1));
 }

 for (i = 1; i < nr; i++)
  perf_thread_map__set_pid(threads, i, i * 100);

 __T("Unexpected value of thread 0",
     perf_thread_map__pid(threads, 0) == (array ? array[0] : -1));

 for (i = 1; i < nr; i++) {
  __T("Unexpected thread value",
      perf_thread_map__pid(threads, i) == i * 100);
 }

 perf_thread_map__put(threads);

 return 0;
}

#define THREADS_NR 10
int test_threadmap(int argc, char **argv)
{
 struct perf_thread_map *threads;
 pid_t thr_array[THREADS_NR];
 int i;

 __T_START;

 libperf_init(libperf_print);

 threads = perf_thread_map__new_dummy();
 if (!threads)
  return -1;

 perf_thread_map__get(threads);
 perf_thread_map__put(threads);
 perf_thread_map__put(threads);

 test_threadmap_array(THREADS_NR, NULL);

 for (i = 0; i < THREADS_NR; i++)
  thr_array[i] = i + 100;

 test_threadmap_array(THREADS_NR, thr_array);

 __T_END;
 return tests_failed == 0 ? 0 : -1;
}