Adding tests for USB extension API

chrome/browser/extensions/api/usb/usb_apitest.cc

+// Copyright (c) 2012 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 "chrome/browser/extensions/api/usb/usb_api.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/usb/usb_service.h"
+#include "chrome/browser/usb/usb_service_factory.h"
+#include "chrome/common/chrome_switches.h"
+#include "net/base/io_buffer.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+using testing::AnyNumber;
+using testing::_;
+
+namespace {
+
+ACTION_TEMPLATE(InvokeUsbTransferCallback,
+                HAS_1_TEMPLATE_PARAMS(int, k),
+                AND_1_VALUE_PARAMS(p1)) {
+  ::std::tr1::get<k>(args).Run(p1);
+}
+
+class MockUsbDevice : public UsbDevice {
+ public:
+  MockUsbDevice() : UsbDevice() {}
+
+  MOCK_METHOD0(Close, void(void));
+
+  MOCK_METHOD10(ControlTransfer, void(const TransferDirection direction,
+      const TransferRequestType request_type, const TransferRecipient recipient,
+      const uint8 request, const uint16 value, const uint16 index,
+      net::IOBuffer* buffer, const size_t length, const unsigned int timeout,
+      const UsbTransferCallback& callback));
+
+  MOCK_METHOD6(BulkTransfer, void(const TransferDirection direction,
+      const uint8 endpoint, net::IOBuffer* buffer, const size_t length,
+      const unsigned int timeout, const UsbTransferCallback& callback));
+
+  MOCK_METHOD6(InterruptTransfer, void(const TransferDirection direction,
+      const uint8 endpoint, net::IOBuffer* buffer, const size_t length,
+      const unsigned int timeout, const UsbTransferCallback& callback));
+
+  MOCK_METHOD8(IsochronousTransfer, void(const TransferDirection direction,
+      const uint8 endpoint, net::IOBuffer* buffer, const size_t length,
+      const unsigned int packets, const unsigned int packet_length,
+      const unsigned int timeout, const UsbTransferCallback& callback));
+
+ protected:
+  virtual ~MockUsbDevice() {}
+};
+
+class UsbApiTest : public ExtensionApiTest {
+ public:
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    ExtensionApiTest::SetUpCommandLine(command_line);
+    command_line->AppendSwitch(switches::kEnableExperimentalExtensionApis);
+  }
+
+  void SetUpOnMainThread() OVERRIDE {
+    mock_device_ = new MockUsbDevice();
+    extensions::UsbFindDeviceFunction::SetDeviceForTest(mock_device_.get());
+  }
+
+ protected:
+  scoped_refptr<MockUsbDevice> mock_device_;
+};
+
+}  // namespace
+
+IN_PROC_BROWSER_TEST_F(UsbApiTest, DeviceHandling) {
+  EXPECT_CALL(*mock_device_, Close()).Times(3);
+  ASSERT_TRUE(RunExtensionTest("usb/device_handling"));
+}
+
+IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferEvent) {
+  EXPECT_CALL(*mock_device_,
+      ControlTransfer(UsbDevice::OUTBOUND, UsbDevice::STANDARD,
+          UsbDevice::DEVICE, 1, 2, 3, _, 1, _, _))
+      .WillOnce(InvokeUsbTransferCallback<9>(USB_TRANSFER_COMPLETED));
+  EXPECT_CALL(*mock_device_,
+      BulkTransfer(UsbDevice::OUTBOUND, 1, _, 1, _, _))
+      .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED));
+  EXPECT_CALL(*mock_device_,
+      InterruptTransfer(UsbDevice::OUTBOUND, 2, _, 1, _, _))
+      .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED));
+  EXPECT_CALL(*mock_device_,
+      IsochronousTransfer(UsbDevice::OUTBOUND, 3, _, 1, 1, 1, _, _))
+      .WillOnce(InvokeUsbTransferCallback<7>(USB_TRANSFER_COMPLETED));
+  EXPECT_CALL(*mock_device_, Close()).Times(AnyNumber());
+  ASSERT_TRUE(RunExtensionTest("usb/transfer_event"));
+}
+
+IN_PROC_BROWSER_TEST_F(UsbApiTest, ZeroLengthTransfer) {
+  EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, 0, _, _))
+      .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED));
+  EXPECT_CALL(*mock_device_, Close());
+  ASSERT_TRUE(RunExtensionTest("usb/zero_length_transfer"));
+}
+
+IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferFailure) {
+  EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, _, _, _))
+      .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_ERROR))
+      .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_TIMEOUT));
+  EXPECT_CALL(*mock_device_, Close()).Times(AnyNumber());
+  ASSERT_TRUE(RunExtensionTest("usb/transfer_failure"));
+}