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 "rtc_base/memory/always_valid_pointer.h"
12
13 #include <string>
14
15 #include "test/gtest.h"
16
17 namespace webrtc {
18
TEST(AlwaysValidPointerTest,DefaultToEmptyValue)19 TEST(AlwaysValidPointerTest, DefaultToEmptyValue) {
20 AlwaysValidPointer<std::string> ptr(nullptr);
21 EXPECT_EQ(*ptr, "");
22 }
TEST(AlwaysValidPointerTest,DefaultWithForwardedArgument)23 TEST(AlwaysValidPointerTest, DefaultWithForwardedArgument) {
24 AlwaysValidPointer<std::string> ptr(nullptr, "test");
25 EXPECT_EQ(*ptr, "test");
26 }
TEST(AlwaysValidPointerTest,DefaultToSubclass)27 TEST(AlwaysValidPointerTest, DefaultToSubclass) {
28 struct A {
29 virtual ~A() {}
30 virtual int f() = 0;
31 };
32 struct B : public A {
33 int b = 0;
34 explicit B(int val) : b(val) {}
35 virtual ~B() {}
36 int f() override { return b; }
37 };
38 AlwaysValidPointer<A, B> ptr(nullptr, 3);
39 EXPECT_EQ(ptr->f(), 3);
40 EXPECT_EQ((*ptr).f(), 3);
41 EXPECT_EQ(ptr.get()->f(), 3);
42 }
TEST(AlwaysValidPointerTest,NonDefaultValue)43 TEST(AlwaysValidPointerTest, NonDefaultValue) {
44 std::string str("keso");
45 AlwaysValidPointer<std::string> ptr(&str, "test");
46 EXPECT_EQ(*ptr, "keso");
47 }
48
TEST(AlwaysValidPointerTest,TakeOverOwnershipOfInstance)49 TEST(AlwaysValidPointerTest, TakeOverOwnershipOfInstance) {
50 std::string str("keso");
51 std::unique_ptr<std::string> str2 = std::make_unique<std::string>("kent");
52 AlwaysValidPointer<std::string> ptr(std::move(str2), &str);
53 EXPECT_EQ(*ptr, "kent");
54 EXPECT_EQ(str2, nullptr);
55 }
56
TEST(AlwaysValidPointerTest,TakeOverOwnershipFallbackOnPointer)57 TEST(AlwaysValidPointerTest, TakeOverOwnershipFallbackOnPointer) {
58 std::string str("keso");
59 std::unique_ptr<std::string> str2;
60 AlwaysValidPointer<std::string> ptr(std::move(str2), &str);
61 EXPECT_EQ(*ptr, "keso");
62 }
63
TEST(AlwaysValidPointerTest,TakeOverOwnershipFallbackOnDefault)64 TEST(AlwaysValidPointerTest, TakeOverOwnershipFallbackOnDefault) {
65 std::unique_ptr<std::string> str;
66 std::string* str_ptr = nullptr;
67 AlwaysValidPointer<std::string> ptr(std::move(str), str_ptr);
68 EXPECT_EQ(*ptr, "");
69 }
70
TEST(AlwaysValidPointerTest,TakeOverOwnershipFallbackOnDefaultWithForwardedArgument)71 TEST(AlwaysValidPointerTest,
72 TakeOverOwnershipFallbackOnDefaultWithForwardedArgument) {
73 std::unique_ptr<std::string> str2;
74 AlwaysValidPointer<std::string> ptr(std::move(str2), nullptr, "keso");
75 EXPECT_EQ(*ptr, "keso");
76 }
77
TEST(AlwaysValidPointerTest,TakeOverOwnershipDoesNotForwardDefaultArguments)78 TEST(AlwaysValidPointerTest, TakeOverOwnershipDoesNotForwardDefaultArguments) {
79 std::unique_ptr<std::string> str = std::make_unique<std::string>("kalle");
80 std::unique_ptr<std::string> str2 = std::make_unique<std::string>("anka");
81 AlwaysValidPointer<std::string> ptr(std::move(str), nullptr, *str2);
82 EXPECT_EQ(*ptr, "kalle");
83 EXPECT_TRUE(!str);
84 EXPECT_EQ(*str2, "anka");
85 }
86
TEST(AlwaysValidPointerTest,DefaultToLambda)87 TEST(AlwaysValidPointerTest, DefaultToLambda) {
88 AlwaysValidPointer<std::string> ptr(
89 nullptr, []() { return std::make_unique<std::string>("onkel skrue"); });
90 EXPECT_EQ(*ptr, "onkel skrue");
91 }
92
TEST(AlwaysValidPointerTest,NoDefaultObjectPassValidPointer)93 TEST(AlwaysValidPointerTest, NoDefaultObjectPassValidPointer) {
94 std::string str("foo");
95 AlwaysValidPointerNoDefault<std::string> ptr(&str);
96 EXPECT_EQ(*ptr, "foo");
97 EXPECT_EQ(ptr, &str);
98 }
99
TEST(AlwaysValidPointerTest,NoDefaultObjectWithTakeOverOwnership)100 TEST(AlwaysValidPointerTest, NoDefaultObjectWithTakeOverOwnership) {
101 std::unique_ptr<std::string> str = std::make_unique<std::string>("yum");
102 AlwaysValidPointerNoDefault<std::string> ptr(std::move(str));
103 EXPECT_EQ(*ptr, "yum");
104 std::unique_ptr<std::string> str2 = std::make_unique<std::string>("fun");
105 AlwaysValidPointerNoDefault<std::string> ptr2(std::move(str), str2.get());
106 EXPECT_EQ(*ptr2, "fun");
107 EXPECT_EQ(ptr2, str2.get());
108 }
109
110 #if GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID)
111
TEST(AlwaysValidPointerTest,NoDefaultObjectPassNullPointer)112 TEST(AlwaysValidPointerTest, NoDefaultObjectPassNullPointer) {
113 auto pass_null = []() {
114 AlwaysValidPointerNoDefault<std::string> ptr(nullptr);
115 };
116 EXPECT_DEATH(pass_null(), "");
117 }
118
TEST(AlwaysValidPointerTest,NoDefaultObjectPassNullUniquePointer)119 TEST(AlwaysValidPointerTest, NoDefaultObjectPassNullUniquePointer) {
120 auto pass_null = []() {
121 std::unique_ptr<std::string> str;
122 AlwaysValidPointerNoDefault<std::string> ptr(std::move(str));
123 };
124 EXPECT_DEATH(pass_null(), "");
125 }
126
127 #endif
128
129 } // namespace webrtc
130