/*
- * 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
- ;
-
-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
- ;
+static bool expectedSetOtherThreadNameResult = folly::canSetOtherThreadName();
+static bool expectedSetSelfThreadNameResult = folly::canSetCurrentThreadName();
TEST(ThreadName, setThreadName_self) {
thread th([] {
expectedSetOtherThreadNameResult,
setThreadName(th.native_handle(), "rockin-thread"));
}
+
+TEST(ThreadName, setThreadName_other_id) {
+ Baton<> let_thread_end;
+ thread th([&] {
+ let_thread_end.wait();
+ });
+ SCOPE_EXIT { th.join(); };
+ SCOPE_EXIT { let_thread_end.post(); };
+ EXPECT_EQ(
+ expectedSetOtherThreadNameResult,
+ setThreadName(th.get_id(), "rockin-thread"));
+}