Add unittests for BufferedSocketWriter and fix some bugs in that code.

remoting/protocol/fake_session.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef REMOTING_PROTOCOL_FAKE_SESSION_H_
 #define REMOTING_PROTOCOL_FAKE_SESSION_H_
 
 #include <map>
 #include <string>
 #include <vector>
@@ skipping 22 matching lines @@
 // PairWith() method, e.g.: a->PairWith(b). After this all data
 // written to |a| can be read from |b| and vica versa. Two connected
 // sockets |a| and |b| must be created and used on the same thread.
 class FakeSocket : public net::StreamSocket {
  public:
   FakeSocket();
   virtual ~FakeSocket();
 
   const std::string& written_data() const { return written_data_; }
 
+  void set_write_limit(int write_limit) { write_limit_ = write_limit; }
+  void set_async_write(bool async_write) { async_write_ = async_write; }
+  void set_next_write_error(int error) { next_write_error_ = error; }
   void set_next_read_error(int error) { next_read_error_ = error; }
   void AppendInputData(const std::vector<char>& data);
   void PairWith(FakeSocket* peer_socket);
   int input_pos() const { return input_pos_; }
   bool read_pending() const { return read_pending_; }
 
   // net::Socket implementation.
   virtual int Read(net::IOBuffer* buf, int buf_len,
                    const net::CompletionCallback& callback) OVERRIDE;
   virtual int Write(net::IOBuffer* buf, int buf_len,
@@ skipping 14 matching lines @@
   virtual void SetOmniboxSpeculation() OVERRIDE;
   virtual bool WasEverUsed() const OVERRIDE;
   virtual bool UsingTCPFastOpen() const OVERRIDE;
   virtual int64 NumBytesRead() const OVERRIDE;
   virtual base::TimeDelta GetConnectTimeMicros() const OVERRIDE;
   virtual bool WasNpnNegotiated() const OVERRIDE;
   virtual net::NextProto GetNegotiatedProtocol() const OVERRIDE;
   virtual bool GetSSLInfo(net::SSLInfo* ssl_info) OVERRIDE;
 
  private:
+  void DoAsyncWrite(scoped_refptr<net::IOBuffer> buf, int buf_len,
+                    const net::CompletionCallback& callback);
+  void DoWrite(net::IOBuffer* buf, int buf_len);
+
+  bool async_write_;
+  bool write_pending_;
+  int write_limit_;
+  int next_write_error_;
+
   int next_read_error_;
   bool read_pending_;
   scoped_refptr<net::IOBuffer> read_buffer_;
   int read_buffer_size_;
   net::CompletionCallback read_callback_;
   base::WeakPtr<FakeSocket> peer_socket_;
 
   std::string written_data_;
   std::string input_data_;
   int input_pos_;
@@ skipping 99 matching lines @@
   ErrorCode error_;
   bool closed_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeSession);
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_FAKE_SESSION_H_