1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
//===--- ThreadSafeRefCntPointerTest.cpp ----------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/Basic/ThreadSafeRefCounted.h"
#include "llvm/ADT/IntrusiveRefCntPtr.h"
#include "gtest/gtest.h"
using llvm::IntrusiveRefCntPtr;
struct TestRelease : llvm::ThreadSafeRefCountedBase<TestRelease> {
bool &destroy;
TestRelease(bool &destroy) : destroy(destroy) {}
~TestRelease() { destroy = true; }
};
TEST(ThreadSafeRefCountedBase, ReleaseSimple) {
bool destroyed = false;
{
IntrusiveRefCntPtr<TestRelease> ref = new TestRelease(destroyed);
}
EXPECT_TRUE(destroyed);
}
TEST(ThreadSafeRefCountedBase, Release) {
bool destroyed = false;
{
IntrusiveRefCntPtr<TestRelease> ref = new TestRelease(destroyed);
ref->Retain();
ref->Release();
}
EXPECT_TRUE(destroyed);
}
struct TestReleaseVPTR : swift::ThreadSafeRefCountedBaseVPTR {
bool &destroy;
TestReleaseVPTR(bool &destroy) : destroy(destroy) {}
~TestReleaseVPTR() override { destroy = true; }
};
TEST(ThreadSafeRefCountedBaseVPTR, ReleaseSimple) {
bool destroyed = false;
{
IntrusiveRefCntPtr<TestReleaseVPTR> ref = new TestReleaseVPTR(destroyed);
}
EXPECT_TRUE(destroyed);
}
TEST(ThreadSafeRefCountedBaseVPTR, Release) {
bool destroyed = false;
{
IntrusiveRefCntPtr<TestReleaseVPTR> ref = new TestReleaseVPTR(destroyed);
ref->Retain();
ref->Release();
}
EXPECT_TRUE(destroyed);
}
|
