llvm/unittests/Support/ThreadLocalTest.cpp

   1 //===- llvm/unittest/Support/ThreadLocalTest.cpp - ThreadLocal tests ------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/Support/ThreadLocal.h"
  10 #include "gtest/gtest.h"
  11 #include <type_traits>
  12
  13 using namespace llvm;
  14 using namespace sys;
  15
  16 namespace {
  17
  18 class ThreadLocalTest : public ::testing::Test {
  19 };
  20
  21 struct S {
  22   int i;
  23 };
  24
  25 TEST_F(ThreadLocalTest, Basics) {
  26   ThreadLocal<const S> x;
  27
  28   static_assert(
  29       std::is_const<std::remove_pointer<decltype(x.get())>::type>::value,
  30       "ThreadLocal::get didn't return a pointer to const object");
  31
  32   EXPECT_EQ(nullptr, x.get());
  33
  34   S s;
  35   x.set(&s);
  36   EXPECT_EQ(&s, x.get());
  37
  38   x.erase();
  39   EXPECT_EQ(nullptr, x.get());
  40
  41   ThreadLocal<S> y;
  42
  43   static_assert(
  44       !std::is_const<std::remove_pointer<decltype(y.get())>::type>::value,
  45       "ThreadLocal::get returned a pointer to const object");
  46
  47   EXPECT_EQ(nullptr, y.get());
  48
  49   y.set(&s);
  50   EXPECT_EQ(&s, y.get());
  51
  52   y.erase();
  53   EXPECT_EQ(nullptr, y.get());
  54 }
  55
  56 }