4 files changed, 44 insertions, 0 deletions

diff --git a/tools/perf/tests/Build b/tools/perf/tests/Build
index ee41e705b2eb..d20d6e6ab65b 100644
--- a/tools/perf/tests/Build
+++ b/tools/perf/tests/Build
@@ -31,6 +31,7 @@ perf-y += code-reading.o
 perf-y += sample-parsing.o
 perf-y += parse-no-sample-id-all.o
 perf-y += kmod-path.o
+perf-y += thread-map.o
 
 perf-$(CONFIG_X86) += perf-time-to-tsc.o
 
diff --git a/tools/perf/tests/builtin-test.c b/tools/perf/tests/builtin-test.c
index 87b9961646e4..c1dde733c3a6 100644
--- a/tools/perf/tests/builtin-test.c
+++ b/tools/perf/tests/builtin-test.c
@@ -171,6 +171,10 @@ static struct test {
                 .func = test__kmod_path__parse,
         },
         {
+                .desc = "Test thread map",
+                .func = test__thread_map,
+        },
+        {
                 .func = NULL,
         },
 };
diff --git a/tools/perf/tests/tests.h b/tools/perf/tests/tests.h
index 8e5038b48ba8..ebb47d96bc0b 100644
--- a/tools/perf/tests/tests.h
+++ b/tools/perf/tests/tests.h
@@ -61,6 +61,7 @@ int test__switch_tracking(void);
 int test__fdarray__filter(void);
 int test__fdarray__add(void);
 int test__kmod_path__parse(void);
+int test__thread_map(void);
 
 #if defined(__x86_64__) || defined(__i386__) || defined(__arm__) || defined(__aarch64__)
 #ifdef HAVE_DWARF_UNWIND_SUPPORT
diff --git a/tools/perf/tests/thread-map.c b/tools/perf/tests/thread-map.c
new file mode 100644
index 000000000000..5acf000939ea
--- /dev/null
+++ b/tools/perf/tests/thread-map.c
@@ -0,0 +1,38 @@
+#include <sys/types.h>
+#include <unistd.h>
+#include "tests.h"
+#include "thread_map.h"
+#include "debug.h"
+
+int test__thread_map(void)
+{
+        struct thread_map *map;
+
+        /* test map on current pid */
+        map = thread_map__new_by_pid(getpid());
+        TEST_ASSERT_VAL("failed to alloc map", map);
+
+        thread_map__read_comms(map);
+
+        TEST_ASSERT_VAL("wrong nr", map->nr == 1);
+        TEST_ASSERT_VAL("wrong pid",
+                        thread_map__pid(map, 0) == getpid());
+        TEST_ASSERT_VAL("wrong comm",
+                        thread_map__comm(map, 0) &&
+                        !strcmp(thread_map__comm(map, 0), "perf"));
+        thread_map__put(map);
+
+        /* test dummy pid */
+        map = thread_map__new_dummy();
+        TEST_ASSERT_VAL("failed to alloc map", map);
+
+        thread_map__read_comms(map);
+
+        TEST_ASSERT_VAL("wrong nr", map->nr == 1);
+        TEST_ASSERT_VAL("wrong pid", thread_map__pid(map, 0) == -1);
+        TEST_ASSERT_VAL("wrong comm",
+                        thread_map__comm(map, 0) &&
+                        !strcmp(thread_map__comm(map, 0), "dummy"));
+        thread_map__put(map);
+        return 0;
+}