Skip to content

Commit 7030db1

Browse files
committed
test: cover GetNumThreads default
1 parent e8c853a commit 7030db1

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

modules/core/util/tests/util.cpp

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
#include <gtest/gtest.h>
44

5+
#include <cstdlib>
56
#include <string_view>
67

78
#include "omp.h"
@@ -26,6 +27,24 @@ TEST(util_tests, threads_control_check_openmp_disabled_valgrind) {
2627
ASSERT_EQ(ppc_num_threads, omp_num_threads);
2728
}
2829

30+
TEST(util_tests, GetNumThreadsEnvUnsetReturnsOne) {
31+
const char* old_value = std::getenv("PPC_NUM_THREADS");
32+
std::string saved_value;
33+
if (old_value != nullptr) {
34+
saved_value = old_value;
35+
}
36+
37+
unsetenv("PPC_NUM_THREADS");
38+
39+
EXPECT_EQ(ppc::util::GetNumThreads(), 1);
40+
41+
if (old_value != nullptr) {
42+
setenv("PPC_NUM_THREADS", saved_value.c_str(), 1);
43+
} else {
44+
unsetenv("PPC_NUM_THREADS");
45+
}
46+
}
47+
2948
namespace test_ns {
3049
struct TypeInNamespace {};
3150
} // namespace test_ns

0 commit comments

Comments
 (0)