threadlocal-test.c++ (2085B)
1 // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors 2 // Licensed under the MIT License: 3 // 4 // Permission is hereby granted, free of charge, to any person obtaining a copy 5 // of this software and associated documentation files (the "Software"), to deal 6 // in the Software without restriction, including without limitation the rights 7 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 8 // copies of the Software, and to permit persons to whom the Software is 9 // furnished to do so, subject to the following conditions: 10 // 11 // The above copyright notice and this permission notice shall be included in 12 // all copies or substantial portions of the Software. 13 // 14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 17 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 18 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 19 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 20 // THE SOFTWARE. 21 22 #include "threadlocal.h" 23 #include "debug.h" 24 #include "thread.h" 25 #include <kj/compat/gtest.h> 26 27 namespace kj { 28 namespace { 29 30 KJ_THREADLOCAL_PTR(uint) tls1 = nullptr; 31 KJ_THREADLOCAL_PTR(uint) tls2; 32 33 TEST(ThreadLocal, Basic) { 34 // Verify that both started out null. 35 uint* p = tls1; 36 EXPECT_EQ(nullptr, p); 37 p = tls2; 38 EXPECT_EQ(nullptr, p); 39 40 // Set tls1, then verify that only tls1 changed, not tls2. 41 uint i = 123; 42 tls1 = &i; 43 44 p = tls1; 45 EXPECT_EQ(&i, p); 46 p = tls2; 47 EXPECT_EQ(nullptr, p); 48 49 // Check that in another thread, tls1 starts null but can be changed. 50 uint j = 456; 51 bool threadDone = false; 52 Thread([&]() { 53 p = tls1; 54 EXPECT_EQ(nullptr, p); 55 tls1 = &j; 56 57 p = tls1; 58 EXPECT_EQ(&j, p); 59 threadDone = true; 60 }); 61 EXPECT_TRUE(threadDone); 62 63 // tls1 didn't change in this thread. 64 p = tls1; 65 EXPECT_EQ(&i, p); 66 } 67 68 } // namespace 69 } // namespace kj