1 files changed, 162 insertions, 0 deletions
diff --git a/extensions/renderer/api/serial/data_sender_unittest.cc b/extensions/renderer/api/serial/data_sender_unittest.cc
new file mode 100644
index 0000000..8fa9251
--- /dev/null
+++ b/extensions/renderer/api/serial/data_sender_unittest.cc
@@ -0,0 +1,162 @@
+// 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 <queue>
+
+#include "device/serial/data_sink_receiver.h"
+#include "device/serial/data_stream.mojom.h"
+#include "extensions/renderer/api_test_base.h"
+#include "grit/extensions_renderer_resources.h"
+
+namespace extensions {
+
+// Runs tests defined in extensions/test/data/data_sender_unittest.js
+class DataSenderTest : public ApiTestBase {
+ public:
+  DataSenderTest() {}
+
+  virtual void SetUp() OVERRIDE {
+    ApiTestBase::SetUp();
+    env()->RegisterModule("async_waiter", IDR_ASYNC_WAITER_JS);
+    env()->RegisterModule("data_sender", IDR_DATA_SENDER_JS);
+    env()->RegisterModule("device/serial/data_stream.mojom",
+                          IDR_DATA_STREAM_MOJOM_JS);
+    service_provider()->AddService(
+        base::Bind(&DataSenderTest::CreateDataSink, base::Unretained(this)));
+  }
+
+  virtual void TearDown() OVERRIDE {
+    if (receiver_) {
+      receiver_->ShutDown();
+      receiver_ = NULL;
+    }
+    EXPECT_FALSE(buffer_);
+    buffer_.reset();
+    ApiTestBase::TearDown();
+  }
+
+  std::queue<int32_t> error_to_report_;
+  std::queue<std::string> expected_data_;
+
+ private:
+  void CreateDataSink(
+      mojo::InterfaceRequest<device::serial::DataSink> request) {
+    receiver_ = mojo::WeakBindToRequest(
+        new device::DataSinkReceiver(
+            base::Bind(&DataSenderTest::ReadyToReceive, base::Unretained(this)),
+            base::Bind(&DataSenderTest::OnCancel, base::Unretained(this)),
+            base::Bind(base::DoNothing)),
+        &request);
+  }
+
+  void ReadyToReceive(scoped_ptr<device::ReadOnlyBuffer> buffer) {
+    std::string data(buffer->GetData(), buffer->GetSize());
+    if (expected_data_.empty()) {
+      buffer_ = buffer.Pass();
+      return;
+    }
+
+    std::string& expected = expected_data_.front();
+    if (expected.size() > buffer->GetSize()) {
+      EXPECT_EQ(expected.substr(0, buffer->GetSize()), data);
+      expected = expected.substr(buffer->GetSize());
+      buffer->Done(buffer->GetSize());
+      return;
+    }
+    if (expected.size() < buffer->GetSize())
+      data = data.substr(0, expected.size());
+    EXPECT_EQ(expected, data);
+    expected_data_.pop();
+    int32_t error = 0;
+    if (!error_to_report_.empty()) {
+      error = error_to_report_.front();
+      error_to_report_.pop();
+    }
+    if (error)
+      buffer->DoneWithError(data.size(), error);
+    else
+      buffer->Done(data.size());
+  }
+
+  void OnCancel(int32_t error) {
+    ASSERT_TRUE(buffer_);
+    buffer_->DoneWithError(0, error);
+    buffer_.reset();
+  }
+
+  scoped_refptr<device::DataSinkReceiver> receiver_;
+  scoped_ptr<device::ReadOnlyBuffer> buffer_;
+
+  DISALLOW_COPY_AND_ASSIGN(DataSenderTest);
+};
+
+TEST_F(DataSenderTest, Send) {
+  expected_data_.push("aa");
+  RunTest("data_sender_unittest.js", "testSend");
+}
+
+TEST_F(DataSenderTest, LargeSend) {
+  std::string pattern = "1234567890";
+  std::string expected_data;
+  for (int i = 0; i < 110; i++)
+    expected_data += pattern;
+  expected_data_.push(expected_data);
+  RunTest("data_sender_unittest.js", "testLargeSend");
+}
+
+TEST_F(DataSenderTest, SendError) {
+  expected_data_.push("");
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendError");
+}
+
+TEST_F(DataSenderTest, SendErrorPartialSuccess) {
+  expected_data_.push(std::string(5, 'b'));
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendErrorPartialSuccess");
+}
+
+TEST_F(DataSenderTest, SendErrorBetweenPackets) {
+  expected_data_.push(std::string(2, 'b'));
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendErrorBetweenPackets");
+}
+
+TEST_F(DataSenderTest, SendErrorInSecondPacket) {
+  expected_data_.push(std::string(3, 'b'));
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendErrorInSecondPacket");
+}
+
+TEST_F(DataSenderTest, SendErrorInLargeSend) {
+  expected_data_.push("123456789012");
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendErrorInLargeSend");
+}
+
+TEST_F(DataSenderTest, SendErrorBeforeLargeSend) {
+  expected_data_.push(std::string(2, 'b'));
+  expected_data_.push("a");
+  error_to_report_.push(1);
+  RunTest("data_sender_unittest.js", "testSendErrorBeforeLargeSend");
+}
+
+TEST_F(DataSenderTest, CancelWithoutSend) {
+  RunTest("data_sender_unittest.js", "testCancelWithoutSend");
+}
+
+TEST_F(DataSenderTest, Cancel) {
+  RunTest("data_sender_unittest.js", "testCancel");
+}
+
+TEST_F(DataSenderTest, Close) {
+  RunTest("data_sender_unittest.js", "testClose");
+}
+
+}  // namespace extensions