device/serial/serial_connection_unittest.cc

   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 <string>
   6
   7 #include "base/bind.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/run_loop.h"
  10 #include "base/strings/string_piece.h"
  11 #include "device/serial/data_receiver.h"
  12 #include "device/serial/data_sender.h"
  13 #include "device/serial/data_stream.mojom.h"
  14 #include "device/serial/serial.mojom.h"
  15 #include "device/serial/serial_connection.h"
  16 #include "device/serial/serial_service_impl.h"
  17 #include "device/serial/test_serial_io_handler.h"
  18 #include "testing/gtest/include/gtest/gtest.h"
  19 #include "third_party/mojo/src/mojo/public/cpp/bindings/error_handler.h"
  20 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_ptr.h"
  21 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_request.h"
  22
  23 namespace device {
  24 namespace {
  25
  26 class FakeSerialDeviceEnumerator : public SerialDeviceEnumerator {
  27   mojo::Array<serial::DeviceInfoPtr> GetDevices() override {
  28     mojo::Array<serial::DeviceInfoPtr> devices(1);
  29     devices[0] = serial::DeviceInfo::New();
  30     devices[0]->path = "device";
  31     return devices.Pass();
  32   }
  33 };
  34
  35 }  // namespace
  36
  37 class SerialConnectionTest : public testing::Test, public mojo::ErrorHandler {
  38  public:
  39   enum Event {
  40     EVENT_NONE,
  41     EVENT_GOT_INFO,
  42     EVENT_SET_OPTIONS,
  43     EVENT_GOT_CONTROL_SIGNALS,
  44     EVENT_SET_CONTROL_SIGNALS,
  45     EVENT_FLUSHED,
  46     EVENT_DATA_AT_IO_HANDLER,
  47     EVENT_DATA_SENT,
  48     EVENT_SEND_ERROR,
  49     EVENT_DATA_RECEIVED,
  50     EVENT_RECEIVE_ERROR,
  51     EVENT_CANCEL_COMPLETE,
  52     EVENT_ERROR,
  53   };
  54
  55   static const uint32_t kBufferSize;
  56
  57   SerialConnectionTest()
  58       : connected_(false),
  59         success_(false),
  60         bytes_sent_(0),
  61         send_error_(serial::SEND_ERROR_NONE),
  62         receive_error_(serial::RECEIVE_ERROR_NONE),
  63         expected_event_(EVENT_NONE) {}
  64
  65   void SetUp() override {
  66     message_loop_.reset(new base::MessageLoop);
  67     mojo::InterfacePtr<serial::SerialService> service;
  68     mojo::BindToProxy(
  69         new SerialServiceImpl(
  70             new SerialConnectionFactory(
  71                 base::Bind(&SerialConnectionTest::CreateIoHandler,
  72                            base::Unretained(this)),
  73                 base::MessageLoopProxy::current()),
  74             scoped_ptr<SerialDeviceEnumerator>(new FakeSerialDeviceEnumerator)),
  75         &service);
  76     service.set_error_handler(this);
  77     mojo::InterfacePtr<serial::DataSink> sink;
  78     mojo::InterfacePtr<serial::DataSource> source;
  79     mojo::InterfacePtr<serial::DataSourceClient> source_client;
  80     mojo::InterfaceRequest<serial::DataSourceClient> source_client_request =
  81         mojo::GetProxy(&source_client);
  82     service->Connect("device", serial::ConnectionOptions::New(),
  83                      mojo::GetProxy(&connection_), mojo::GetProxy(&sink),
  84                      mojo::GetProxy(&source), source_client.Pass());
  85     sender_.reset(new DataSender(sink.Pass(), kBufferSize,
  86                                  serial::SEND_ERROR_DISCONNECTED));
  87     receiver_ =
  88         new DataReceiver(source.Pass(), source_client_request.Pass(),
  89                          kBufferSize, serial::RECEIVE_ERROR_DISCONNECTED);
  90     connection_.set_error_handler(this);
  91     connection_->GetInfo(
  92         base::Bind(&SerialConnectionTest::StoreInfo, base::Unretained(this)));
  93     WaitForEvent(EVENT_GOT_INFO);
  94     ASSERT_TRUE(io_handler_.get());
  95   }
  96
  97   void StoreInfo(serial::ConnectionInfoPtr options) {
  98     info_ = options.Pass();
  99     EventReceived(EVENT_GOT_INFO);
 100   }
 101
 102   void StoreControlSignals(serial::DeviceControlSignalsPtr signals) {
 103     signals_ = signals.Pass();
 104     EventReceived(EVENT_GOT_CONTROL_SIGNALS);
 105   }
 106
 107   void StoreSuccess(Event event_to_report, bool success) {
 108     success_ = success;
 109     EventReceived(event_to_report);
 110   }
 111
 112   void Send(const base::StringPiece& data) {
 113     ASSERT_TRUE(sender_->Send(
 114         data,
 115         base::Bind(&SerialConnectionTest::OnDataSent, base::Unretained(this)),
 116         base::Bind(&SerialConnectionTest::OnSendError,
 117                    base::Unretained(this))));
 118   }
 119
 120   void Receive() {
 121     ASSERT_TRUE(
 122         receiver_->Receive(base::Bind(&SerialConnectionTest::OnDataReceived,
 123                                       base::Unretained(this)),
 124                            base::Bind(&SerialConnectionTest::OnReceiveError,
 125                                       base::Unretained(this))));
 126   }
 127
 128   void WaitForEvent(Event event) {
 129     expected_event_ = event;
 130     base::RunLoop run_loop;
 131     stop_run_loop_ = run_loop.QuitClosure();
 132     run_loop.Run();
 133   }
 134
 135   void EventReceived(Event event) {
 136     if (event != expected_event_)
 137       return;
 138     expected_event_ = EVENT_NONE;
 139     ASSERT_TRUE(message_loop_);
 140     ASSERT_TRUE(!stop_run_loop_.is_null());
 141     message_loop_->PostTask(FROM_HERE, stop_run_loop_);
 142   }
 143
 144   scoped_refptr<SerialIoHandler> CreateIoHandler() {
 145     io_handler_ = new TestSerialIoHandler;
 146     return io_handler_;
 147   }
 148
 149   void OnDataSent(uint32_t bytes_sent) {
 150     bytes_sent_ += bytes_sent;
 151     send_error_ = serial::SEND_ERROR_NONE;
 152     EventReceived(EVENT_DATA_SENT);
 153   }
 154
 155   void OnSendError(uint32_t bytes_sent, int32_t error) {
 156     bytes_sent_ += bytes_sent;
 157     send_error_ = static_cast<serial::SendError>(error);
 158     EventReceived(EVENT_SEND_ERROR);
 159   }
 160
 161   void OnDataReceived(scoped_ptr<ReadOnlyBuffer> buffer) {
 162     data_received_ += std::string(buffer->GetData(), buffer->GetSize());
 163     buffer->Done(buffer->GetSize());
 164     receive_error_ = serial::RECEIVE_ERROR_NONE;
 165     EventReceived(EVENT_DATA_RECEIVED);
 166   }
 167
 168   void OnReceiveError(int32_t error) {
 169     receive_error_ = static_cast<serial::ReceiveError>(error);
 170     EventReceived(EVENT_RECEIVE_ERROR);
 171   }
 172
 173   void OnConnectionError() override {
 174     EventReceived(EVENT_ERROR);
 175     FAIL() << "Connection error";
 176   }
 177
 178   mojo::Array<serial::DeviceInfoPtr> devices_;
 179   serial::ConnectionInfoPtr info_;
 180   serial::DeviceControlSignalsPtr signals_;
 181   bool connected_;
 182   bool success_;
 183   int bytes_sent_;
 184   serial::SendError send_error_;
 185   serial::ReceiveError receive_error_;
 186   std::string data_received_;
 187   Event expected_event_;
 188
 189   scoped_ptr<base::MessageLoop> message_loop_;
 190   base::Closure stop_run_loop_;
 191   mojo::InterfacePtr<serial::Connection> connection_;
 192   scoped_ptr<DataSender> sender_;
 193   scoped_refptr<DataReceiver> receiver_;
 194   scoped_refptr<TestSerialIoHandler> io_handler_;
 195
 196  private:
 197   DISALLOW_COPY_AND_ASSIGN(SerialConnectionTest);
 198 };
 199
 200 const uint32_t SerialConnectionTest::kBufferSize = 10;
 201
 202 TEST_F(SerialConnectionTest, GetInfo) {
 203   // |info_| is filled in during SetUp().
 204   ASSERT_TRUE(info_);
 205   EXPECT_EQ(9600u, info_->bitrate);
 206   EXPECT_EQ(serial::DATA_BITS_EIGHT, info_->data_bits);
 207   EXPECT_EQ(serial::PARITY_BIT_NO, info_->parity_bit);
 208   EXPECT_EQ(serial::STOP_BITS_ONE, info_->stop_bits);
 209   EXPECT_FALSE(info_->cts_flow_control);
 210 }
 211
 212 TEST_F(SerialConnectionTest, SetOptions) {
 213   serial::ConnectionOptionsPtr options(serial::ConnectionOptions::New());
 214   options->bitrate = 12345;
 215   options->data_bits = serial::DATA_BITS_SEVEN;
 216   options->has_cts_flow_control = true;
 217   options->cts_flow_control = true;
 218   connection_->SetOptions(options.Pass(),
 219                           base::Bind(&SerialConnectionTest::StoreSuccess,
 220                                      base::Unretained(this),
 221                                      EVENT_SET_OPTIONS));
 222   WaitForEvent(EVENT_SET_OPTIONS);
 223   ASSERT_TRUE(success_);
 224   serial::ConnectionInfo* info = io_handler_->connection_info();
 225   EXPECT_EQ(12345u, info->bitrate);
 226   EXPECT_EQ(serial::DATA_BITS_SEVEN, info->data_bits);
 227   EXPECT_EQ(serial::PARITY_BIT_NO, info->parity_bit);
 228   EXPECT_EQ(serial::STOP_BITS_ONE, info->stop_bits);
 229   EXPECT_TRUE(info->cts_flow_control);
 230 }
 231
 232 TEST_F(SerialConnectionTest, GetControlSignals) {
 233   connection_->GetControlSignals(base::Bind(
 234       &SerialConnectionTest::StoreControlSignals, base::Unretained(this)));
 235   serial::DeviceControlSignals* signals = io_handler_->device_control_signals();
 236   signals->dcd = true;
 237   signals->dsr = true;
 238
 239   WaitForEvent(EVENT_GOT_CONTROL_SIGNALS);
 240   ASSERT_TRUE(signals_);
 241   EXPECT_TRUE(signals_->dcd);
 242   EXPECT_FALSE(signals_->cts);
 243   EXPECT_FALSE(signals_->ri);
 244   EXPECT_TRUE(signals_->dsr);
 245 }
 246
 247 TEST_F(SerialConnectionTest, SetControlSignals) {
 248   serial::HostControlSignalsPtr signals(serial::HostControlSignals::New());
 249   signals->has_dtr = true;
 250   signals->dtr = true;
 251   signals->has_rts = true;
 252   signals->rts = true;
 253
 254   connection_->SetControlSignals(signals.Pass(),
 255                                  base::Bind(&SerialConnectionTest::StoreSuccess,
 256                                             base::Unretained(this),
 257                                             EVENT_SET_CONTROL_SIGNALS));
 258   WaitForEvent(EVENT_SET_CONTROL_SIGNALS);
 259   ASSERT_TRUE(success_);
 260   EXPECT_TRUE(io_handler_->dtr());
 261   EXPECT_TRUE(io_handler_->rts());
 262 }
 263
 264 TEST_F(SerialConnectionTest, Flush) {
 265   ASSERT_EQ(0, io_handler_->flushes());
 266   connection_->Flush(base::Bind(&SerialConnectionTest::StoreSuccess,
 267                                 base::Unretained(this),
 268                                 EVENT_FLUSHED));
 269   WaitForEvent(EVENT_FLUSHED);
 270   ASSERT_TRUE(success_);
 271   EXPECT_EQ(1, io_handler_->flushes());
 272 }
 273
 274 TEST_F(SerialConnectionTest, DisconnectWithSend) {
 275   connection_.reset();
 276   io_handler_->set_send_callback(base::Bind(base::DoNothing));
 277   ASSERT_NO_FATAL_FAILURE(Send("data"));
 278   WaitForEvent(EVENT_SEND_ERROR);
 279   EXPECT_EQ(serial::SEND_ERROR_DISCONNECTED, send_error_);
 280   EXPECT_EQ(0, bytes_sent_);
 281   EXPECT_TRUE(io_handler_->HasOneRef());
 282 }
 283
 284 TEST_F(SerialConnectionTest, DisconnectWithReceive) {
 285   connection_.reset();
 286   ASSERT_NO_FATAL_FAILURE(Receive());
 287   WaitForEvent(EVENT_RECEIVE_ERROR);
 288   EXPECT_EQ(serial::RECEIVE_ERROR_DISCONNECTED, receive_error_);
 289   EXPECT_EQ("", data_received_);
 290   EXPECT_TRUE(io_handler_->HasOneRef());
 291 }
 292
 293 TEST_F(SerialConnectionTest, Echo) {
 294   ASSERT_NO_FATAL_FAILURE(Send("data"));
 295   WaitForEvent(EVENT_DATA_SENT);
 296   EXPECT_EQ(serial::SEND_ERROR_NONE, send_error_);
 297   EXPECT_EQ(4, bytes_sent_);
 298   ASSERT_NO_FATAL_FAILURE(Receive());
 299   WaitForEvent(EVENT_DATA_RECEIVED);
 300   EXPECT_EQ("data", data_received_);
 301   EXPECT_EQ(serial::RECEIVE_ERROR_NONE, receive_error_);
 302 }
 303
 304 TEST_F(SerialConnectionTest, Cancel) {
 305   // To test that cancels are correctly passed to the IoHandler, we need a send
 306   // to be in progress because otherwise, the DataSinkReceiver would handle the
 307   // cancel internally.
 308   io_handler_->set_send_callback(
 309       base::Bind(&SerialConnectionTest::EventReceived,
 310                  base::Unretained(this),
 311                  EVENT_DATA_AT_IO_HANDLER));
 312   ASSERT_NO_FATAL_FAILURE(Send("something else"));
 313   WaitForEvent(EVENT_DATA_AT_IO_HANDLER);
 314   EXPECT_EQ(0, bytes_sent_);
 315
 316   ASSERT_TRUE(sender_->Cancel(serial::SEND_ERROR_TIMEOUT,
 317                               base::Bind(&SerialConnectionTest::EventReceived,
 318                                          base::Unretained(this),
 319                                          EVENT_CANCEL_COMPLETE)));
 320
 321   WaitForEvent(EVENT_CANCEL_COMPLETE);
 322   EXPECT_EQ(serial::SEND_ERROR_TIMEOUT, send_error_);
 323
 324   ASSERT_NO_FATAL_FAILURE(Send("data"));
 325   WaitForEvent(EVENT_DATA_SENT);
 326   EXPECT_EQ(serial::SEND_ERROR_NONE, send_error_);
 327   EXPECT_EQ(4, bytes_sent_);
 328   ASSERT_NO_FATAL_FAILURE(Receive());
 329   WaitForEvent(EVENT_DATA_RECEIVED);
 330   EXPECT_EQ("data", data_received_);
 331   EXPECT_EQ(serial::RECEIVE_ERROR_NONE, receive_error_);
 332 }
 333
 334 }  // namespace device