base/threading/thread_local_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/logging.h"
   6 #include "base/threading/simple_thread.h"
   7 #include "base/threading/thread_local.h"
   8 #include "base/synchronization/waitable_event.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace base {
  12
  13 namespace {
  14
  15 class ThreadLocalTesterBase : public base::DelegateSimpleThreadPool::Delegate {
  16  public:
  17   typedef base::ThreadLocalPointer<char> TLPType;
  18
  19   ThreadLocalTesterBase(TLPType* tlp, base::WaitableEvent* done)
  20       : tlp_(tlp),
  21         done_(done) {
  22   }
  23   ~ThreadLocalTesterBase() override {}
  24
  25  protected:
  26   TLPType* tlp_;
  27   base::WaitableEvent* done_;
  28 };
  29
  30 class SetThreadLocal : public ThreadLocalTesterBase {
  31  public:
  32   SetThreadLocal(TLPType* tlp, base::WaitableEvent* done)
  33       : ThreadLocalTesterBase(tlp, done),
  34         val_(NULL) {
  35   }
  36   ~SetThreadLocal() override {}
  37
  38   void set_value(char* val) { val_ = val; }
  39
  40   void Run() override {
  41     DCHECK(!done_->IsSignaled());
  42     tlp_->Set(val_);
  43     done_->Signal();
  44   }
  45
  46  private:
  47   char* val_;
  48 };
  49
  50 class GetThreadLocal : public ThreadLocalTesterBase {
  51  public:
  52   GetThreadLocal(TLPType* tlp, base::WaitableEvent* done)
  53       : ThreadLocalTesterBase(tlp, done),
  54         ptr_(NULL) {
  55   }
  56   ~GetThreadLocal() override {}
  57
  58   void set_ptr(char** ptr) { ptr_ = ptr; }
  59
  60   void Run() override {
  61     DCHECK(!done_->IsSignaled());
  62     *ptr_ = tlp_->Get();
  63     done_->Signal();
  64   }
  65
  66  private:
  67   char** ptr_;
  68 };
  69
  70 }  // namespace
  71
  72 // In this test, we start 2 threads which will access a ThreadLocalPointer.  We
  73 // make sure the default is NULL, and the pointers are unique to the threads.
  74 TEST(ThreadLocalTest, Pointer) {
  75   base::DelegateSimpleThreadPool tp1("ThreadLocalTest tp1", 1);
  76   base::DelegateSimpleThreadPool tp2("ThreadLocalTest tp1", 1);
  77   tp1.Start();
  78   tp2.Start();
  79
  80   base::ThreadLocalPointer<char> tlp;
  81
  82   static char* const kBogusPointer = reinterpret_cast<char*>(0x1234);
  83
  84   char* tls_val;
  85   base::WaitableEvent done(true, false);
  86
  87   GetThreadLocal getter(&tlp, &done);
  88   getter.set_ptr(&tls_val);
  89
  90   // Check that both threads defaulted to NULL.
  91   tls_val = kBogusPointer;
  92   done.Reset();
  93   tp1.AddWork(&getter);
  94   done.Wait();
  95   EXPECT_EQ(static_cast<char*>(NULL), tls_val);
  96
  97   tls_val = kBogusPointer;
  98   done.Reset();
  99   tp2.AddWork(&getter);
 100   done.Wait();
 101   EXPECT_EQ(static_cast<char*>(NULL), tls_val);
 102
 103
 104   SetThreadLocal setter(&tlp, &done);
 105   setter.set_value(kBogusPointer);
 106
 107   // Have thread 1 set their pointer value to kBogusPointer.
 108   done.Reset();
 109   tp1.AddWork(&setter);
 110   done.Wait();
 111
 112   tls_val = NULL;
 113   done.Reset();
 114   tp1.AddWork(&getter);
 115   done.Wait();
 116   EXPECT_EQ(kBogusPointer, tls_val);
 117
 118   // Make sure thread 2 is still NULL
 119   tls_val = kBogusPointer;
 120   done.Reset();
 121   tp2.AddWork(&getter);
 122   done.Wait();
 123   EXPECT_EQ(static_cast<char*>(NULL), tls_val);
 124
 125   // Set thread 2 to kBogusPointer + 1.
 126   setter.set_value(kBogusPointer + 1);
 127
 128   done.Reset();
 129   tp2.AddWork(&setter);
 130   done.Wait();
 131
 132   tls_val = NULL;
 133   done.Reset();
 134   tp2.AddWork(&getter);
 135   done.Wait();
 136   EXPECT_EQ(kBogusPointer + 1, tls_val);
 137
 138   // Make sure thread 1 is still kBogusPointer.
 139   tls_val = NULL;
 140   done.Reset();
 141   tp1.AddWork(&getter);
 142   done.Wait();
 143   EXPECT_EQ(kBogusPointer, tls_val);
 144
 145   tp1.JoinAll();
 146   tp2.JoinAll();
 147 }
 148
 149 TEST(ThreadLocalTest, Boolean) {
 150   {
 151     base::ThreadLocalBoolean tlb;
 152     EXPECT_FALSE(tlb.Get());
 153
 154     tlb.Set(false);
 155     EXPECT_FALSE(tlb.Get());
 156
 157     tlb.Set(true);
 158     EXPECT_TRUE(tlb.Get());
 159   }
 160
 161   // Our slot should have been freed, we're all reset.
 162   {
 163     base::ThreadLocalBoolean tlb;
 164     EXPECT_FALSE(tlb.Get());
 165   }
 166 }
 167
 168 }  // namespace base