| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | |
| 3 * | |
| 4 * Use of this source code is governed by a BSD-style license | |
| 5 * that can be found in the LICENSE file in the root of the source | |
| 6 * tree. An additional intellectual property rights grant can be found | |
| 7 * in the file PATENTS. All contributing project authors may | |
| 8 * be found in the AUTHORS file in the root of the source tree. | |
| 9 */ | |
| 10 | |
| 11 #include "webrtc/base/bind.h" | |
| 12 #include "webrtc/base/callback.h" | |
| 13 #include "webrtc/base/gunit.h" | |
| 14 #include "webrtc/base/keep_ref_until_done.h" | |
| 15 #include "webrtc/base/refcount.h" | |
| 16 | |
| 17 namespace rtc { | |
| 18 | |
| 19 namespace { | |
| 20 | |
| 21 void f() {} | |
| 22 int g() { return 42; } | |
| 23 int h(int x) { return x * x; } | |
| 24 void i(int& x) { x *= x; } // NOLINT: Testing refs | |
| 25 | |
| 26 struct BindTester { | |
| 27 int a() { return 24; } | |
| 28 int b(int x) const { return x * x; } | |
| 29 }; | |
| 30 | |
| 31 class RefCountedBindTester : public RefCountInterface { | |
| 32 public: | |
| 33 RefCountedBindTester() : count_(0) {} | |
| 34 int AddRef() const override { | |
| 35 return ++count_; | |
| 36 } | |
| 37 int Release() const override { | |
| 38 return --count_; | |
| 39 } | |
| 40 int RefCount() const { return count_; } | |
| 41 | |
| 42 private: | |
| 43 mutable int count_; | |
| 44 }; | |
| 45 | |
| 46 } // namespace | |
| 47 | |
| 48 TEST(CallbackTest, VoidReturn) { | |
| 49 Callback0<void> cb; | |
| 50 EXPECT_TRUE(cb.empty()); | |
| 51 cb(); // Executing an empty callback should not crash. | |
| 52 cb = Callback0<void>(&f); | |
| 53 EXPECT_FALSE(cb.empty()); | |
| 54 cb(); | |
| 55 } | |
| 56 | |
| 57 TEST(CallbackTest, IntReturn) { | |
| 58 Callback0<int> cb; | |
| 59 EXPECT_TRUE(cb.empty()); | |
| 60 cb = Callback0<int>(&g); | |
| 61 EXPECT_FALSE(cb.empty()); | |
| 62 EXPECT_EQ(42, cb()); | |
| 63 EXPECT_EQ(42, cb()); | |
| 64 } | |
| 65 | |
| 66 TEST(CallbackTest, OneParam) { | |
| 67 Callback1<int, int> cb1(&h); | |
| 68 EXPECT_FALSE(cb1.empty()); | |
| 69 EXPECT_EQ(9, cb1(-3)); | |
| 70 EXPECT_EQ(100, cb1(10)); | |
| 71 | |
| 72 // Try clearing a callback. | |
| 73 cb1 = Callback1<int, int>(); | |
| 74 EXPECT_TRUE(cb1.empty()); | |
| 75 | |
| 76 // Try a callback with a ref parameter. | |
| 77 Callback1<void, int&> cb2(&i); | |
| 78 int x = 3; | |
| 79 cb2(x); | |
| 80 EXPECT_EQ(9, x); | |
| 81 cb2(x); | |
| 82 EXPECT_EQ(81, x); | |
| 83 } | |
| 84 | |
| 85 TEST(CallbackTest, WithBind) { | |
| 86 BindTester t; | |
| 87 Callback0<int> cb1 = Bind(&BindTester::a, &t); | |
| 88 EXPECT_EQ(24, cb1()); | |
| 89 EXPECT_EQ(24, cb1()); | |
| 90 cb1 = Bind(&BindTester::b, &t, 10); | |
| 91 EXPECT_EQ(100, cb1()); | |
| 92 EXPECT_EQ(100, cb1()); | |
| 93 cb1 = Bind(&BindTester::b, &t, 5); | |
| 94 EXPECT_EQ(25, cb1()); | |
| 95 EXPECT_EQ(25, cb1()); | |
| 96 } | |
| 97 | |
| 98 TEST(KeepRefUntilDoneTest, simple) { | |
| 99 RefCountedBindTester t; | |
| 100 EXPECT_EQ(0, t.RefCount()); | |
| 101 { | |
| 102 Callback0<void> cb = KeepRefUntilDone(&t); | |
| 103 EXPECT_EQ(1, t.RefCount()); | |
| 104 cb(); | |
| 105 EXPECT_EQ(1, t.RefCount()); | |
| 106 cb(); | |
| 107 EXPECT_EQ(1, t.RefCount()); | |
| 108 } | |
| 109 EXPECT_EQ(0, t.RefCount()); | |
| 110 } | |
| 111 | |
| 112 TEST(KeepRefUntilDoneTest, copy) { | |
| 113 RefCountedBindTester t; | |
| 114 EXPECT_EQ(0, t.RefCount()); | |
| 115 Callback0<void> cb2; | |
| 116 { | |
| 117 Callback0<void> cb = KeepRefUntilDone(&t); | |
| 118 EXPECT_EQ(1, t.RefCount()); | |
| 119 cb2 = cb; | |
| 120 } | |
| 121 EXPECT_EQ(1, t.RefCount()); | |
| 122 cb2 = Callback0<void>(); | |
| 123 EXPECT_EQ(0, t.RefCount()); | |
| 124 } | |
| 125 | |
| 126 TEST(KeepRefUntilDoneTest, scopedref) { | |
| 127 RefCountedBindTester t; | |
| 128 EXPECT_EQ(0, t.RefCount()); | |
| 129 { | |
| 130 scoped_refptr<RefCountedBindTester> t_scoped_ref(&t); | |
| 131 Callback0<void> cb = KeepRefUntilDone(t_scoped_ref); | |
| 132 t_scoped_ref = nullptr; | |
| 133 EXPECT_EQ(1, t.RefCount()); | |
| 134 cb(); | |
| 135 EXPECT_EQ(1, t.RefCount()); | |
| 136 } | |
| 137 EXPECT_EQ(0, t.RefCount()); | |
| 138 } | |
| 139 | |
| 140 } // namespace rtc | |
| OLD | NEW |