b7239a9dc8
The refcount.h file doesn't depend on anything from refcountedobject.h. The motivation of this change to make it possible to add additional declarations to refcount.h, and include it from refcountedobject.h. Bug: webrtc:8270 Change-Id: I24f6131f471e675570968d00065ff9b1f55e3373 Reviewed-on: https://webrtc-review.googlesource.com/5760 Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Commit-Queue: Niels Moller <nisse@webrtc.org> Cr-Commit-Position: refs/heads/master@{#20106}
93 lines
2.1 KiB
C++
93 lines
2.1 KiB
C++
/*
|
|
* Copyright 2016 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 <string>
|
|
|
|
#include "rtc_base/gunit.h"
|
|
#include "rtc_base/refcount.h"
|
|
#include "rtc_base/refcountedobject.h"
|
|
|
|
namespace rtc {
|
|
|
|
namespace {
|
|
|
|
class A {
|
|
public:
|
|
A() {}
|
|
|
|
private:
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(A);
|
|
};
|
|
|
|
class RefClass : public RefCountInterface {
|
|
public:
|
|
RefClass() {}
|
|
|
|
protected:
|
|
~RefClass() override {}
|
|
};
|
|
|
|
class RefClassWithRvalue : public RefCountInterface {
|
|
public:
|
|
explicit RefClassWithRvalue(std::unique_ptr<A> a) : a_(std::move(a)) {}
|
|
|
|
protected:
|
|
~RefClassWithRvalue() override {}
|
|
|
|
public:
|
|
std::unique_ptr<A> a_;
|
|
};
|
|
|
|
class RefClassWithMixedValues : public RefCountInterface {
|
|
public:
|
|
RefClassWithMixedValues(std::unique_ptr<A> a, int b, const std::string& c)
|
|
: a_(std::move(a)), b_(b), c_(c) {}
|
|
|
|
protected:
|
|
~RefClassWithMixedValues() override {}
|
|
|
|
public:
|
|
std::unique_ptr<A> a_;
|
|
int b_;
|
|
std::string c_;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
TEST(RefCountedObject, Basic) {
|
|
scoped_refptr<RefCountedObject<RefClass>> aref(
|
|
new RefCountedObject<RefClass>());
|
|
EXPECT_TRUE(aref->HasOneRef());
|
|
EXPECT_EQ(2, aref->AddRef());
|
|
EXPECT_EQ(1, aref->Release());
|
|
}
|
|
|
|
TEST(RefCountedObject, SupportRValuesInCtor) {
|
|
std::unique_ptr<A> a(new A());
|
|
scoped_refptr<RefClassWithRvalue> ref(
|
|
new RefCountedObject<RefClassWithRvalue>(std::move(a)));
|
|
EXPECT_TRUE(ref->a_.get() != nullptr);
|
|
EXPECT_TRUE(a.get() == nullptr);
|
|
}
|
|
|
|
TEST(RefCountedObject, SupportMixedTypesInCtor) {
|
|
std::unique_ptr<A> a(new A());
|
|
int b = 9;
|
|
std::string c = "hello";
|
|
scoped_refptr<RefClassWithMixedValues> ref(
|
|
new RefCountedObject<RefClassWithMixedValues>(std::move(a), b, c));
|
|
EXPECT_TRUE(ref->a_.get() != nullptr);
|
|
EXPECT_TRUE(a.get() == nullptr);
|
|
EXPECT_EQ(b, ref->b_);
|
|
EXPECT_EQ(c, ref->c_);
|
|
}
|
|
|
|
} // namespace rtc
|