1 files changed, 14 insertions, 0 deletions
diff --git a/net/socket/ssl_server_socket_unittest.cc b/net/socket/ssl_server_socket_unittest.cc
index eb9dc7c..1ec6b4b 100644
--- a/net/socket/ssl_server_socket_unittest.cc
+++ b/net/socket/ssl_server_socket_unittest.cc
@@ -85,6 +85,14 @@ class FakeDataChannel {
             &FakeDataChannel::DoReadCallback));
     return buf_len;
   }
+  virtual int Write(IOBuffer* buf, int buf_len,
+                    const CompletionCallback& callback) {
+    data_.push(new net::DrainableIOBuffer(buf, buf_len));
+    MessageLoop::current()->PostTask(
+        FROM_HERE, task_factory_.NewRunnableMethod(
+            &FakeDataChannel::DoReadCallback));
+    return buf_len;
+  }
 
  private:
   void DoReadCallback() {
@@ -160,6 +168,12 @@ class FakeSocket : public StreamSocket {
     buf_len = rand() % buf_len + 1;
     return outgoing_->Write(buf, buf_len, callback);
   }
+  virtual int Write(IOBuffer* buf, int buf_len,
+                    const CompletionCallback& callback) {
+    // Write random number of bytes.
+    buf_len = rand() % buf_len + 1;
+    return outgoing_->Write(buf, buf_len, callback);
+  }
 
   virtual bool SetReceiveBufferSize(int32 size) {
     return true;