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/async_tcp_socket.h" 12 13 #include <memory> 14 #include <string> 15 16 #include "rtc_base/gunit.h" 17 #include "rtc_base/virtual_socket_server.h" 18 19 namespace rtc { 20 21 class AsyncTCPSocketTest : public ::testing::Test, public sigslot::has_slots<> { 22 public: AsyncTCPSocketTest()23 AsyncTCPSocketTest() 24 : vss_(new rtc::VirtualSocketServer()), 25 socket_(vss_->CreateAsyncSocket(SOCK_STREAM)), 26 tcp_socket_(new AsyncTCPSocket(socket_, true)), 27 ready_to_send_(false) { 28 tcp_socket_->SignalReadyToSend.connect(this, 29 &AsyncTCPSocketTest::OnReadyToSend); 30 } 31 OnReadyToSend(rtc::AsyncPacketSocket * socket)32 void OnReadyToSend(rtc::AsyncPacketSocket* socket) { ready_to_send_ = true; } 33 34 protected: 35 std::unique_ptr<VirtualSocketServer> vss_; 36 AsyncSocket* socket_; 37 std::unique_ptr<AsyncTCPSocket> tcp_socket_; 38 bool ready_to_send_; 39 }; 40 TEST_F(AsyncTCPSocketTest,OnWriteEvent)41TEST_F(AsyncTCPSocketTest, OnWriteEvent) { 42 EXPECT_FALSE(ready_to_send_); 43 socket_->SignalWriteEvent(socket_); 44 EXPECT_TRUE(ready_to_send_); 45 } 46 47 } // namespace rtc 48