mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
6f72f56b6c
AddRef() now returns void, and Release() returns an enum RefCountReleaseStatus, to indicate whether or not this Release call implied deletion. Bug: webrtc:8270 Change-Id: If2fb77f26118b61751b51c856af187c72112c630 Reviewed-on: https://webrtc-review.googlesource.com/3320 Commit-Queue: Niels Moller <nisse@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Reviewed-by: Taylor Brandstetter <deadbeef@webrtc.org> Cr-Commit-Position: refs/heads/master@{#20366}
140 lines
3.2 KiB
C++
140 lines
3.2 KiB
C++
/*
|
|
* Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "rtc_base/callback.h"
|
|
#include "rtc_base/bind.h"
|
|
#include "rtc_base/gunit.h"
|
|
#include "rtc_base/keep_ref_until_done.h"
|
|
#include "rtc_base/refcount.h"
|
|
|
|
namespace rtc {
|
|
|
|
namespace {
|
|
|
|
void f() {}
|
|
int g() { return 42; }
|
|
int h(int x) { return x * x; }
|
|
void i(int& x) { x *= x; } // NOLINT: Testing refs
|
|
|
|
struct BindTester {
|
|
int a() { return 24; }
|
|
int b(int x) const { return x * x; }
|
|
};
|
|
|
|
class RefCountedBindTester : public RefCountInterface {
|
|
public:
|
|
RefCountedBindTester() : count_(0) {}
|
|
void AddRef() const override { ++count_; }
|
|
RefCountReleaseStatus Release() const override {
|
|
--count_;
|
|
return count_ == 0 ? RefCountReleaseStatus::kDroppedLastRef
|
|
: RefCountReleaseStatus::kOtherRefsRemained;
|
|
}
|
|
int RefCount() const { return count_; }
|
|
|
|
private:
|
|
mutable int count_;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
TEST(CallbackTest, VoidReturn) {
|
|
Callback0<void> cb;
|
|
EXPECT_TRUE(cb.empty());
|
|
cb(); // Executing an empty callback should not crash.
|
|
cb = Callback0<void>(&f);
|
|
EXPECT_FALSE(cb.empty());
|
|
cb();
|
|
}
|
|
|
|
TEST(CallbackTest, IntReturn) {
|
|
Callback0<int> cb;
|
|
EXPECT_TRUE(cb.empty());
|
|
cb = Callback0<int>(&g);
|
|
EXPECT_FALSE(cb.empty());
|
|
EXPECT_EQ(42, cb());
|
|
EXPECT_EQ(42, cb());
|
|
}
|
|
|
|
TEST(CallbackTest, OneParam) {
|
|
Callback1<int, int> cb1(&h);
|
|
EXPECT_FALSE(cb1.empty());
|
|
EXPECT_EQ(9, cb1(-3));
|
|
EXPECT_EQ(100, cb1(10));
|
|
|
|
// Try clearing a callback.
|
|
cb1 = Callback1<int, int>();
|
|
EXPECT_TRUE(cb1.empty());
|
|
|
|
// Try a callback with a ref parameter.
|
|
Callback1<void, int&> cb2(&i);
|
|
int x = 3;
|
|
cb2(x);
|
|
EXPECT_EQ(9, x);
|
|
cb2(x);
|
|
EXPECT_EQ(81, x);
|
|
}
|
|
|
|
TEST(CallbackTest, WithBind) {
|
|
BindTester t;
|
|
Callback0<int> cb1 = Bind(&BindTester::a, &t);
|
|
EXPECT_EQ(24, cb1());
|
|
EXPECT_EQ(24, cb1());
|
|
cb1 = Bind(&BindTester::b, &t, 10);
|
|
EXPECT_EQ(100, cb1());
|
|
EXPECT_EQ(100, cb1());
|
|
cb1 = Bind(&BindTester::b, &t, 5);
|
|
EXPECT_EQ(25, cb1());
|
|
EXPECT_EQ(25, cb1());
|
|
}
|
|
|
|
TEST(KeepRefUntilDoneTest, simple) {
|
|
RefCountedBindTester t;
|
|
EXPECT_EQ(0, t.RefCount());
|
|
{
|
|
Callback0<void> cb = KeepRefUntilDone(&t);
|
|
EXPECT_EQ(1, t.RefCount());
|
|
cb();
|
|
EXPECT_EQ(1, t.RefCount());
|
|
cb();
|
|
EXPECT_EQ(1, t.RefCount());
|
|
}
|
|
EXPECT_EQ(0, t.RefCount());
|
|
}
|
|
|
|
TEST(KeepRefUntilDoneTest, copy) {
|
|
RefCountedBindTester t;
|
|
EXPECT_EQ(0, t.RefCount());
|
|
Callback0<void> cb2;
|
|
{
|
|
Callback0<void> cb = KeepRefUntilDone(&t);
|
|
EXPECT_EQ(1, t.RefCount());
|
|
cb2 = cb;
|
|
}
|
|
EXPECT_EQ(1, t.RefCount());
|
|
cb2 = Callback0<void>();
|
|
EXPECT_EQ(0, t.RefCount());
|
|
}
|
|
|
|
TEST(KeepRefUntilDoneTest, scopedref) {
|
|
RefCountedBindTester t;
|
|
EXPECT_EQ(0, t.RefCount());
|
|
{
|
|
scoped_refptr<RefCountedBindTester> t_scoped_ref(&t);
|
|
Callback0<void> cb = KeepRefUntilDone(t_scoped_ref);
|
|
t_scoped_ref = nullptr;
|
|
EXPECT_EQ(1, t.RefCount());
|
|
cb();
|
|
EXPECT_EQ(1, t.RefCount());
|
|
}
|
|
EXPECT_EQ(0, t.RefCount());
|
|
}
|
|
|
|
} // namespace rtc
|