OLD | NEW |
(Empty) | |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include <queue> |
| 6 |
| 7 #include "device/serial/data_sink_receiver.h" |
| 8 #include "device/serial/data_stream.mojom.h" |
| 9 #include "extensions/renderer/api_test_base.h" |
| 10 #include "grit/extensions_renderer_resources.h" |
| 11 |
| 12 namespace extensions { |
| 13 |
| 14 // Runs tests defined in extensions/test/data/data_sender_unittest.js |
| 15 class DataSenderTest : public ApiTestBase { |
| 16 public: |
| 17 DataSenderTest() {} |
| 18 |
| 19 virtual void SetUp() OVERRIDE { |
| 20 ApiTestBase::SetUp(); |
| 21 env()->RegisterModule("async_waiter", IDR_ASYNC_WAITER_JS); |
| 22 env()->RegisterModule("data_sender", IDR_DATA_SENDER_JS); |
| 23 env()->RegisterModule("device/serial/data_stream.mojom", |
| 24 IDR_DATA_STREAM_MOJOM_JS); |
| 25 service_provider()->AddService( |
| 26 base::Bind(&DataSenderTest::CreateDataSink, base::Unretained(this))); |
| 27 } |
| 28 |
| 29 virtual void TearDown() OVERRIDE { |
| 30 if (receiver_) { |
| 31 receiver_->ShutDown(); |
| 32 receiver_ = NULL; |
| 33 } |
| 34 EXPECT_FALSE(buffer_); |
| 35 buffer_.reset(); |
| 36 ApiTestBase::TearDown(); |
| 37 } |
| 38 |
| 39 std::queue<int32_t> error_to_report_; |
| 40 std::queue<std::string> expected_data_; |
| 41 |
| 42 private: |
| 43 void CreateDataSink( |
| 44 mojo::InterfaceRequest<device::serial::DataSink> request) { |
| 45 receiver_ = mojo::WeakBindToRequest( |
| 46 new device::DataSinkReceiver( |
| 47 base::Bind(&DataSenderTest::ReadyToReceive, base::Unretained(this)), |
| 48 base::Bind(&DataSenderTest::OnCancel, base::Unretained(this)), |
| 49 base::Bind(base::DoNothing)), |
| 50 &request); |
| 51 } |
| 52 |
| 53 void ReadyToReceive(scoped_ptr<device::ReadOnlyBuffer> buffer) { |
| 54 std::string data(buffer->GetData(), buffer->GetSize()); |
| 55 if (expected_data_.empty()) { |
| 56 buffer_ = buffer.Pass(); |
| 57 return; |
| 58 } |
| 59 |
| 60 std::string& expected = expected_data_.front(); |
| 61 if (expected.size() > buffer->GetSize()) { |
| 62 EXPECT_EQ(expected.substr(0, buffer->GetSize()), data); |
| 63 expected = expected.substr(buffer->GetSize()); |
| 64 buffer->Done(buffer->GetSize()); |
| 65 return; |
| 66 } |
| 67 if (expected.size() < buffer->GetSize()) |
| 68 data = data.substr(0, expected.size()); |
| 69 EXPECT_EQ(expected, data); |
| 70 expected_data_.pop(); |
| 71 int32_t error = 0; |
| 72 if (!error_to_report_.empty()) { |
| 73 error = error_to_report_.front(); |
| 74 error_to_report_.pop(); |
| 75 } |
| 76 if (error) |
| 77 buffer->DoneWithError(data.size(), error); |
| 78 else |
| 79 buffer->Done(data.size()); |
| 80 } |
| 81 |
| 82 void OnCancel(int32_t error) { |
| 83 ASSERT_TRUE(buffer_); |
| 84 buffer_->DoneWithError(0, error); |
| 85 buffer_.reset(); |
| 86 } |
| 87 |
| 88 scoped_refptr<device::DataSinkReceiver> receiver_; |
| 89 scoped_ptr<device::ReadOnlyBuffer> buffer_; |
| 90 |
| 91 DISALLOW_COPY_AND_ASSIGN(DataSenderTest); |
| 92 }; |
| 93 |
| 94 TEST_F(DataSenderTest, Send) { |
| 95 expected_data_.push("aa"); |
| 96 RunTest("data_sender_unittest.js", "testSend"); |
| 97 } |
| 98 |
| 99 TEST_F(DataSenderTest, LargeSend) { |
| 100 expected_data_.push(std::string(1000, 'a')); |
| 101 RunTest("data_sender_unittest.js", "testLargeSend"); |
| 102 } |
| 103 |
| 104 TEST_F(DataSenderTest, SendError) { |
| 105 expected_data_.push(""); |
| 106 expected_data_.push("a"); |
| 107 error_to_report_.push(1); |
| 108 RunTest("data_sender_unittest.js", "testSendError"); |
| 109 } |
| 110 |
| 111 TEST_F(DataSenderTest, SendErrorPartialSuccess) { |
| 112 expected_data_.push(std::string(5, 'b')); |
| 113 expected_data_.push("a"); |
| 114 error_to_report_.push(1); |
| 115 RunTest("data_sender_unittest.js", "testSendErrorPartialSuccess"); |
| 116 } |
| 117 |
| 118 TEST_F(DataSenderTest, SendErrorBetweenPackets) { |
| 119 expected_data_.push(std::string(2, 'b')); |
| 120 expected_data_.push("a"); |
| 121 error_to_report_.push(1); |
| 122 RunTest("data_sender_unittest.js", "testSendErrorBetweenPackets"); |
| 123 } |
| 124 |
| 125 TEST_F(DataSenderTest, SendErrorInSecondPacket) { |
| 126 expected_data_.push(std::string(3, 'b')); |
| 127 expected_data_.push("a"); |
| 128 error_to_report_.push(1); |
| 129 RunTest("data_sender_unittest.js", "testSendErrorInSecondPacket"); |
| 130 } |
| 131 |
| 132 TEST_F(DataSenderTest, SendErrorInLargeSend) { |
| 133 expected_data_.push(std::string(2, 'b')); |
| 134 expected_data_.push("a"); |
| 135 error_to_report_.push(1); |
| 136 RunTest("data_sender_unittest.js", "testSendErrorInLargeSend"); |
| 137 } |
| 138 |
| 139 TEST_F(DataSenderTest, SendErrorBeforeLargeSend) { |
| 140 expected_data_.push(std::string(2, 'b')); |
| 141 expected_data_.push("a"); |
| 142 error_to_report_.push(1); |
| 143 RunTest("data_sender_unittest.js", "testSendErrorBeforeLargeSend"); |
| 144 } |
| 145 |
| 146 TEST_F(DataSenderTest, CancelWithoutSend) { |
| 147 RunTest("data_sender_unittest.js", "testCancelWithoutSend"); |
| 148 } |
| 149 |
| 150 TEST_F(DataSenderTest, Cancel) { |
| 151 RunTest("data_sender_unittest.js", "testCancel"); |
| 152 } |
| 153 |
| 154 TEST_F(DataSenderTest, Close) { |
| 155 RunTest("data_sender_unittest.js", "testClose"); |
| 156 } |
| 157 |
| 158 } // namespace extensions |
OLD | NEW |