folly/test/ThreadNameTest.cpp

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <thread>
  18 #include <folly/Baton.h>
  19 #include <folly/ScopeGuard.h>
  20 #include <folly/ThreadName.h>
  21 #include <folly/portability/GTest.h>
  22
  23 using namespace std;
  24 using namespace folly;
  25
  26 constexpr bool expectedSetOtherThreadNameResult =
  27 #ifdef FOLLY_HAS_PTHREAD_SETNAME_NP_THREAD_NAME
  28     true
  29 #else
  30     false // This system has no known way to set the name of another thread
  31 #endif
  32     ;
  33
  34 constexpr bool expectedSetSelfThreadNameResult =
  35 #if defined(FOLLY_HAS_PTHREAD_SETNAME_NP_THREAD_NAME) || \
  36     defined(FOLLY_HAS_PTHREAD_SETNAME_NP_NAME)
  37     true
  38 #else
  39     false // This system has no known way to set its own thread name
  40 #endif
  41     ;
  42
  43 TEST(ThreadName, setThreadName_self) {
  44   thread th([] {
  45     EXPECT_EQ(expectedSetSelfThreadNameResult, setThreadName("rockin-thread"));
  46   });
  47   SCOPE_EXIT { th.join(); };
  48 }
  49
  50 TEST(ThreadName, setThreadName_other_pthread) {
  51   Baton<> handle_set;
  52   Baton<> let_thread_end;
  53   pthread_t handle;
  54   thread th([&] {
  55       handle = pthread_self();
  56       handle_set.post();
  57       let_thread_end.wait();
  58   });
  59   SCOPE_EXIT { th.join(); };
  60   handle_set.wait();
  61   SCOPE_EXIT { let_thread_end.post(); };
  62   EXPECT_EQ(
  63       expectedSetOtherThreadNameResult, setThreadName(handle, "rockin-thread"));
  64 }
  65
  66 TEST(ThreadName, setThreadName_other_native) {
  67   Baton<> let_thread_end;
  68   thread th([&] {
  69       let_thread_end.wait();
  70   });
  71   SCOPE_EXIT { th.join(); };
  72   SCOPE_EXIT { let_thread_end.post(); };
  73   EXPECT_EQ(
  74       expectedSetOtherThreadNameResult,
  75       setThreadName(th.native_handle(), "rockin-thread"));
  76 }