Mercurial > hg > CbC > CbC_llvm

annotate unittests/ADT/IntrusiveRefCntPtrTest.cpp @ 131:f476a9ba4795

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
http://llvm.org/svn/llvm-project/compiler-rt/trunk compiler-rt
author mir3636
date Fri, 16 Feb 2018 21:02:11 +0900
parents 803732b1fca8
children c2174574ed3a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
1 //===- unittest/ADT/IntrusiveRefCntPtrTest.cpp ----------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
2 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
3 // The LLVM Compiler Infrastructure
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
4 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
5 // This file is distributed under the University of Illinois Open Source
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
6 // License. See LICENSE.TXT for details.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
7 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
8 //===----------------------------------------------------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
9
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
10 #include "llvm/ADT/IntrusiveRefCntPtr.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
11 #include "gtest/gtest.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
12
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
13 namespace llvm {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
14
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
15 namespace {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
16 struct SimpleRefCounted : public RefCountedBase<SimpleRefCounted> {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
17 SimpleRefCounted() { ++NumInstances; }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
18 SimpleRefCounted(const SimpleRefCounted &) : RefCountedBase() {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
19 ++NumInstances;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
20 }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
21 ~SimpleRefCounted() { --NumInstances; }
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
22
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
23 static int NumInstances;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
24 };
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
25 int SimpleRefCounted::NumInstances = 0;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
26 } // anonymous namespace
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
27
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
28 TEST(IntrusiveRefCntPtr, RefCountedBaseCopyDoesNotLeak) {
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
29 EXPECT_EQ(0, SimpleRefCounted::NumInstances);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
30 {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
31 SimpleRefCounted *S1 = new SimpleRefCounted;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
32 IntrusiveRefCntPtr<SimpleRefCounted> R1 = S1;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
33 SimpleRefCounted *S2 = new SimpleRefCounted(*S1);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
34 IntrusiveRefCntPtr<SimpleRefCounted> R2 = S2;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
35 EXPECT_EQ(2, SimpleRefCounted::NumInstances);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
36 }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
37 EXPECT_EQ(0, SimpleRefCounted::NumInstances);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
38 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
39
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
40 struct InterceptRefCounted : public RefCountedBase<InterceptRefCounted> {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
41 InterceptRefCounted(bool *Released, bool *Retained)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
42 : Released(Released), Retained(Retained) {}
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
43 bool * const Released;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
44 bool * const Retained;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
45 };
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
46 template <> struct IntrusiveRefCntPtrInfo<InterceptRefCounted> {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
47 static void retain(InterceptRefCounted *I) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
48 *I->Retained = true;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
49 I->Retain();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
50 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
51 static void release(InterceptRefCounted *I) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
52 *I->Released = true;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
53 I->Release();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
54 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
55 };
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
56 TEST(IntrusiveRefCntPtr, UsesTraitsToRetainAndRelease) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
57 bool Released = false;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
58 bool Retained = false;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
59 {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
60 InterceptRefCounted *I = new InterceptRefCounted(&Released, &Retained);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
61 IntrusiveRefCntPtr<InterceptRefCounted> R = I;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
62 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
63 EXPECT_TRUE(Released);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
64 EXPECT_TRUE(Retained);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
65 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
66
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
67 } // end namespace llvm