| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #ifndef WEBRTC_BASE_TESTCLIENT_H_ | 11 #ifndef WEBRTC_BASE_TESTCLIENT_H_ |
| 12 #define WEBRTC_BASE_TESTCLIENT_H_ | 12 #define WEBRTC_BASE_TESTCLIENT_H_ |
| 13 | 13 |
| 14 #include <memory> | |
| 15 #include <vector> | |
| 16 #include "webrtc/base/asyncudpsocket.h" | |
| 17 #include "webrtc/base/constructormagic.h" | |
| 18 #include "webrtc/base/criticalsection.h" | |
| 19 #include "webrtc/base/fakeclock.h" | |
| 20 | 14 |
| 21 namespace rtc { | 15 // This header is deprecated and is just left here temporarily during |
| 22 | 16 // refactoring. See https://bugs.webrtc.org/7634 for more details. |
| 23 // A simple client that can send TCP or UDP data and check that it receives | 17 #include "webrtc/rtc_base/testclient.h" |
| 24 // what it expects to receive. Useful for testing server functionality. | |
| 25 class TestClient : public sigslot::has_slots<> { | |
| 26 public: | |
| 27 // Records the contents of a packet that was received. | |
| 28 struct Packet { | |
| 29 Packet(const SocketAddress& a, | |
| 30 const char* b, | |
| 31 size_t s, | |
| 32 const PacketTime& packet_time); | |
| 33 Packet(const Packet& p); | |
| 34 virtual ~Packet(); | |
| 35 | |
| 36 SocketAddress addr; | |
| 37 char* buf; | |
| 38 size_t size; | |
| 39 PacketTime packet_time; | |
| 40 }; | |
| 41 | |
| 42 // Default timeout for NextPacket reads. | |
| 43 static const int kTimeoutMs = 5000; | |
| 44 | |
| 45 // Creates a client that will send and receive with the given socket and | |
| 46 // will post itself messages with the given thread. | |
| 47 explicit TestClient(std::unique_ptr<AsyncPacketSocket> socket); | |
| 48 // Create a test client that will use a fake clock. NextPacket needs to wait | |
| 49 // for a packet to be received, and thus it needs to advance the fake clock | |
| 50 // if the test is using one, rather than just sleeping. | |
| 51 TestClient(std::unique_ptr<AsyncPacketSocket> socket, FakeClock* fake_clock); | |
| 52 ~TestClient() override; | |
| 53 | |
| 54 SocketAddress address() const { return socket_->GetLocalAddress(); } | |
| 55 SocketAddress remote_address() const { return socket_->GetRemoteAddress(); } | |
| 56 | |
| 57 // Checks that the socket moves to the specified connect state. | |
| 58 bool CheckConnState(AsyncPacketSocket::State state); | |
| 59 | |
| 60 // Checks that the socket is connected to the remote side. | |
| 61 bool CheckConnected() { | |
| 62 return CheckConnState(AsyncPacketSocket::STATE_CONNECTED); | |
| 63 } | |
| 64 | |
| 65 // Sends using the clients socket. | |
| 66 int Send(const char* buf, size_t size); | |
| 67 | |
| 68 // Sends using the clients socket to the given destination. | |
| 69 int SendTo(const char* buf, size_t size, const SocketAddress& dest); | |
| 70 | |
| 71 // Returns the next packet received by the client or null if none is received | |
| 72 // within the specified timeout. | |
| 73 std::unique_ptr<Packet> NextPacket(int timeout_ms); | |
| 74 | |
| 75 // Checks that the next packet has the given contents. Returns the remote | |
| 76 // address that the packet was sent from. | |
| 77 bool CheckNextPacket(const char* buf, size_t len, SocketAddress* addr); | |
| 78 | |
| 79 // Checks that no packets have arrived or will arrive in the next second. | |
| 80 bool CheckNoPacket(); | |
| 81 | |
| 82 int GetError(); | |
| 83 int SetOption(Socket::Option opt, int value); | |
| 84 | |
| 85 bool ready_to_send() const { return ready_to_send_count() > 0; } | |
| 86 | |
| 87 // How many times SignalReadyToSend has been fired. | |
| 88 int ready_to_send_count() const { return ready_to_send_count_; } | |
| 89 | |
| 90 private: | |
| 91 // Timeout for reads when no packet is expected. | |
| 92 static const int kNoPacketTimeoutMs = 1000; | |
| 93 // Workaround for the fact that AsyncPacketSocket::GetConnState doesn't exist. | |
| 94 Socket::ConnState GetState(); | |
| 95 // Slot for packets read on the socket. | |
| 96 void OnPacket(AsyncPacketSocket* socket, const char* buf, size_t len, | |
| 97 const SocketAddress& remote_addr, | |
| 98 const PacketTime& packet_time); | |
| 99 void OnReadyToSend(AsyncPacketSocket* socket); | |
| 100 bool CheckTimestamp(int64_t packet_timestamp); | |
| 101 void AdvanceTime(int ms); | |
| 102 | |
| 103 FakeClock* fake_clock_ = nullptr; | |
| 104 CriticalSection crit_; | |
| 105 std::unique_ptr<AsyncPacketSocket> socket_; | |
| 106 std::vector<std::unique_ptr<Packet>> packets_; | |
| 107 int ready_to_send_count_ = 0; | |
| 108 int64_t prev_packet_timestamp_; | |
| 109 RTC_DISALLOW_COPY_AND_ASSIGN(TestClient); | |
| 110 }; | |
| 111 | |
| 112 } // namespace rtc | |
| 113 | 18 |
| 114 #endif // WEBRTC_BASE_TESTCLIENT_H_ | 19 #endif // WEBRTC_BASE_TESTCLIENT_H_ |
| OLD | NEW |