| /* |
| * Copyright 2009 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. |
| */ |
| |
| #ifndef WEBRTC_BASE_SOCKET_UNITTEST_H_ |
| #define WEBRTC_BASE_SOCKET_UNITTEST_H_ |
| |
| #include "webrtc/base/gunit.h" |
| #include "webrtc/base/thread.h" |
| |
| namespace rtc { |
| |
| // Generic socket tests, to be used when testing individual socketservers. |
| // Derive your specific test class from SocketTest, install your |
| // socketserver, and call the SocketTest test methods. |
| class SocketTest : public testing::Test { |
| protected: |
| SocketTest() : kIPv4Loopback(INADDR_LOOPBACK), |
| kIPv6Loopback(in6addr_loopback), |
| ss_(nullptr) {} |
| virtual void SetUp() { ss_ = Thread::Current()->socketserver(); } |
| void TestConnectIPv4(); |
| void TestConnectIPv6(); |
| void TestConnectWithDnsLookupIPv4(); |
| void TestConnectWithDnsLookupIPv6(); |
| void TestConnectFailIPv4(); |
| void TestConnectFailIPv6(); |
| void TestConnectWithDnsLookupFailIPv4(); |
| void TestConnectWithDnsLookupFailIPv6(); |
| void TestConnectWithClosedSocketIPv4(); |
| void TestConnectWithClosedSocketIPv6(); |
| void TestConnectWhileNotClosedIPv4(); |
| void TestConnectWhileNotClosedIPv6(); |
| void TestServerCloseDuringConnectIPv4(); |
| void TestServerCloseDuringConnectIPv6(); |
| void TestClientCloseDuringConnectIPv4(); |
| void TestClientCloseDuringConnectIPv6(); |
| void TestServerCloseIPv4(); |
| void TestServerCloseIPv6(); |
| void TestCloseInClosedCallbackIPv4(); |
| void TestCloseInClosedCallbackIPv6(); |
| void TestSocketServerWaitIPv4(); |
| void TestSocketServerWaitIPv6(); |
| void TestTcpIPv4(); |
| void TestTcpIPv6(); |
| void TestSingleFlowControlCallbackIPv4(); |
| void TestSingleFlowControlCallbackIPv6(); |
| void TestUdpIPv4(); |
| void TestUdpIPv6(); |
| void TestUdpReadyToSendIPv4(); |
| void TestUdpReadyToSendIPv6(); |
| void TestGetSetOptionsIPv4(); |
| void TestGetSetOptionsIPv6(); |
| void TestSocketRecvTimestampIPv4(); |
| void TestSocketRecvTimestampIPv6(); |
| |
| static const int kTimeout = 5000; // ms |
| const IPAddress kIPv4Loopback; |
| const IPAddress kIPv6Loopback; |
| |
| protected: |
| void TcpInternal(const IPAddress& loopback, size_t data_size, |
| ptrdiff_t max_send_size); |
| |
| private: |
| void ConnectInternal(const IPAddress& loopback); |
| void ConnectWithDnsLookupInternal(const IPAddress& loopback, |
| const std::string& host); |
| void ConnectFailInternal(const IPAddress& loopback); |
| |
| void ConnectWithDnsLookupFailInternal(const IPAddress& loopback); |
| void ConnectWithClosedSocketInternal(const IPAddress& loopback); |
| void ConnectWhileNotClosedInternal(const IPAddress& loopback); |
| void ServerCloseDuringConnectInternal(const IPAddress& loopback); |
| void ClientCloseDuringConnectInternal(const IPAddress& loopback); |
| void ServerCloseInternal(const IPAddress& loopback); |
| void CloseInClosedCallbackInternal(const IPAddress& loopback); |
| void SocketServerWaitInternal(const IPAddress& loopback); |
| void SingleFlowControlCallbackInternal(const IPAddress& loopback); |
| void UdpInternal(const IPAddress& loopback); |
| void UdpReadyToSend(const IPAddress& loopback); |
| void GetSetOptionsInternal(const IPAddress& loopback); |
| void SocketRecvTimestamp(const IPAddress& loopback); |
| |
| SocketServer* ss_; |
| }; |
| |
| // For unbound sockets, GetLocalAddress / GetRemoteAddress return AF_UNSPEC |
| // values on Windows, but an empty address of the same family on Linux/MacOS X. |
| bool IsUnspecOrEmptyIP(const IPAddress& address); |
| |
| } // namespace rtc |
| |
| #endif // WEBRTC_BASE_SOCKET_UNITTEST_H_ |