Chromium Code Reviews| Index: net/socket/socket_test_util.h |
| diff --git a/net/socket/socket_test_util.h b/net/socket/socket_test_util.h |
| index fa47d26fd0f36da03aaedf37b3e0775c6543fc64..56844974bace79f1d316f1c081c495f4749e57ec 100644 |
| --- a/net/socket/socket_test_util.h |
| +++ b/net/socket/socket_test_util.h |
| @@ -587,6 +587,8 @@ class MockClientSocket : public net::SSLClientSocket { |
| // Socket implementation. |
| virtual int Read(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) = 0; |
| + virtual int Read(net::IOBuffer* buf, int buf_len, |
| + const net::CompletionCallback& callback) = 0; |
| virtual int Write(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) = 0; |
| virtual bool SetReceiveBufferSize(int32 size) OVERRIDE; |
| @@ -640,6 +642,8 @@ class MockTCPClientSocket : public MockClientSocket, public AsyncSocket { |
| // Socket implementation. |
| virtual int Read(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| + virtual int Read(net::IOBuffer* buf, int buf_len, |
| + const net::CompletionCallback& callback) OVERRIDE; |
| virtual int Write(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| @@ -677,7 +681,8 @@ class MockTCPClientSocket : public MockClientSocket, public AsyncSocket { |
| // While an asynchronous IO is pending, we save our user-buffer state. |
| net::IOBuffer* pending_buf_; |
| int pending_buf_len_; |
| - net::OldCompletionCallback* pending_callback_; |
| + net::OldCompletionCallback* old_pending_callback_; |
| + net::CompletionCallback pending_callback_; |
|
James Hawkins
2011/12/07 00:01:57
#include "net/base/completion_callback.h"
James Hawkins
2011/12/07 00:08:11
Done.
|
| bool was_used_to_convey_data_; |
| }; |
| @@ -695,11 +700,13 @@ class DeterministicMockTCPClientSocket : public MockClientSocket, |
| void CompleteWrite(); |
| int CompleteRead(); |
| - // Socket: |
| + // Socket implementation. |
| virtual int Write(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| virtual int Read(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| + virtual int Read(net::IOBuffer* buf, int buf_len, |
| + const net::CompletionCallback& callback) OVERRIDE; |
| // StreamSocket implementation. |
| virtual int Connect(net::OldCompletionCallback* callback) OVERRIDE; |
| @@ -725,7 +732,8 @@ class DeterministicMockTCPClientSocket : public MockClientSocket, |
| net::IOBuffer* read_buf_; |
| int read_buf_len_; |
| bool read_pending_; |
| - net::OldCompletionCallback* read_callback_; |
| + net::OldCompletionCallback* old_read_callback_; |
| + net::CompletionCallback read_callback_; |
| net::DeterministicSocketData* data_; |
| bool was_used_to_convey_data_; |
| }; |
| @@ -743,6 +751,8 @@ class MockSSLClientSocket : public MockClientSocket, public AsyncSocket { |
| // Socket implementation. |
| virtual int Read(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| + virtual int Read(net::IOBuffer* buf, int buf_len, |
| + const net::CompletionCallback& callback) OVERRIDE; |
| virtual int Write(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| @@ -785,31 +795,35 @@ class MockUDPClientSocket : public DatagramClientSocket, |
| MockUDPClientSocket(SocketDataProvider* data, net::NetLog* net_log); |
| virtual ~MockUDPClientSocket(); |
| - // Socket interface |
| + // Socket implementation. |
| virtual int Read(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| + virtual int Read(net::IOBuffer* buf, int buf_len, |
| + const net::CompletionCallback& callback) OVERRIDE; |
| virtual int Write(net::IOBuffer* buf, int buf_len, |
| net::OldCompletionCallback* callback) OVERRIDE; |
| virtual bool SetReceiveBufferSize(int32 size) OVERRIDE; |
| virtual bool SetSendBufferSize(int32 size) OVERRIDE; |
| - // DatagramSocket interface |
| + // DatagramSocket implementation. |
| virtual void Close() OVERRIDE; |
| virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; |
| virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; |
| virtual const BoundNetLog& NetLog() const OVERRIDE; |
| - // DatagramClientSocket interface |
| + // DatagramClientSocket implementation. |
| virtual int Connect(const IPEndPoint& address) OVERRIDE; |
| - // AsyncSocket interface |
| + // AsyncSocket implementation. |
| virtual void OnReadComplete(const MockRead& data) OVERRIDE; |
| private: |
| int CompleteRead(); |
| void RunCallbackAsync(net::OldCompletionCallback* callback, int result); |
| - void RunCallback(net::OldCompletionCallback* callback, int result); |
| + void RunCallbackAsync(const net::CompletionCallback& callback, int result); |
| + void RunOldCallback(net::OldCompletionCallback* callback, int result); |
| + void RunCallback(const net::CompletionCallback& callback, int result); |
| bool connected_; |
| SocketDataProvider* data_; |
| @@ -820,7 +834,8 @@ class MockUDPClientSocket : public DatagramClientSocket, |
| // While an asynchronous IO is pending, we save our user-buffer state. |
| net::IOBuffer* pending_buf_; |
| int pending_buf_len_; |
| - net::OldCompletionCallback* pending_callback_; |
| + net::OldCompletionCallback* old_pending_callback_; |
| + net::CompletionCallback pending_callback_; |
| BoundNetLog net_log_; |