100 lines
3.6 KiB
C++
100 lines
3.6 KiB
C++
/*
|
|
* 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 TestSocketRecvTimestamp();
|
|
|
|
static const int kTimeout = 5000; // ms
|
|
const IPAddress kIPv4Loopback;
|
|
const IPAddress kIPv6Loopback;
|
|
|
|
protected:
|
|
void TcpInternal(const IPAddress& loopback, size_t data_size,
|
|
ssize_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_
|