[DevTools] Added browser tests for adb_client_socket.cc and android_device.cc

chrome/browser/devtools/adb_client_socket_browsertest.cc

Index: chrome/browser/devtools/adb_client_socket_browsertest.cc
diff --git a/chrome/browser/devtools/adb_client_socket_browsertest.cc b/chrome/browser/devtools/adb_client_socket_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..02ac090c7f8bea5a01cc78aae03607f1dbdeb286
--- /dev/null
+++ b/chrome/browser/devtools/adb_client_socket_browsertest.cc
@@ -0,0 +1,281 @@
+// Copyright (c) 2013 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/strings/string_number_conversions.h"
+#include "chrome/browser/devtools/android_device.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/test/browser_test.h"
+#include "content/public/test/test_utils.h"
+#include "net/base/host_port_pair.h"
+#include "net/base/io_buffer.h"
+#include "net/base/ip_endpoint.h"
+#include "net/base/net_errors.h"
+#include "net/base/net_log.h"
+#include "net/socket/stream_socket.h"
+#include "net/socket/tcp_server_socket.h"
+
+using namespace net;
+
+static const char kDeviceModelCommand[] = "shell:getprop ro.product.model";
+static const char kDeviceModel[] = "Nexus 8";
+
+static const int kBufferSize = 16*1024;
+static const int kAdbPort = 5037;
+
+static const int kAdbMessageHeaderSize = 4;
+
+class AdbClientSocketTest : public InProcessBrowserTest {
+public:
+  AdbClientSocketTest() {
+    input_buffer_ = new net::GrowableIOBuffer();
+    input_buffer_->SetCapacity(kBufferSize);
+
+    output_buffer_ = new net::GrowableIOBuffer();
+
+  }
+
+  void InitServerSocket() {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
+
+    IPAddressNumber address;
+    ParseIPLiteralToNumber("127.0.0.1", &address);
+
+    IPEndPoint endpoint(address, kAdbPort);
+
+    server_socket_.reset(new TCPServerSocket(NULL, NetLog::Source()));
+    server_socket_->Listen(endpoint, 2);
+    AcceptConnection();
+
+    content::BrowserThread::PostTask(
+                  content::BrowserThread::UI, FROM_HERE,
+                  base::Bind(&AdbClientSocketTest::StartTest,
+                             base::Unretained(this)));
+  }
+
+private:
+  void StartTest() {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+    adb_thread_ = RefCountedAdbThread::GetInstance();
+
+    adb_provider_ = AndroidDeviceProvider::GetAdbDeviceProvider();
+    adb_provider_->QueryDevices(
+        base::Bind(&AdbClientSocketTest::AdbDevicesReceived,
+                   base::Unretained(this)));
+  }
+
+  void AdbDevicesReceived(
+      const AndroidDeviceProvider::AndroidDevices& received_devices) {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+#if defined(DEBUG_DEVTOOLS)
+    // Mock device is added
+    ASSERT_EQ(2U, received_devices.size());
+#else
+    ASSERT_EQ(1U, received_devices.size());
+#endif
+    android_device_ = received_devices.back();
+    ASSERT_EQ(android_device_->serial(), "1498B321301A00A");

[Vladislav Kaznacheev, 2013/10/25 15:18:19]

lets test "offline" device as well
this will probably require a separate test since if you have two devices
connected you cannot send commands such as getprop

[Dmitry Zvorygin, 2013/10/28 14:00:16]

Done.

+    ASSERT_TRUE(android_device_->is_connected());
+

[Vladislav Kaznacheev, 2013/10/25 15:18:19]

extra blank line

[Dmitry Zvorygin, 2013/10/28 14:00:16]

Done.

+
+    adb_thread_->message_loop()->PostTask(
+        FROM_HERE,
+        base::Bind(&AndroidDevice::RunCommand,
+            android_device_,
+            kDeviceModelCommand,
+            base::Bind(&AdbClientSocketTest::AdbDeviceModelReceived,
+                       base::Unretained(this))));
+  }
+
+  void AdbDeviceModelReceived(int result_code, const std::string& model) {
+    DCHECK_EQ(adb_thread_->message_loop(), base::MessageLoop::current());
+    ASSERT_EQ(result_code, 0);
+    ASSERT_EQ(model, kDeviceModel);
+    EndTest();
+  }
+
+  void EndTest() {
+    content::BrowserThread::PostTask(
+            content::BrowserThread::IO, FROM_HERE,
+            base::Bind(&AdbClientSocketTest::ShutDownSockets,
+                       base::Unretained(this)));
+  }
+
+  void ShutDownSockets() {
+    server_socket_.reset();
+
+    if (client_socket_) {
+      client_socket_->Disconnect();
+      client_socket_.reset();
+    }
+
+    content::BrowserThread::PostTask(
+        content::BrowserThread::UI, FROM_HERE,
+        base::Bind(&content::MessageLoopRunner::Quit, runner));
+  }
+
+  // Adb server implementation

[Vladislav Kaznacheev, 2013/10/25 15:18:19]

Please extract a separate class for the server. You are going to need it, trust
me.

[Dmitry Zvorygin, 2013/10/28 14:00:16]

Done.

+
+  void AcceptConnection() {
+    if (client_socket_) {
+      client_socket_->Disconnect();
+      client_socket_.reset();
+    }
+
+    int accept_result = server_socket_->Accept(&client_socket_,
+        base::Bind(&AdbClientSocketTest::OnAccepted, base::Unretained(this)));
+    if (accept_result != ERR_IO_PENDING)
+        OnAccepted(accept_result);
+  }
+
+  void OnAccepted(int result) {
+    ASSERT_EQ(result, 0);
+    ReadData();
+  }
+
+  void ReadData() {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
+
+    int read_result = client_socket_->Read(
+        input_buffer_.get(),
+        input_buffer_->RemainingCapacity(),
+        base::Bind(&AdbClientSocketTest::OnDataRead, base::Unretained(this)));
+
+    if (read_result != ERR_IO_PENDING)
+      OnDataRead(read_result);
+  }
+
+  void OnDataRead(int count) {
+    if (count <= 0) {
+      AcceptConnection();
+      return;
+    }
+
+    input_buffer_->set_offset(input_buffer_->offset() + count);
+
+    if (!TryProcessCommand()) {
+      // Posting is needed not to enter deep recursion in case too synchronous
+      // replies.
+      content::BrowserThread::PostTask( content::BrowserThread::IO, FROM_HERE,
+          base::Bind(&AdbClientSocketTest::ReadData, base::Unretained(this)));
+    }
+  }
+
+  void WriteData() {
+    ASSERT_NE(output_buffer_->offset(), output_buffer_->capacity());
+    int write_result = client_socket_->Write(
+        output_buffer_,
+        output_buffer_->capacity() - output_buffer_->offset(),
+        base::Bind(&AdbClientSocketTest::OnDataWritten,
+                   base::Unretained(this)));
+    if (write_result != ERR_IO_PENDING)
+      OnDataWritten(write_result);
+  }
+
+  void OnDataWritten(int count) {
+    if (count < 0) {
+      AcceptConnection();
+      return;
+    }
+
+    ASSERT_TRUE(count > 0);
+    ASSERT_TRUE(output_buffer_->offset() + count <= output_buffer_->capacity());
+
+    output_buffer_->set_offset(output_buffer_->offset() + count);
+    if (output_buffer_->offset() != output_buffer_->capacity())
+      WriteData();
+    else if (!TryProcessCommand())
+      ReadData();
+  }
+
+  bool TryProcessCommand() {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
+    char* data = input_buffer_->StartOfBuffer();
+    int data_size = input_buffer_->offset();
+
+    if (data_size >= kAdbMessageHeaderSize) {
+      std::string message_header(data, kAdbMessageHeaderSize);
+      int message_size;
+
+      EXPECT_TRUE(base::HexStringToInt(message_header, &message_size));
+
+      if (data_size >= message_size + kAdbMessageHeaderSize) {
+        std::string message_body(data + kAdbMessageHeaderSize, message_size );
+
+        memmove(data, data + message_size,
+                data_size - message_size - kAdbMessageHeaderSize);
+        input_buffer_->set_offset(
+            data_size - message_size - kAdbMessageHeaderSize);
+
+        ProcessCommand(message_body);
+
+        return true;
+      }
+    }
+
+    return false;
+  }
+
+  void ProcessCommand(const std::string& command) {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
+
+    if (command == "host:devices") {
+      SendResponse("1498B321301A00A\tdevice");
+    } else if (command == "host:transport:1498B321301A00A") {
+      SendResponse("");
+    } else if (command == kDeviceModelCommand) {
+      SendResponse(kDeviceModel);
+    } else {
+      NOTREACHED() << "Unknown command - " << command;
+    }
+  }
+
+  void SendResponse(const std::string& response) {
+    DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
+
+    std::stringstream response_stream;
+    response_stream << "OKAY";
+
+    int size = response.size();
+    if (size > 0) {
+      static const char kHexChars[] = "0123456789ABCDEF";
+      for (int i = 3; i >= 0; i--)
+        response_stream << kHexChars[ (size >> 4*i) & 0x0f ];
+      response_stream << response;
+    }
+
+    std::string response_data = response_stream.str();
+
+    output_buffer_->SetCapacity(response_data.size());
+    output_buffer_->set_offset(0);
+    mempcpy(output_buffer_->StartOfBuffer(), response_data.c_str(),
+            response_data.size());
+    WriteData();
+  }
+
+public:
+  scoped_refptr<content::MessageLoopRunner> runner;
+
+private:
+  scoped_refptr<RefCountedAdbThread> adb_thread_;
+  scoped_refptr<AndroidDeviceProvider> adb_provider_;
+  scoped_refptr<AndroidDevice> android_device_;
+  scoped_ptr<TCPServerSocket> server_socket_;
+  scoped_ptr<StreamSocket> client_socket_;
+  scoped_refptr<net::GrowableIOBuffer> input_buffer_;
+  scoped_refptr<net::GrowableIOBuffer> output_buffer_;
+};
+
+IN_PROC_BROWSER_TEST_F(AdbClientSocketTest, TestAdbClientSocket) {
+  runner = new content::MessageLoopRunner;
+
+  content::BrowserThread::PostTask(
+              content::BrowserThread::IO, FROM_HERE,
+              base::Bind(&AdbClientSocketTest::InitServerSocket,
+                         base::Unretained(this)));
+
+  runner->Run();
+}