| Index: net/socket/socket_test_util.h
|
| diff --git a/net/socket/socket_test_util.h b/net/socket/socket_test_util.h
|
| index fc393eb8924bac73fee506dd348271cd80523c6f..a09b18a344ea485e834553637b9e599b1e42f359 100644
|
| --- a/net/socket/socket_test_util.h
|
| +++ b/net/socket/socket_test_util.h
|
| @@ -498,7 +498,7 @@ class DeterministicMockTCPClientSocket;
|
| //
|
| // For examples of how to use this class, see:
|
| // deterministic_socket_data_unittests.cc
|
| -class DeterministicSocketData : public StaticSocketDataProvider {
|
| +class DeterministicSocketData {
|
| public:
|
| // The Delegate is an abstract interface which handles the communication from
|
| // the DeterministicSocketData to the Deterministic MockSocket. The
|
| @@ -530,7 +530,7 @@ class DeterministicSocketData : public StaticSocketDataProvider {
|
| size_t reads_count,
|
| MockWrite* writes,
|
| size_t writes_count);
|
| - ~DeterministicSocketData() override;
|
| + ~DeterministicSocketData();
|
|
|
| // Consume all the data up to the give stop point (via SetStop()).
|
| void Run();
|
| @@ -539,29 +539,31 @@ class DeterministicSocketData : public StaticSocketDataProvider {
|
| void RunFor(int steps);
|
|
|
| // Stop at step |seq|, which must be in the future.
|
| - virtual void SetStop(int seq);
|
| + void SetStop(int seq);
|
|
|
| // Stop |seq| steps after the current step.
|
| - virtual void StopAfter(int seq);
|
| + void StopAfter(int seq);
|
| +
|
| bool stopped() const { return stopped_; }
|
| void SetStopped(bool val) { stopped_ = val; }
|
| MockRead& current_read() { return current_read_; }
|
| MockWrite& current_write() { return current_write_; }
|
| int sequence_number() const { return sequence_number_; }
|
| void set_delegate(base::WeakPtr<Delegate> delegate) { delegate_ = delegate; }
|
| -
|
| - // StaticSocketDataProvider:
|
| + MockConnect connect_data() const { return connect_; }
|
| + void set_connect_data(const MockConnect& connect) { connect_ = connect; }
|
|
|
| // When the socket calls Read(), that calls OnRead(), and expects either
|
| // ERR_IO_PENDING or data.
|
| - MockRead OnRead() override;
|
| + MockRead OnRead();
|
|
|
| // When the socket calls Write(), it always completes synchronously. OnWrite()
|
| // checks to make sure the written data matches the expected data. The
|
| // callback will not be invoked until its sequence number is reached.
|
| - MockWriteResult OnWrite(const std::string& data) override;
|
| - void Reset() override;
|
| - void CompleteRead() override {}
|
| + MockWriteResult OnWrite(const std::string& data);
|
| +
|
| + bool AllReadDataConsumed() const;
|
| + bool AllWriteDataConsumed() const;
|
|
|
| private:
|
| // Invoke the read and write callbacks, if the timing is appropriate.
|
| @@ -573,7 +575,8 @@ class DeterministicSocketData : public StaticSocketDataProvider {
|
| size_t reads_count,
|
| MockWrite* writes,
|
| size_t writes_count);
|
| -
|
| + StaticSocketDataHelper helper_;
|
| + MockConnect connect_;
|
| int sequence_number_;
|
| MockRead current_read_;
|
| MockWrite current_write_;
|
| @@ -836,7 +839,6 @@ class DeterministicSocketHelper {
|
| // Mock UDP socket to be used in conjunction with DeterministicSocketData.
|
| class DeterministicMockUDPClientSocket
|
| : public DatagramClientSocket,
|
| - public AsyncSocket,
|
| public DeterministicSocketData::Delegate,
|
| public base::SupportsWeakPtr<DeterministicMockUDPClientSocket> {
|
| public:
|
| @@ -869,11 +871,6 @@ class DeterministicMockUDPClientSocket
|
| // DatagramClientSocket implementation.
|
| int Connect(const IPEndPoint& address) override;
|
|
|
| - // AsyncSocket implementation.
|
| - void OnReadComplete(const MockRead& data) override;
|
| - void OnWriteComplete(int rv) override;
|
| - void OnConnectComplete(const MockConnect& data) override;
|
| -
|
| void set_source_port(uint16 port) { source_port_ = port; }
|
|
|
| private:
|
| @@ -888,7 +885,6 @@ class DeterministicMockUDPClientSocket
|
| // Mock TCP socket to be used in conjunction with DeterministicSocketData.
|
| class DeterministicMockTCPClientSocket
|
| : public MockClientSocket,
|
| - public AsyncSocket,
|
| public DeterministicSocketData::Delegate,
|
| public base::SupportsWeakPtr<DeterministicMockTCPClientSocket> {
|
| public:
|
| @@ -920,11 +916,6 @@ class DeterministicMockTCPClientSocket
|
| bool WasNpnNegotiated() const override;
|
| bool GetSSLInfo(SSLInfo* ssl_info) override;
|
|
|
| - // AsyncSocket:
|
| - void OnReadComplete(const MockRead& data) override;
|
| - void OnWriteComplete(int rv) override;
|
| - void OnConnectComplete(const MockConnect& data) override;
|
| -
|
| private:
|
| DeterministicSocketHelper helper_;
|
|
|
|
|