Allow gaming_seat to use ozone gamepad as back-end

components/exo/gaming_seat_joydev_unittest.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.

[reveman, 2017/05/31 22:42:23]

why would we just be testing the joydev version? doesn't the same test apply to
the ozone impl?

[jkwang, 2017/06/01 19:43:15]

I have added test for device connect/disconnect.
Most part of the ozone impl just forward the event to arc++. 

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/exo/gaming_seat_joydev.h"
+#include "ash/shell.h"
+#include "base/command_line.h"
+#include "base/run_loop.h"
+#include "base/test/test_simple_task_runner.h"
+#include "components/exo/buffer.h"
+#include "components/exo/gamepad_delegate.h"
+#include "components/exo/gaming_seat_delegate.h"
+#include "components/exo/shell_surface.h"
+#include "components/exo/surface.h"
+#include "components/exo/test/exo_test_base.h"
+#include "components/exo/test/exo_test_helper.h"
+#include "device/gamepad/gamepad_test_helpers.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/aura/client/focus_client.h"
+
+namespace exo {
+namespace {
+
+class MockGamingSeatDelegate : public GamingSeatDelegate {
+ public:
+  MOCK_CONST_METHOD1(CanAcceptGamepadEventsForSurface, bool(Surface*));
+  MOCK_METHOD0(GamepadAdded, GamepadDelegate*());
+  MOCK_METHOD0(Die, void());
+  void OnGamingSeatDestroying(GamingSeat*) override { delete this; };
+  ~MockGamingSeatDelegate() { Die(); };
+};
+
+class MockGamepadDelegate : public GamepadDelegate {
+ public:
+  MockGamepadDelegate() {}
+
+  // Overridden from GamepadDelegate:
+  MOCK_METHOD0(OnRemoved, void());
+  MOCK_METHOD2(OnAxis, void(int, double));
+  MOCK_METHOD3(OnButton, void(int, bool, double));
+  MOCK_METHOD0(OnFrame, void());
+};
+
+class GamingSeatJoydevTest : public test::ExoTestBase {
+ public:
+  GamingSeatJoydevTest() {}
+
+  std::unique_ptr<device::GamepadDataFetcher> MockDataFetcherFactory() {
+    device::Gamepads initial_data;
+    std::unique_ptr<device::MockGamepadDataFetcher> fetcher(
+        new device::MockGamepadDataFetcher(initial_data));
+    mock_data_fetcher_ = fetcher.get();
+    return std::move(fetcher);
+  }
+
+  void InitializeGamingSeat(MockGamingSeatDelegate* delegate) {
+    polling_task_runner_ = new base::TestSimpleTaskRunner();
+    gaming_seat_.reset(new GamingSeatJoydev(
+        delegate, polling_task_runner_.get(),
+        base::Bind(&GamingSeatJoydevTest::MockDataFetcherFactory,
+                   base::Unretained(this))));
+    // Run the polling task runner to have it create the data fetcher.
+    polling_task_runner_->RunPendingTasks();
+  }
+
+  void DestroyGamingSeat(MockGamingSeatDelegate* delegate) {
+    EXPECT_CALL(*delegate, Die()).Times(1);
+    mock_data_fetcher_ = nullptr;
+    gaming_seat_.reset();
+    // Process tasks until polling is shut down.
+    polling_task_runner_->RunPendingTasks();
+    polling_task_runner_ = nullptr;
+  }
+
+  void SetDataAndPostToDelegate(const device::Gamepads& new_data) {
+    ASSERT_TRUE(mock_data_fetcher_ != nullptr);
+    mock_data_fetcher_->SetTestData(new_data);
+    // Run one polling cycle, which will post a task to the origin task runner.
+    polling_task_runner_->RunPendingTasks();
+    // Run origin task runner to invoke delegate.
+    base::RunLoop().RunUntilIdle();
+  }
+
+ protected:
+  std::unique_ptr<GamingSeatJoydev> gaming_seat_;
+
+  // Task runner to simulate the polling thread.
+  scoped_refptr<base::TestSimpleTaskRunner> polling_task_runner_;
+
+  // Weak reference to the mock data fetcher provided by MockDataFetcherFactory.
+  // This instance is valid until both gamepad_ and polling_task_runner_ are
+  // shut down.
+  device::MockGamepadDataFetcher* mock_data_fetcher_;
+
+  DISALLOW_COPY_AND_ASSIGN(GamingSeatJoydevTest);
+};
+
+TEST_F(GamingSeatJoydevTest, ConnectionChange) {
+  std::unique_ptr<Surface> surface(new Surface);
+  std::unique_ptr<ShellSurface> shell_surface(new ShellSurface(surface.get()));
+  gfx::Size buffer_size(10, 10);
+  std::unique_ptr<Buffer> buffer(
+      new Buffer(exo_test_helper()->CreateGpuMemoryBuffer(buffer_size)));
+  surface->Attach(buffer.get());
+  surface->Commit();
+
+  testing::StrictMock<MockGamingSeatDelegate>* gaming_seat_delegate =
+      new testing::StrictMock<MockGamingSeatDelegate>();
+  EXPECT_CALL(*gaming_seat_delegate,
+              CanAcceptGamepadEventsForSurface(testing::_))
+      .WillOnce(testing::Return(true));
+
+  InitializeGamingSeat(gaming_seat_delegate);
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate0;
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate1;
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate2;
+
+  {  // Test sequence
+    testing::InSequence s;
+    // connect gamepad 0
+    // connect gamepad 2
+    // connect gamepad 1
+    EXPECT_CALL(*gaming_seat_delegate, GamepadAdded())
+        .WillOnce(testing::Return(&gamepad_delegate0))
+        .WillOnce(testing::Return(&gamepad_delegate2))
+        .WillOnce(testing::Return(&gamepad_delegate1));
+    // disconnect gamepad 1
+    EXPECT_CALL(gamepad_delegate1, OnRemoved()).Times(1);
+    // disconnect other gamepads
+    EXPECT_CALL(gamepad_delegate0, OnRemoved()).Times(1);
+    EXPECT_CALL(gamepad_delegate2, OnRemoved()).Times(1);
+  }
+  // Gamepad connected.
+  device::Gamepads gamepad_connected;
+  gamepad_connected.items[0].connected = true;
+  gamepad_connected.items[0].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  gamepad_connected.items[2].connected = true;
+  gamepad_connected.items[2].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  gamepad_connected.items[1].connected = true;
+  gamepad_connected.items[1].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  // Gamepad 1 is dis connected
+  gamepad_connected.items[1].connected = false;
+  gamepad_connected.items[1].timestamp = 2;
+
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  // Gamepad disconnected.
+  device::Gamepads all_disconnected;
+  SetDataAndPostToDelegate(all_disconnected);
+
+  DestroyGamingSeat(gaming_seat_delegate);
+}
+
+TEST_F(GamingSeatJoydevTest, OnAxis) {
+  std::unique_ptr<Surface> surface(new Surface);
+  std::unique_ptr<ShellSurface> shell_surface(new ShellSurface(surface.get()));
+  gfx::Size buffer_size(10, 10);
+  std::unique_ptr<Buffer> buffer(
+      new Buffer(exo_test_helper()->CreateGpuMemoryBuffer(buffer_size)));
+  surface->Attach(buffer.get());
+  surface->Commit();
+
+  testing::StrictMock<MockGamingSeatDelegate>* gaming_seat_delegate =
+      new testing::StrictMock<MockGamingSeatDelegate>();
+  EXPECT_CALL(*gaming_seat_delegate,
+              CanAcceptGamepadEventsForSurface(testing::_))
+      .WillOnce(testing::Return(true));
+
+  InitializeGamingSeat(gaming_seat_delegate);
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate0;
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate2;
+
+  // connect gamepad 0 and 2
+  EXPECT_CALL(*gaming_seat_delegate, GamepadAdded())
+      .WillOnce(testing::Return(&gamepad_delegate0))
+      .WillOnce(testing::Return(&gamepad_delegate2));
+
+  device::Gamepads gamepad_connected;
+  gamepad_connected.items[0].connected = true;
+  gamepad_connected.items[0].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  gamepad_connected.items[2].connected = true;
+  gamepad_connected.items[2].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  // send axis event to 2 and then 0
+  device::Gamepads axis_moved;
+  axis_moved.items[0].connected = true;
+  axis_moved.items[0].timestamp = 1;
+  axis_moved.items[2].connected = true;
+  axis_moved.items[2].timestamp = 2;
+  axis_moved.items[2].axes_length = 1;
+  axis_moved.items[2].axes[0] = 1.0;
+
+  EXPECT_CALL(gamepad_delegate2, OnAxis(0, 1.0)).Times(1);
+  EXPECT_CALL(gamepad_delegate2, OnFrame()).Times(1);
+  SetDataAndPostToDelegate(axis_moved);
+
+  axis_moved.items[0].timestamp = 2;
+  axis_moved.items[0].axes_length = 1;
+  axis_moved.items[0].axes[0] = 2.0;
+
+  EXPECT_CALL(gamepad_delegate0, OnAxis(0, 2.0)).Times(1);
+  EXPECT_CALL(gamepad_delegate0, OnFrame()).Times(1);
+  SetDataAndPostToDelegate(axis_moved);
+
+  EXPECT_CALL(gamepad_delegate0, OnRemoved()).Times(1);
+  EXPECT_CALL(gamepad_delegate2, OnRemoved()).Times(1);
+  DestroyGamingSeat(gaming_seat_delegate);
+}
+
+TEST_F(GamingSeatJoydevTest, OnButton) {
+  std::unique_ptr<Surface> surface(new Surface);
+  std::unique_ptr<ShellSurface> shell_surface(new ShellSurface(surface.get()));
+  gfx::Size buffer_size(10, 10);
+  std::unique_ptr<Buffer> buffer(
+      new Buffer(exo_test_helper()->CreateGpuMemoryBuffer(buffer_size)));
+  surface->Attach(buffer.get());
+  surface->Commit();
+
+  testing::StrictMock<MockGamingSeatDelegate>* gaming_seat_delegate =
+      new testing::StrictMock<MockGamingSeatDelegate>();
+  EXPECT_CALL(*gaming_seat_delegate,
+              CanAcceptGamepadEventsForSurface(testing::_))
+      .WillOnce(testing::Return(true));
+
+  InitializeGamingSeat(gaming_seat_delegate);
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate0;
+  testing::StrictMock<MockGamepadDelegate> gamepad_delegate2;
+
+  // connect gamepad 0 and 2
+  EXPECT_CALL(*gaming_seat_delegate, GamepadAdded())
+      .WillOnce(testing::Return(&gamepad_delegate0))
+      .WillOnce(testing::Return(&gamepad_delegate2));
+
+  device::Gamepads gamepad_connected;
+  gamepad_connected.items[0].connected = true;
+  gamepad_connected.items[0].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  gamepad_connected.items[2].connected = true;
+  gamepad_connected.items[2].timestamp = 1;
+  SetDataAndPostToDelegate(gamepad_connected);
+
+  // send axis event to 2 and then 0
+  device::Gamepads axis_moved;
+  axis_moved.items[0].connected = true;
+  axis_moved.items[0].timestamp = 1;
+  axis_moved.items[2].connected = true;
+  axis_moved.items[2].timestamp = 2;
+
+  axis_moved.items[2].buttons_length = 1;
+  axis_moved.items[2].buttons[0].pressed = true;
+  axis_moved.items[2].buttons[0].value = 1.0;
+
+  EXPECT_CALL(gamepad_delegate2, OnButton(0, true, 1.0)).Times(1);
+  EXPECT_CALL(gamepad_delegate2, OnFrame()).Times(1);
+  SetDataAndPostToDelegate(axis_moved);
+
+  axis_moved.items[0].timestamp = 2;
+  axis_moved.items[0].buttons_length = 1;
+  axis_moved.items[0].buttons[0].pressed = true;
+  axis_moved.items[0].buttons[0].value = 2.0;
+
+  EXPECT_CALL(gamepad_delegate0, OnButton(0, true, 2.0)).Times(1);
+  EXPECT_CALL(gamepad_delegate0, OnFrame()).Times(1);
+  SetDataAndPostToDelegate(axis_moved);
+
+  EXPECT_CALL(gamepad_delegate0, OnRemoved()).Times(1);
+  EXPECT_CALL(gamepad_delegate2, OnRemoved()).Times(1);
+
+  DestroyGamingSeat(gaming_seat_delegate);
+}
+
+TEST_F(GamingSeatJoydevTest, OnWindowFocused) {
+  // Create surface and move focus to it.
+  std::unique_ptr<Surface> surface(new Surface);
+  std::unique_ptr<ShellSurface> shell_surface(new ShellSurface(surface.get()));
+  gfx::Size buffer_size(10, 10);
+  std::unique_ptr<Buffer> buffer(
+      new Buffer(exo_test_helper()->CreateGpuMemoryBuffer(buffer_size)));
+  surface->Attach(buffer.get());
+  surface->Commit();
+
+  testing::StrictMock<MockGamingSeatDelegate>* gaming_seat_delegate =
+      new testing::StrictMock<MockGamingSeatDelegate>();
+  EXPECT_CALL(*gaming_seat_delegate,
+              CanAcceptGamepadEventsForSurface(testing::_))
+      .WillOnce(testing::Return(true));
+
+  InitializeGamingSeat(gaming_seat_delegate);
+
+  // In focus. Should be polling indefinitely, check a couple of time that the
+  // poll task is re-posted.
+  for (size_t i = 0; i < 5; ++i) {
+    polling_task_runner_->RunPendingTasks();
+    ASSERT_TRUE(polling_task_runner_->HasPendingTask());
+  }
+
+  // Remove focus from window.
+  aura::client::FocusClient* focus_client =
+      aura::client::GetFocusClient(ash::Shell::GetPrimaryRootWindow());
+  focus_client->FocusWindow(nullptr);
+
+  // Run EnablePolling and OnPoll task, no more polls should be scheduled.
+  // In the first round of RunPendingTasks we will execute
+  // EnablePollingOnPollingThread, which will cause the polling to stop being
+  // scheduled in the next round.
+  polling_task_runner_->RunPendingTasks();
+  polling_task_runner_->RunPendingTasks();
+  ASSERT_FALSE(polling_task_runner_->HasPendingTask());
+
+  DestroyGamingSeat(gaming_seat_delegate);
+}
+
+}  // namespace
+}  // namespace exo