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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#include "tensorflow/core/lib/core/refcount.h"
#include <gtest/gtest.h>
namespace tensorflow {
namespace core {
namespace {
static int constructed = 0;
static int destroyed = 0;
class MyRef : public RefCounted {
public:
MyRef() { constructed++; }
~MyRef() override { destroyed++; }
};
class RefTest : public testing::Test {
public:
RefTest() {
constructed = 0;
destroyed = 0;
}
};
TEST_F(RefTest, New) {
MyRef* ref = new MyRef;
ASSERT_EQ(1, constructed);
ASSERT_EQ(0, destroyed);
ref->Unref();
ASSERT_EQ(1, constructed);
ASSERT_EQ(1, destroyed);
}
TEST_F(RefTest, RefUnref) {
MyRef* ref = new MyRef;
ASSERT_EQ(1, constructed);
ASSERT_EQ(0, destroyed);
ref->Ref();
ASSERT_EQ(0, destroyed);
ref->Unref();
ASSERT_EQ(0, destroyed);
ref->Unref();
ASSERT_EQ(1, destroyed);
}
TEST_F(RefTest, RefCountOne) {
MyRef* ref = new MyRef;
ASSERT_TRUE(ref->RefCountIsOne());
ref->Unref();
}
TEST_F(RefTest, RefCountNotOne) {
MyRef* ref = new MyRef;
ref->Ref();
ASSERT_FALSE(ref->RefCountIsOne());
ref->Unref();
ref->Unref();
}
TEST_F(RefTest, ConstRefUnref) {
const MyRef* cref = new MyRef;
ASSERT_EQ(1, constructed);
ASSERT_EQ(0, destroyed);
cref->Ref();
ASSERT_EQ(0, destroyed);
cref->Unref();
ASSERT_EQ(0, destroyed);
cref->Unref();
ASSERT_EQ(1, destroyed);
}
TEST_F(RefTest, ReturnOfUnref) {
MyRef* ref = new MyRef;
ref->Ref();
EXPECT_FALSE(ref->Unref());
EXPECT_TRUE(ref->Unref());
}
TEST_F(RefTest, ScopedUnref) {
{ ScopedUnref unref(new MyRef); }
EXPECT_EQ(destroyed, 1);
}
TEST_F(RefTest, ScopedUnref_Nullptr) {
{ ScopedUnref unref(nullptr); }
EXPECT_EQ(destroyed, 0);
}
} // namespace
} // namespace core
} // namespace tensorflow
|
