cannam@147: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
cannam@147: // Licensed under the MIT License:
cannam@147: //
cannam@147: // Permission is hereby granted, free of charge, to any person obtaining a copy
cannam@147: // of this software and associated documentation files (the "Software"), to deal
cannam@147: // in the Software without restriction, including without limitation the rights
cannam@147: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
cannam@147: // copies of the Software, and to permit persons to whom the Software is
cannam@147: // furnished to do so, subject to the following conditions:
cannam@147: //
cannam@147: // The above copyright notice and this permission notice shall be included in
cannam@147: // all copies or substantial portions of the Software.
cannam@147: //
cannam@147: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
cannam@147: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
cannam@147: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
cannam@147: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
cannam@147: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
cannam@147: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
cannam@147: // THE SOFTWARE.
cannam@147: 
cannam@147: #include "threadlocal.h"
cannam@147: #include "debug.h"
cannam@147: #include "thread.h"
cannam@147: #include <kj/compat/gtest.h>
cannam@147: 
cannam@147: namespace kj {
cannam@147: namespace {
cannam@147: 
cannam@147: KJ_THREADLOCAL_PTR(uint) tls1 = nullptr;
cannam@147: KJ_THREADLOCAL_PTR(uint) tls2;
cannam@147: 
cannam@147: TEST(ThreadLocal, Basic) {
cannam@147:   // Verify that both started out null.
cannam@147:   uint* p = tls1;
cannam@147:   EXPECT_EQ(nullptr, p);
cannam@147:   p = tls2;
cannam@147:   EXPECT_EQ(nullptr, p);
cannam@147: 
cannam@147:   // Set tls1, then verify that only tls1 changed, not tls2.
cannam@147:   uint i = 123;
cannam@147:   tls1 = &i;
cannam@147: 
cannam@147:   p = tls1;
cannam@147:   EXPECT_EQ(&i, p);
cannam@147:   p = tls2;
cannam@147:   EXPECT_EQ(nullptr, p);
cannam@147: 
cannam@147:   // Check that in another thread, tls1 starts null but can be changed.
cannam@147:   uint j = 456;
cannam@147:   bool threadDone = false;
cannam@147:   Thread([&]() {
cannam@147:     p = tls1;
cannam@147:     EXPECT_EQ(nullptr, p);
cannam@147:     tls1 = &j;
cannam@147: 
cannam@147:     p = tls1;
cannam@147:     EXPECT_EQ(&j, p);
cannam@147:     threadDone = true;
cannam@147:   });
cannam@147:   EXPECT_TRUE(threadDone);
cannam@147: 
cannam@147:   // tls1 didn't change in this thread.
cannam@147:   p = tls1;
cannam@147:   EXPECT_EQ(&i, p);
cannam@147: }
cannam@147: 
cannam@147: }  // namespace
cannam@147: }  // namespace kj