unittests/Support/ThreadLocalTest.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

//===- llvm/unittest/Support/ThreadLocalTest.cpp - ThreadLocal tests ------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/ThreadLocal.h"
#include "gtest/gtest.h"
#include <type_traits>

using namespace llvm;
using namespace sys;

namespace {

class ThreadLocalTest : public ::testing::Test {
};

struct S {
  int i;
};

TEST_F(ThreadLocalTest, Basics) {
  ThreadLocal<const S> x;

  static_assert(
      std::is_const<std::remove_pointer<decltype(x.get())>::type>::value,
      "ThreadLocal::get didn't return a pointer to const object");

  EXPECT_EQ(nullptr, x.get());

  S s;
  x.set(&s);
  EXPECT_EQ(&s, x.get());

  x.erase();
  EXPECT_EQ(nullptr, x.get());

  ThreadLocal<S> y;

  static_assert(
      !std::is_const<std::remove_pointer<decltype(y.get())>::type>::value,
      "ThreadLocal::get returned a pointer to const object");

  EXPECT_EQ(nullptr, y.get());

  y.set(&s);
  EXPECT_EQ(&s, y.get());

  y.erase();
  EXPECT_EQ(nullptr, y.get());
}

}