Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1028)

Unified Diff: remoting/protocol/fake_datagram_socket.h

Issue 580243003: Cleanup Fake* classes in remoting/protocol (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@sctp
Patch Set: Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: remoting/protocol/fake_datagram_socket.h
diff --git a/remoting/protocol/fake_datagram_socket.h b/remoting/protocol/fake_datagram_socket.h
new file mode 100644
index 0000000000000000000000000000000000000000..feb5511f871bd8cfcd08a1b058b0521e9985a2af
--- /dev/null
+++ b/remoting/protocol/fake_datagram_socket.h
@@ -0,0 +1,117 @@
+// Copyright 2014 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_DATAGRAM_SOCKET_H_
+#define REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_
+
+#include <map>
+#include <string>
+#include <vector>
+
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "net/base/completion_callback.h"
+#include "net/socket/socket.h"
+#include "remoting/protocol/datagram_channel_factory.h"
+
+namespace base {
+class SingleThreadTaskRunner;
+}
+
+namespace remoting {
+namespace protocol {
+
+// FakeDatagramSocket implement net::StreamSocket interface. All data written to
+// FakeDatagramSocket is stored in a buffer returned by written_packets().
+// Read() reads data from another buffer that can be set with
+// AppendInputPacket(). Pending reads are supported, so if there is a pending
+// read AppendInputPacket() calls the read callback.
+//
+// Two fake sockets can be connected to each other using the
+// PairWith() method, e.g.: a->PairWith(b). After this all data
+// written to |a| can be read from |b| and vice versa. Two connected
+// sockets |a| and |b| must be created and used on the same thread.
+class FakeDatagramSocket : public net::Socket {
+ public:
+ FakeDatagramSocket();
+ virtual ~FakeDatagramSocket();
+
+ const std::vector<std::string>& written_packets() const {
+ return written_packets_;
+ }
+
+ void AppendInputPacket(const std::string& data);
Wez 2014/09/20 00:25:09 Consider replacing the term "input" with "to read"
Wez 2014/09/20 00:25:10 Would it be simpler to always create these things
Sergey Ulanov 2014/09/22 19:22:11 We had AppendInputPacket() before this CL. I think
Sergey Ulanov 2014/09/22 19:22:11 No. For some tests it's easier to have not connect
+ int input_pos() const { return input_pos_; }
Wez 2014/09/20 00:25:09 What does |input_pos| mean? Is it a position in pa
Sergey Ulanov 2014/09/22 19:22:11 Added comment
+ void PairWith(FakeDatagramSocket* peer_socket);
Wez 2014/09/20 00:25:09 Clarify that caller retains ownership of |peer_soc
Sergey Ulanov 2014/09/22 19:22:11 it's not scoped_ptr<> so it should be clear that o
Wez 2014/09/23 17:07:38 Except that there is still old code in the codebas
Sergey Ulanov 2014/09/23 19:06:08 I believe all code in src/remoting should be using
+
+ // 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,
+ const net::CompletionCallback& callback) OVERRIDE;
+
Wez 2014/09/20 00:25:09 nit: Remove blank line here
Sergey Ulanov 2014/09/22 19:22:11 Done.
+ virtual int SetReceiveBufferSize(int32 size) OVERRIDE;
+ virtual int SetSendBufferSize(int32 size) OVERRIDE;
+
+ private:
+ base::WeakPtr<FakeDatagramSocket> peer_socket_;
+
+ bool read_pending_;
Wez 2014/09/20 00:25:10 nit: has_pending_read_ or has_read_pending_ or is_
Sergey Ulanov 2014/09/22 19:22:11 We use read_pending_ in other places as well, and
+ scoped_refptr<net::IOBuffer> read_buffer_;
+ int read_buffer_size_;
+ net::CompletionCallback read_callback_;
+
+ std::vector<std::string> written_packets_;
+ std::vector<std::string> input_packets_;
+ int input_pos_;
+
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
+ base::WeakPtrFactory<FakeDatagramSocket> weak_factory_;
+
+ DISALLOW_COPY_AND_ASSIGN(FakeDatagramSocket);
+};
+
+class FakeDatagramChannelFactory : public DatagramChannelFactory {
+ public:
+ FakeDatagramChannelFactory();
+ virtual ~FakeDatagramChannelFactory();
+
+ void set_async_creation(bool async_creation) {
Wez 2014/09/20 00:25:10 nit: set_asynchronous, or set_is_asynchronous or s
Sergey Ulanov 2014/09/22 19:22:11 Done.
+ async_creation_ = async_creation;
+ }
+
+ void set_fail(bool fail) { fail_ = fail; }
Wez 2014/09/20 00:25:09 nit: set_fail_create_channel or set_fail_create?
Sergey Ulanov 2014/09/22 19:22:11 Done.
+
+ void PairWith(FakeDatagramChannelFactory* peer_factory);
Wez 2014/09/20 00:25:10 What does it mean to pair two factories?
Sergey Ulanov 2014/09/22 19:22:11 It means that the sockets they create are paired a
Wez 2014/09/23 17:07:38 Suggest tweaking comment to make clear that the cr
Sergey Ulanov 2014/09/23 19:06:08 Done.
+
+ FakeDatagramSocket* GetChannel(const std::string& name);
Wez 2014/09/20 00:25:09 What are the ownership semantics of GetChannel - l
Sergey Ulanov 2014/09/22 19:22:11 Added comment
+
+ // DatagramChannelFactory interface.
+ virtual void CreateChannel(const std::string& name,
+ const ChannelCreatedCallback& callback) OVERRIDE;
+ virtual void CancelChannelCreation(const std::string& name) OVERRIDE;
+
+ private:
+ typedef std::map<std::string, FakeDatagramSocket*> ChannelsMap;
+
+ void NotifyChannelCreated(const std::string& name,
+ const ChannelCreatedCallback& callback);
+
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
+ bool async_creation_;
+ ChannelsMap channels_;
+
+ base::WeakPtr<FakeDatagramChannelFactory> peer_factory_;
Wez 2014/09/20 00:25:10 nit: In the FakeDatagramSocket the peer weakptrfac
Sergey Ulanov 2014/09/22 19:22:11 Done.
+
+ bool fail_;
+
+ base::WeakPtrFactory<FakeDatagramChannelFactory> weak_factory_;
+
+ DISALLOW_COPY_AND_ASSIGN(FakeDatagramChannelFactory);
+};
+
+} // namespace protocol
+} // namespace remoting
+
+#endif // REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_

Powered by Google App Engine
This is Rietveld 408576698