/*
- * Copyright 2016 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*/
#include <thread>
+
#include <folly/Baton.h>
#include <folly/ScopeGuard.h>
#include <folly/ThreadName.h>
using namespace std;
using namespace folly;
-constexpr bool expectedSetOtherThreadNameResult =
-#ifdef FOLLY_HAS_PTHREAD_SETNAME_NP_THREAD_NAME
- true
-#else
- false // This system has no known way to set the name of another thread
-#endif
- ;
+namespace {
+
+const bool expectedSetOtherThreadNameResult = folly::canSetOtherThreadName();
+const bool expectedSetSelfThreadNameResult = folly::canSetCurrentThreadName();
+constexpr StringPiece kThreadName{"rockin-thread"};
-constexpr bool expectedSetSelfThreadNameResult =
-#if defined(FOLLY_HAS_PTHREAD_SETNAME_NP_THREAD_NAME) || \
- defined(FOLLY_HAS_PTHREAD_SETNAME_NP_NAME)
- true
-#else
- false // This system has no known way to set its own thread name
-#endif
- ;
+} // namespace
-TEST(ThreadName, setThreadName_self) {
+TEST(ThreadName, getCurrentThreadName) {
thread th([] {
- EXPECT_EQ(expectedSetSelfThreadNameResult, setThreadName("rockin-thread"));
+ EXPECT_EQ(expectedSetSelfThreadNameResult, setThreadName(kThreadName));
+ if (expectedSetSelfThreadNameResult) {
+ EXPECT_EQ(kThreadName.toString(), *getCurrentThreadName());
+ }
});
SCOPE_EXIT { th.join(); };
}
handle_set.wait();
SCOPE_EXIT { let_thread_end.post(); };
EXPECT_EQ(
- expectedSetOtherThreadNameResult, setThreadName(handle, "rockin-thread"));
+ expectedSetOtherThreadNameResult, setThreadName(handle, kThreadName));
}
-TEST(ThreadName, setThreadName_other_native) {
+TEST(ThreadName, setThreadName_other_id) {
Baton<> let_thread_end;
thread th([&] {
let_thread_end.wait();
SCOPE_EXIT { let_thread_end.post(); };
EXPECT_EQ(
expectedSetOtherThreadNameResult,
- setThreadName(th.native_handle(), "rockin-thread"));
+ setThreadName(th.get_id(), kThreadName));
+ if (expectedSetOtherThreadNameResult) {
+ EXPECT_EQ(*getThreadName(th.get_id()), kThreadName);
+ }
}