Index: extensions/renderer/api/serial/data_sender_unittest.cc |
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 0000000000000000000000000000000000000000..488826610e7f150efb8b6f610035a1015cb0438e |
--- /dev/null |
+++ b/extensions/renderer/api/serial/data_sender_unittest.cc |
@@ -0,0 +1,158 @@ |
+// 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) { |
+ expected_data_.push(std::string(1000, 'a')); |
+ 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(std::string(2, 'b')); |
+ 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 |