diff options
Diffstat (limited to 'net/socket/socket_test_util.h')
-rw-r--r-- | net/socket/socket_test_util.h | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/net/socket/socket_test_util.h b/net/socket/socket_test_util.h index 147a4ba..0a01df3 100644 --- a/net/socket/socket_test_util.h +++ b/net/socket/socket_test_util.h @@ -172,6 +172,7 @@ class StaticSocketDataProvider : public SocketDataProvider { virtual MockRead GetNextRead(); virtual MockWriteResult OnWrite(const std::string& data); virtual void Reset(); + virtual void CompleteRead() {} // These functions get access to the next available read and write data. const MockRead& PeekRead() const; @@ -284,7 +285,7 @@ class DelayedSocketData : public StaticSocketDataProvider, virtual MockRead GetNextRead(); virtual MockWriteResult OnWrite(const std::string& data); virtual void Reset(); - void CompleteRead(); + virtual void CompleteRead(); void ForceNextRead(); private: @@ -327,6 +328,8 @@ class OrderedSocketData : public StaticSocketDataProvider, virtual MockRead GetNextRead(); virtual MockWriteResult OnWrite(const std::string& data); virtual void Reset(); + virtual void CompleteRead(); + void SetCompletionCallback(CompletionCallback* callback) { callback_ = callback; } @@ -334,8 +337,6 @@ class OrderedSocketData : public StaticSocketDataProvider, // Posts a quit message to the current message loop, if one is running. void EndLoop(); - void CompleteRead(); - private: friend class base::RefCounted<OrderedSocketData>; virtual ~OrderedSocketData(); @@ -425,6 +426,8 @@ class DeterministicSocketData : public StaticSocketDataProvider, virtual void Reset(); + virtual void CompleteRead() {} + // Consume all the data up to the give stop point (via SetStop()). void Run(); @@ -442,7 +445,6 @@ class DeterministicSocketData : public StaticSocketDataProvider, virtual void StopAfter(int seq) { SetStop(sequence_number_ + seq); } - void CompleteRead(); bool stopped() const { return stopped_; } void SetStopped(bool val) { stopped_ = val; } MockRead& current_read() { return current_read_; } |