/*
- * Copyright 2015 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>
-#include <gtest/gtest.h>
+#include <folly/portability/GTest.h>
using namespace std;
using namespace folly;
-TEST(ThreadName, setThreadName_self) {
+namespace {
+
+const bool expectedSetOtherThreadNameResult = folly::canSetOtherThreadName();
+const bool expectedSetSelfThreadNameResult = folly::canSetCurrentThreadName();
+constexpr StringPiece kThreadName{"rockin-thread"};
+
+} // namespace
+
+TEST(ThreadName, getCurrentThreadName) {
thread th([] {
- EXPECT_TRUE(setThreadName("rockin-thread"));
+ EXPECT_EQ(expectedSetSelfThreadNameResult, setThreadName(kThreadName));
+ if (expectedSetSelfThreadNameResult) {
+ EXPECT_EQ(kThreadName.toString(), *getCurrentThreadName());
+ }
});
SCOPE_EXIT { th.join(); };
}
SCOPE_EXIT { th.join(); };
handle_set.wait();
SCOPE_EXIT { let_thread_end.post(); };
- EXPECT_TRUE(setThreadName(handle, "rockin-thread"));
+ EXPECT_EQ(
+ 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 { th.join(); };
SCOPE_EXIT { let_thread_end.post(); };
- EXPECT_TRUE(setThreadName(th.native_handle(), "rockin-thread"));
+ EXPECT_EQ(
+ expectedSetOtherThreadNameResult,
+ setThreadName(th.get_id(), kThreadName));
+ if (expectedSetOtherThreadNameResult) {
+ EXPECT_EQ(*getThreadName(th.get_id()), kThreadName);
+ }
}