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