From 04e53de9e5406d7fc177da2c3f886709268ec466 Mon Sep 17 00:00:00 2001 From: Tyler Retzlaff Date: Thu, 12 May 2022 06:14:30 -0700 Subject: [PATCH] test/threads: add unit test Establish unit test for testing thread api. Initial unit tests for rte_thread_{get,set}_affinity_by_id(). Signed-off-by: Narcisa Vasile Signed-off-by: Tyler Retzlaff --- MAINTAINERS | 1 + app/test/meson.build | 2 + app/test/test_threads.c | 81 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 84 insertions(+) create mode 100644 app/test/test_threads.c diff --git a/MAINTAINERS b/MAINTAINERS index 8f4e9c3479..17a0559ee7 100644 --- a/MAINTAINERS +++ b/MAINTAINERS @@ -181,6 +181,7 @@ F: app/test/test_rwlock.c F: app/test/test_spinlock.c F: app/test/test_string_fns.c F: app/test/test_tailq.c +F: app/test/test_threads.c F: app/test/test_version.c Trace - EXPERIMENTAL diff --git a/app/test/meson.build b/app/test/meson.build index bb4621ed2a..15591ce5cf 100644 --- a/app/test/meson.build +++ b/app/test/meson.build @@ -133,6 +133,7 @@ test_sources = files( 'test_tailq.c', 'test_thash.c', 'test_thash_perf.c', + 'test_threads.c', 'test_timer.c', 'test_timer_perf.c', 'test_timer_racecond.c', @@ -239,6 +240,7 @@ fast_tests = [ ['reorder_autotest', true, true], ['service_autotest', true, true], ['thash_autotest', true, true], + ['threads_autotest', true, true], ['trace_autotest', true, true], ] diff --git a/app/test/test_threads.c b/app/test/test_threads.c new file mode 100644 index 0000000000..e1a2ea5be1 --- /dev/null +++ b/app/test/test_threads.c @@ -0,0 +1,81 @@ +/* SPDX-License-Identifier: BSD-3-Clause + * Copyright (C) 2022 Microsoft Corporation + */ + +#include +#include + +#include +#include + +#include "test.h" + +RTE_LOG_REGISTER(threads_logtype_test, test.threads, INFO); + +static uint32_t thread_id_ready; + +static void * +thread_main(void *arg) +{ + *(rte_thread_t *)arg = rte_thread_self(); + __atomic_store_n(&thread_id_ready, 1, __ATOMIC_RELEASE); + + return NULL; +} + +static int +test_thread_affinity(void) +{ + pthread_t id; + rte_thread_t thread_id; + rte_cpuset_t cpuset0; + rte_cpuset_t cpuset1; + + RTE_TEST_ASSERT(pthread_create(&id, NULL, thread_main, &thread_id) == 0, + "Failed to create thread"); + + while (__atomic_load_n(&thread_id_ready, __ATOMIC_ACQUIRE) == 0) + ; + + RTE_TEST_ASSERT(rte_thread_get_affinity_by_id(thread_id, &cpuset0) == 0, + "Failed to get thread affinity"); + RTE_TEST_ASSERT(rte_thread_get_affinity_by_id(thread_id, &cpuset1) == 0, + "Failed to get thread affinity"); + RTE_TEST_ASSERT(memcmp(&cpuset0, &cpuset1, sizeof(rte_cpuset_t)) == 0, + "Affinity should be stable"); + + size_t i; + for (i = 1; i < CPU_SETSIZE; i++) + if (CPU_ISSET(i, &cpuset0)) { + CPU_ZERO(&cpuset0); + CPU_SET(i, &cpuset0); + + break; + } + RTE_TEST_ASSERT(rte_thread_set_affinity_by_id(thread_id, &cpuset0) == 0, + "Failed to set thread affinity"); + RTE_TEST_ASSERT(rte_thread_get_affinity_by_id(thread_id, &cpuset1) == 0, + "Failed to get thread affinity"); + RTE_TEST_ASSERT(memcmp(&cpuset0, &cpuset1, sizeof(rte_cpuset_t)) == 0, + "Affinity should be stable"); + + return 0; +} + +static struct unit_test_suite threads_test_suite = { + .suite_name = "threads autotest", + .setup = NULL, + .teardown = NULL, + .unit_test_cases = { + TEST_CASE(test_thread_affinity), + TEST_CASES_END() + } +}; + +static int +test_threads(void) +{ + return unit_test_suite_runner(&threads_test_suite); +} + +REGISTER_TEST_COMMAND(threads_autotest, test_threads);