Add data pipe wrappers to be used to implement serial receive.

device/serial/data_pipe_receiver_unittest.cc

+// 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.
+
+#include "base/bind.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "base/strings/string_piece.h"
+#include "device/serial/async_waiter.h"
+#include "device/serial/buffer.h"
+#include "device/serial/data_pipe.mojom.h"
+#include "device/serial/data_pipe_producer.h"
+#include "device/serial/data_pipe_receiver.h"
+#include "mojo/public/cpp/bindings/interface_ptr.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace device {
+
+class DataPipeReceiverTest : public testing::Test {
+ public:
+  DataPipeReceiverTest() : error_(0), seen_connection_error_(false) {}
+
+  virtual void SetUp() OVERRIDE {
+    message_loop_.reset(new base::MessageLoop);
+    mojo::InterfacePtr<serial::DataPipeProducer> producer_handle;
+    producer_ = mojo::WeakBindToProxy(
+        new DataPipeProducer(
+            base::Bind(&DataPipeReceiverTest::CanWriteData,
+                       base::Unretained(this)),
+            base::Bind(&DataPipeReceiverTest::OnError,
+                       base::Unretained(this))),
+        &producer_handle);
+    receiver_ = new DataPipeReceiver(producer_handle.Pass(), 100, -10);
+  }
+
+  virtual void TearDown() OVERRIDE {
+    write_buffer_.reset();
+    buffer_.reset();
+    message_loop_.reset();
+  }
+
+  void OnError() {
+    seen_connection_error_ = true;
+    StopMessageLoop();
+  }
+
+  void RunMessageLoop() {
+    base::RunLoop run_loop;
+    stop_run_loop_ = run_loop.QuitClosure();
+    run_loop.Run();
+  }
+
+  void StopMessageLoop() {
+    if (!stop_run_loop_.is_null())
+      stop_run_loop_.Run();
+  }
+
+  bool Receive() {
+    return receiver_->Receive(base::Bind(&DataPipeReceiverTest::OnDataReceived,
+                                         base::Unretained(this)),
+                              base::Bind(&DataPipeReceiverTest::OnReceiveError,
+                                         base::Unretained(this)));
+  }
+
+  bool FillWriteBuffer(const base::StringPiece& data, int32_t error) {
+    if (!write_buffer_)
+      RunMessageLoop();
+    if (!write_buffer_)
+      return false;
+    if (write_buffer_->GetSize() < static_cast<uint32_t>(data.size())) {
+      write_buffer_->Done(0);
+      write_buffer_.reset();
+      return false;
+    }
+
+    memcpy(write_buffer_->GetData(), data.data(), data.size());
+    if (error)
+      write_buffer_->DoneWithError(static_cast<uint32_t>(data.size()), error);
+    else
+      write_buffer_->Done(static_cast<uint32_t>(data.size()));
+    write_buffer_.reset();
+    return true;
+  }
+
+  bool ReceiveAndWait() {
+    if (!Receive())
+      return false;
+    RunMessageLoop();
+    return true;
+  }
+
+  void OnDataReceived(scoped_ptr<ReadOnlyBuffer> buffer) {
+    ASSERT_TRUE(buffer);
+    error_ = 0;
+    buffer_ = buffer.Pass();
+    buffer_contents_ = std::string(buffer_->GetData(), buffer_->GetSize());
+    StopMessageLoop();
+  }
+
+  void OnReceiveError(int32_t error) {
+    buffer_contents_.clear();
+    error_ = error;
+    StopMessageLoop();
+  }
+
+  void CanWriteData(scoped_ptr<WritableBuffer> buffer) {
+    write_buffer_ = buffer.Pass();
+    StopMessageLoop();
+  }
+
+ protected:
+  scoped_ptr<base::MessageLoop> message_loop_;
+  base::Closure stop_run_loop_;
+
+  scoped_refptr<DataPipeProducer> producer_;
+  scoped_refptr<DataPipeReceiver> receiver_;
+
+  scoped_ptr<ReadOnlyBuffer> buffer_;
+  std::string buffer_contents_;
+  int32_t error_;
+  scoped_ptr<WritableBuffer> write_buffer_;
+
+  bool seen_connection_error_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(DataPipeReceiverTest);
+};
+
+TEST_F(DataPipeReceiverTest, Basic) {
+  FillWriteBuffer("a", 0);
+
+  // Wait until we get a new write_buffer_ before we wait to receive.
+  RunMessageLoop();
+  ASSERT_TRUE(write_buffer_);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("a", buffer_contents_);
+  buffer_->Done(1);
+
+  FillWriteBuffer("b", 0);
+
+  // Wait until we get a new write_buffer_ before we wait to receive.
+  RunMessageLoop();
+  ASSERT_TRUE(write_buffer_);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("b", buffer_contents_);
+}
+
+TEST_F(DataPipeReceiverTest, PartialReceive) {
+  FillWriteBuffer("ab", 0);
+
+  // Wait until we get a new write_buffer_ before we wait to receive.
+  RunMessageLoop();
+  ASSERT_TRUE(write_buffer_);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("ab", buffer_contents_);
+  buffer_->Done(1);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("b", buffer_contents_);
+}
+
+TEST_F(DataPipeReceiverTest, ErrorAndData) {
+  FillWriteBuffer("abc", -1);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("abc", buffer_contents_);
+  buffer_->Done(1);
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(ReceiveAndWait());
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("bc", buffer_contents_);
+  buffer_->Done(1);
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(ReceiveAndWait());
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("c", buffer_contents_);
+  buffer_->Done(1);
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(-1, error_);
+  ASSERT_FALSE(write_buffer_);
+
+  ASSERT_TRUE(Receive());
+  FillWriteBuffer("d", -2);
+
+  RunMessageLoop();
+  ASSERT_TRUE(buffer_);
+  EXPECT_EQ("d", buffer_contents_);
+  buffer_->Done(1);
+  EXPECT_EQ(0, error_);
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_EQ(-2, error_);
+}
+
+TEST_F(DataPipeReceiverTest, ErrorOnly) {
+  FillWriteBuffer("", -1);
+
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_FALSE(buffer_);
+  EXPECT_EQ(-1, error_);
+  ASSERT_FALSE(write_buffer_);
+
+  ASSERT_TRUE(Receive());
+  FillWriteBuffer("", -2);
+
+  RunMessageLoop();
+  EXPECT_FALSE(buffer_);
+  EXPECT_EQ(-2, error_);
+  ASSERT_FALSE(write_buffer_);
+}
+
+TEST_F(DataPipeReceiverTest, ProducerShutdown) {
+  producer_ = NULL;
+  ASSERT_TRUE(ReceiveAndWait());
+  EXPECT_FALSE(buffer_);
+  EXPECT_EQ(-10, error_);
+  ASSERT_FALSE(write_buffer_);
+  ASSERT_FALSE(Receive());
+}
+
+TEST_F(DataPipeReceiverTest, ReceiverShutdown) {
+  Receive();
+  receiver_ = NULL;
+  EXPECT_EQ(-10, error_);
+  RunMessageLoop();
+  EXPECT_TRUE(seen_connection_error_);
+}
+
+}  // namespace device