| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright 2013 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/api/proxy.h" | |
| 12 | |
| 13 #include <memory> | |
| 14 #include <string> | |
| 15 | |
| 16 #include "webrtc/base/gunit.h" | |
| 17 #include "webrtc/base/refcount.h" | |
| 18 #include "webrtc/base/thread.h" | |
| 19 #include "webrtc/test/gmock.h" | |
| 20 | |
| 21 using ::testing::_; | |
| 22 using ::testing::DoAll; | |
| 23 using ::testing::Exactly; | |
| 24 using ::testing::InvokeWithoutArgs; | |
| 25 using ::testing::Return; | |
| 26 | |
| 27 namespace webrtc { | |
| 28 | |
| 29 // Interface used for testing here. | |
| 30 class FakeInterface : public rtc::RefCountInterface { | |
| 31 public: | |
| 32 virtual void VoidMethod0() = 0; | |
| 33 virtual std::string Method0() = 0; | |
| 34 virtual std::string ConstMethod0() const = 0; | |
| 35 virtual std::string Method1(std::string s) = 0; | |
| 36 virtual std::string ConstMethod1(std::string s) const = 0; | |
| 37 virtual std::string Method2(std::string s1, std::string s2) = 0; | |
| 38 | |
| 39 protected: | |
| 40 virtual ~FakeInterface() {} | |
| 41 }; | |
| 42 | |
| 43 // Implementation of the test interface. | |
| 44 class Fake : public FakeInterface { | |
| 45 public: | |
| 46 static rtc::scoped_refptr<Fake> Create() { | |
| 47 return new rtc::RefCountedObject<Fake>(); | |
| 48 } | |
| 49 // Used to verify destructor is called on the correct thread. | |
| 50 MOCK_METHOD0(Destroy, void()); | |
| 51 | |
| 52 MOCK_METHOD0(VoidMethod0, void()); | |
| 53 MOCK_METHOD0(Method0, std::string()); | |
| 54 MOCK_CONST_METHOD0(ConstMethod0, std::string()); | |
| 55 | |
| 56 MOCK_METHOD1(Method1, std::string(std::string)); | |
| 57 MOCK_CONST_METHOD1(ConstMethod1, std::string(std::string)); | |
| 58 | |
| 59 MOCK_METHOD2(Method2, std::string(std::string, std::string)); | |
| 60 | |
| 61 protected: | |
| 62 Fake() {} | |
| 63 ~Fake() { Destroy(); } | |
| 64 }; | |
| 65 | |
| 66 // Proxies for the test interface. | |
| 67 BEGIN_PROXY_MAP(Fake) | |
| 68 PROXY_WORKER_THREAD_DESTRUCTOR() | |
| 69 PROXY_METHOD0(void, VoidMethod0) | |
| 70 PROXY_METHOD0(std::string, Method0) | |
| 71 PROXY_CONSTMETHOD0(std::string, ConstMethod0) | |
| 72 PROXY_WORKER_METHOD1(std::string, Method1, std::string) | |
| 73 PROXY_CONSTMETHOD1(std::string, ConstMethod1, std::string) | |
| 74 PROXY_WORKER_METHOD2(std::string, Method2, std::string, std::string) | |
| 75 END_PROXY_MAP() | |
| 76 | |
| 77 // Preprocessor hack to get a proxy class a name different than FakeProxy. | |
| 78 #define FakeProxy FakeSignalingProxy | |
| 79 #define FakeProxyWithInternal FakeSignalingProxyWithInternal | |
| 80 BEGIN_SIGNALING_PROXY_MAP(Fake) | |
| 81 PROXY_SIGNALING_THREAD_DESTRUCTOR() | |
| 82 PROXY_METHOD0(void, VoidMethod0) | |
| 83 PROXY_METHOD0(std::string, Method0) | |
| 84 PROXY_CONSTMETHOD0(std::string, ConstMethod0) | |
| 85 PROXY_METHOD1(std::string, Method1, std::string) | |
| 86 PROXY_CONSTMETHOD1(std::string, ConstMethod1, std::string) | |
| 87 PROXY_METHOD2(std::string, Method2, std::string, std::string) | |
| 88 END_PROXY_MAP() | |
| 89 #undef FakeProxy | |
| 90 | |
| 91 class SignalingProxyTest : public testing::Test { | |
| 92 public: | |
| 93 // Checks that the functions are called on the right thread. | |
| 94 void CheckSignalingThread() { EXPECT_TRUE(signaling_thread_->IsCurrent()); } | |
| 95 | |
| 96 protected: | |
| 97 void SetUp() override { | |
| 98 signaling_thread_.reset(new rtc::Thread()); | |
| 99 ASSERT_TRUE(signaling_thread_->Start()); | |
| 100 fake_ = Fake::Create(); | |
| 101 fake_signaling_proxy_ = | |
| 102 FakeSignalingProxy::Create(signaling_thread_.get(), fake_.get()); | |
| 103 } | |
| 104 | |
| 105 protected: | |
| 106 std::unique_ptr<rtc::Thread> signaling_thread_; | |
| 107 rtc::scoped_refptr<FakeInterface> fake_signaling_proxy_; | |
| 108 rtc::scoped_refptr<Fake> fake_; | |
| 109 }; | |
| 110 | |
| 111 TEST_F(SignalingProxyTest, SignalingThreadDestructor) { | |
| 112 EXPECT_CALL(*fake_, Destroy()) | |
| 113 .Times(Exactly(1)) | |
| 114 .WillOnce( | |
| 115 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread)); | |
| 116 fake_ = nullptr; | |
| 117 fake_signaling_proxy_ = nullptr; | |
| 118 } | |
| 119 | |
| 120 TEST_F(SignalingProxyTest, VoidMethod0) { | |
| 121 EXPECT_CALL(*fake_, VoidMethod0()) | |
| 122 .Times(Exactly(1)) | |
| 123 .WillOnce( | |
| 124 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread)); | |
| 125 fake_signaling_proxy_->VoidMethod0(); | |
| 126 } | |
| 127 | |
| 128 TEST_F(SignalingProxyTest, Method0) { | |
| 129 EXPECT_CALL(*fake_, Method0()) | |
| 130 .Times(Exactly(1)) | |
| 131 .WillOnce(DoAll( | |
| 132 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread), | |
| 133 Return("Method0"))); | |
| 134 EXPECT_EQ("Method0", fake_signaling_proxy_->Method0()); | |
| 135 } | |
| 136 | |
| 137 TEST_F(SignalingProxyTest, ConstMethod0) { | |
| 138 EXPECT_CALL(*fake_, ConstMethod0()) | |
| 139 .Times(Exactly(1)) | |
| 140 .WillOnce(DoAll( | |
| 141 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread), | |
| 142 Return("ConstMethod0"))); | |
| 143 EXPECT_EQ("ConstMethod0", fake_signaling_proxy_->ConstMethod0()); | |
| 144 } | |
| 145 | |
| 146 TEST_F(SignalingProxyTest, Method1) { | |
| 147 const std::string arg1 = "arg1"; | |
| 148 EXPECT_CALL(*fake_, Method1(arg1)) | |
| 149 .Times(Exactly(1)) | |
| 150 .WillOnce(DoAll( | |
| 151 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread), | |
| 152 Return("Method1"))); | |
| 153 EXPECT_EQ("Method1", fake_signaling_proxy_->Method1(arg1)); | |
| 154 } | |
| 155 | |
| 156 TEST_F(SignalingProxyTest, ConstMethod1) { | |
| 157 const std::string arg1 = "arg1"; | |
| 158 EXPECT_CALL(*fake_, ConstMethod1(arg1)) | |
| 159 .Times(Exactly(1)) | |
| 160 .WillOnce(DoAll( | |
| 161 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread), | |
| 162 Return("ConstMethod1"))); | |
| 163 EXPECT_EQ("ConstMethod1", fake_signaling_proxy_->ConstMethod1(arg1)); | |
| 164 } | |
| 165 | |
| 166 TEST_F(SignalingProxyTest, Method2) { | |
| 167 const std::string arg1 = "arg1"; | |
| 168 const std::string arg2 = "arg2"; | |
| 169 EXPECT_CALL(*fake_, Method2(arg1, arg2)) | |
| 170 .Times(Exactly(1)) | |
| 171 .WillOnce(DoAll( | |
| 172 InvokeWithoutArgs(this, &SignalingProxyTest::CheckSignalingThread), | |
| 173 Return("Method2"))); | |
| 174 EXPECT_EQ("Method2", fake_signaling_proxy_->Method2(arg1, arg2)); | |
| 175 } | |
| 176 | |
| 177 class ProxyTest : public testing::Test { | |
| 178 public: | |
| 179 // Checks that the functions are called on the right thread. | |
| 180 void CheckSignalingThread() { EXPECT_TRUE(signaling_thread_->IsCurrent()); } | |
| 181 void CheckWorkerThread() { EXPECT_TRUE(worker_thread_->IsCurrent()); } | |
| 182 | |
| 183 protected: | |
| 184 void SetUp() override { | |
| 185 signaling_thread_.reset(new rtc::Thread()); | |
| 186 worker_thread_.reset(new rtc::Thread()); | |
| 187 ASSERT_TRUE(signaling_thread_->Start()); | |
| 188 ASSERT_TRUE(worker_thread_->Start()); | |
| 189 fake_ = Fake::Create(); | |
| 190 fake_proxy_ = FakeProxy::Create(signaling_thread_.get(), | |
| 191 worker_thread_.get(), fake_.get()); | |
| 192 } | |
| 193 | |
| 194 protected: | |
| 195 std::unique_ptr<rtc::Thread> signaling_thread_; | |
| 196 std::unique_ptr<rtc::Thread> worker_thread_; | |
| 197 rtc::scoped_refptr<FakeInterface> fake_proxy_; | |
| 198 rtc::scoped_refptr<Fake> fake_; | |
| 199 }; | |
| 200 | |
| 201 TEST_F(ProxyTest, WorkerThreadDestructor) { | |
| 202 EXPECT_CALL(*fake_, Destroy()) | |
| 203 .Times(Exactly(1)) | |
| 204 .WillOnce(InvokeWithoutArgs(this, &ProxyTest::CheckWorkerThread)); | |
| 205 fake_ = nullptr; | |
| 206 fake_proxy_ = nullptr; | |
| 207 } | |
| 208 | |
| 209 TEST_F(ProxyTest, VoidMethod0) { | |
| 210 EXPECT_CALL(*fake_, VoidMethod0()) | |
| 211 .Times(Exactly(1)) | |
| 212 .WillOnce(InvokeWithoutArgs(this, &ProxyTest::CheckSignalingThread)); | |
| 213 fake_proxy_->VoidMethod0(); | |
| 214 } | |
| 215 | |
| 216 TEST_F(ProxyTest, Method0) { | |
| 217 EXPECT_CALL(*fake_, Method0()) | |
| 218 .Times(Exactly(1)) | |
| 219 .WillOnce( | |
| 220 DoAll(InvokeWithoutArgs(this, &ProxyTest::CheckSignalingThread), | |
| 221 Return("Method0"))); | |
| 222 EXPECT_EQ("Method0", | |
| 223 fake_proxy_->Method0()); | |
| 224 } | |
| 225 | |
| 226 TEST_F(ProxyTest, ConstMethod0) { | |
| 227 EXPECT_CALL(*fake_, ConstMethod0()) | |
| 228 .Times(Exactly(1)) | |
| 229 .WillOnce( | |
| 230 DoAll(InvokeWithoutArgs(this, &ProxyTest::CheckSignalingThread), | |
| 231 Return("ConstMethod0"))); | |
| 232 EXPECT_EQ("ConstMethod0", | |
| 233 fake_proxy_->ConstMethod0()); | |
| 234 } | |
| 235 | |
| 236 TEST_F(ProxyTest, WorkerMethod1) { | |
| 237 const std::string arg1 = "arg1"; | |
| 238 EXPECT_CALL(*fake_, Method1(arg1)) | |
| 239 .Times(Exactly(1)) | |
| 240 .WillOnce(DoAll(InvokeWithoutArgs(this, &ProxyTest::CheckWorkerThread), | |
| 241 Return("Method1"))); | |
| 242 EXPECT_EQ("Method1", fake_proxy_->Method1(arg1)); | |
| 243 } | |
| 244 | |
| 245 TEST_F(ProxyTest, ConstMethod1) { | |
| 246 const std::string arg1 = "arg1"; | |
| 247 EXPECT_CALL(*fake_, ConstMethod1(arg1)) | |
| 248 .Times(Exactly(1)) | |
| 249 .WillOnce( | |
| 250 DoAll(InvokeWithoutArgs(this, &ProxyTest::CheckSignalingThread), | |
| 251 Return("ConstMethod1"))); | |
| 252 EXPECT_EQ("ConstMethod1", fake_proxy_->ConstMethod1(arg1)); | |
| 253 } | |
| 254 | |
| 255 TEST_F(ProxyTest, WorkerMethod2) { | |
| 256 const std::string arg1 = "arg1"; | |
| 257 const std::string arg2 = "arg2"; | |
| 258 EXPECT_CALL(*fake_, Method2(arg1, arg2)) | |
| 259 .Times(Exactly(1)) | |
| 260 .WillOnce(DoAll(InvokeWithoutArgs(this, &ProxyTest::CheckWorkerThread), | |
| 261 Return("Method2"))); | |
| 262 EXPECT_EQ("Method2", fake_proxy_->Method2(arg1, arg2)); | |
| 263 } | |
| 264 | |
| 265 // Interface for testing OWNED_PROXY_MAP. | |
| 266 class FooInterface { | |
| 267 public: | |
| 268 virtual ~FooInterface() {} | |
| 269 virtual void Bar() = 0; | |
| 270 }; | |
| 271 | |
| 272 class Foo : public FooInterface { | |
| 273 public: | |
| 274 Foo() {} | |
| 275 MOCK_METHOD0(Bar, void()); | |
| 276 }; | |
| 277 | |
| 278 BEGIN_OWNED_PROXY_MAP(Foo) | |
| 279 PROXY_SIGNALING_THREAD_DESTRUCTOR() | |
| 280 PROXY_METHOD0(void, Bar) | |
| 281 END_PROXY_MAP() | |
| 282 | |
| 283 class OwnedProxyTest : public testing::Test { | |
| 284 public: | |
| 285 OwnedProxyTest() | |
| 286 : foo_(new Foo()), | |
| 287 foo_proxy_( | |
| 288 FooProxy::Create(&signaling_thread_, &worker_thread_, foo_)) { | |
| 289 signaling_thread_.Start(); | |
| 290 worker_thread_.Start(); | |
| 291 } | |
| 292 | |
| 293 void CheckSignalingThread() { EXPECT_TRUE(signaling_thread_.IsCurrent()); } | |
| 294 void CheckWorkerThread() { EXPECT_TRUE(worker_thread_.IsCurrent()); } | |
| 295 | |
| 296 protected: | |
| 297 rtc::Thread signaling_thread_; | |
| 298 rtc::Thread worker_thread_; | |
| 299 Foo* foo_; // Owned by foo_proxy_, not this class. | |
| 300 std::unique_ptr<FooInterface> foo_proxy_; | |
| 301 }; | |
| 302 | |
| 303 // Just tests that a method can be invoked using an "owned proxy" (as opposed | |
| 304 // to normal ref-counted version). | |
| 305 TEST_F(OwnedProxyTest, BasicTest) { | |
| 306 EXPECT_CALL(*foo_, Bar()) | |
| 307 .Times(Exactly(1)) | |
| 308 .WillOnce(InvokeWithoutArgs(this, &OwnedProxyTest::CheckSignalingThread)); | |
| 309 foo_proxy_->Bar(); | |
| 310 } | |
| 311 | |
| 312 } // namespace webrtc | |
| OLD | NEW |