// SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
#include "debug.h"
#include "evlist.h"
#include "parse-events.h"
#include "tests.h"
#include "tool_pmu.h"

static int do_test(enum tool_pmu_event ev, bool with_pmu)
{
 struct evlist *evlist = evlist__new();
 struct evsel *evsel;
 struct parse_events_error err;
 int ret;
 char str[128];
 bool found = false;

 if (!evlist) {
  pr_err("evlist allocation failed\n");
  return TEST_FAIL;
 }

 if (with_pmu)
  snprintf(str, sizeof(str), "tool/%s/", tool_pmu__event_to_str(ev));
 else
  snprintf(str, sizeof(str), "%s", tool_pmu__event_to_str(ev));

 parse_events_error__init(&err);
 ret = parse_events(evlist, str, &err);
 if (ret) {
  if (!tool_pmu__event_to_str(ev)) {
   ret = TEST_OK;
   goto out;
  }

  pr_debug("FAILED %s:%d failed to parse event '%s', err %d\n",
    __FILE__, __LINE__, str, ret);
  parse_events_error__print(&err, str);
  ret = TEST_FAIL;
  goto out;
 }

 ret = TEST_OK;
 if (with_pmu ? (evlist->core.nr_entries != 1) : (evlist->core.nr_entries < 1)) {
  pr_debug("FAILED %s:%d Unexpected number of events for '%s' of %d\n",
    __FILE__, __LINE__, str, evlist->core.nr_entries);
  ret = TEST_FAIL;
  goto out;
 }

 evlist__for_each_entry(evlist, evsel) {
  if (perf_pmu__is_tool(evsel->pmu)) {
   if (evsel->core.attr.config != ev) {
    pr_debug("FAILED %s:%d Unexpected config for '%s', %lld != %d\n",
     __FILE__, __LINE__, str, evsel->core.attr.config, ev);
    ret = TEST_FAIL;
    goto out;
   }
   found = true;
  }
 }

 if (!found && tool_pmu__event_to_str(ev)) {
  pr_debug("FAILED %s:%d Didn't find tool event '%s' in parsed evsels\n",
    __FILE__, __LINE__, str);
  ret = TEST_FAIL;
 }

out:
 parse_events_error__exit(&err);
 evlist__delete(evlist);
 return ret;
}

static int test__tool_pmu_without_pmu(struct test_suite *test __maybe_unused,
          int subtest __maybe_unused)
{
 int i;

 tool_pmu__for_each_event(i) {
  int ret = do_test(i, /*with_pmu=*/false);

  if (ret != TEST_OK)
   return ret;
 }
 return TEST_OK;
}

static int test__tool_pmu_with_pmu(struct test_suite *test __maybe_unused,
       int subtest __maybe_unused)
{
 int i;

 tool_pmu__for_each_event(i) {
  int ret = do_test(i, /*with_pmu=*/true);

  if (ret != TEST_OK)
   return ret;
 }
 return TEST_OK;
}

static struct test_case tests__tool_pmu[] = {
 TEST_CASE("Parsing without PMU name", tool_pmu_without_pmu),
 TEST_CASE("Parsing with PMU name", tool_pmu_with_pmu),
 { .name = NULL, }
};

struct test_suite suite__tool_pmu = {
 .desc = "Tool PMU",
 .test_cases = tests__tool_pmu,
};