Mercurial > hg > CbC > CbC_llvm

annotate unittests/ADT/IntrusiveRefCntPtrTest.cpp @ 147:c2174574ed3a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
LLVM 10
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Wed, 14 Aug 2019 16:55:33 +0900
parents 803732b1fca8
children
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 //
147
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 121
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 121
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 121
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
6 //
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 #include "llvm/ADT/IntrusiveRefCntPtr.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
10 #include "gtest/gtest.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
11
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
12 namespace llvm {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
13
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
14 namespace {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
15 struct SimpleRefCounted : public RefCountedBase<SimpleRefCounted> {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
16 SimpleRefCounted() { ++NumInstances; }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
17 SimpleRefCounted(const SimpleRefCounted &) : RefCountedBase() {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
18 ++NumInstances;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
19 }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
20 ~SimpleRefCounted() { --NumInstances; }
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
21
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
22 static int NumInstances;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
23 };
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
24 int SimpleRefCounted::NumInstances = 0;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
25 } // anonymous namespace
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
26
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
27 TEST(IntrusiveRefCntPtr, RefCountedBaseCopyDoesNotLeak) {
121
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
28 EXPECT_EQ(0, SimpleRefCounted::NumInstances);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
29 {
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
30 SimpleRefCounted *S1 = new SimpleRefCounted;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
31 IntrusiveRefCntPtr<SimpleRefCounted> R1 = S1;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
32 SimpleRefCounted *S2 = new SimpleRefCounted(*S1);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
33 IntrusiveRefCntPtr<SimpleRefCounted> R2 = S2;
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
34 EXPECT_EQ(2, SimpleRefCounted::NumInstances);
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
35 }
803732b1fca8 LLVM 5.0
kono
parents: 0
diff changeset
36 EXPECT_EQ(0, SimpleRefCounted::NumInstances);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
37 }
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 struct InterceptRefCounted : public RefCountedBase<InterceptRefCounted> {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
40 InterceptRefCounted(bool *Released, bool *Retained)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
41 : Released(Released), Retained(Retained) {}
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
42 bool * const Released;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
43 bool * const Retained;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
44 };
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
45 template <> struct IntrusiveRefCntPtrInfo<InterceptRefCounted> {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
46 static void retain(InterceptRefCounted *I) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
47 *I->Retained = true;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
48 I->Retain();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
49 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
50 static void release(InterceptRefCounted *I) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
51 *I->Released = true;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
52 I->Release();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
53 }
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 TEST(IntrusiveRefCntPtr, UsesTraitsToRetainAndRelease) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
56 bool Released = false;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
57 bool Retained = false;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
58 {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
59 InterceptRefCounted *I = new InterceptRefCounted(&Released, &Retained);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
60 IntrusiveRefCntPtr<InterceptRefCounted> R = I;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
61 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
62 EXPECT_TRUE(Released);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
63 EXPECT_TRUE(Retained);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
64 }
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 } // end namespace llvm