remoting/client/key_event_mapper_unittest.cc

   1 // Copyright (c) 2012 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 "remoting/client/key_event_mapper.h"
   6
   7 #include "remoting/proto/event.pb.h"
   8 #include "remoting/protocol/protocol_mock_objects.h"
   9 #include "testing/gmock/include/gmock/gmock.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 using ::testing::_;
  13 using ::testing::ExpectationSet;
  14 using ::testing::InSequence;
  15
  16 namespace remoting {
  17
  18 using protocol::InputStub;
  19 using protocol::KeyEvent;
  20 using protocol::MockInputStub;
  21
  22 MATCHER_P2(EqualsUsbEvent, usb_keycode, pressed, "") {
  23   return arg.usb_keycode() == static_cast<uint32>(usb_keycode) &&
  24          arg.pressed() == pressed;
  25 }
  26
  27 static KeyEvent NewUsbEvent(uint32 usb_keycode, bool pressed) {
  28   KeyEvent event;
  29   event.set_usb_keycode(usb_keycode);
  30   event.set_pressed(pressed);
  31   return event;
  32 }
  33
  34 static void PressAndReleaseUsb(InputStub* input_stub,
  35                                uint32 usb_keycode) {
  36   input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, true));
  37   input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, false));
  38 }
  39
  40 static void InjectTestSequence(InputStub* input_stub) {
  41   for (int i = 1; i <= 5; ++i)
  42     PressAndReleaseUsb(input_stub, i);
  43 }
  44
  45 // Verify that keys are passed through the KeyEventMapper by default.
  46 TEST(KeyEventMapperTest, NoMappingOrTrapping) {
  47   MockInputStub mock_stub;
  48   KeyEventMapper event_mapper(&mock_stub);
  49
  50   {
  51     InSequence s;
  52
  53     for (int i = 1; i <= 5; ++i) {
  54       EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, true)));
  55       EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, false)));
  56     }
  57
  58     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
  59     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  60   }
  61
  62   InjectTestSequence(&event_mapper);
  63   PressAndReleaseUsb(&event_mapper, 3);
  64 }
  65
  66 // Verify that USB keys are remapped at most once.
  67 TEST(KeyEventMapperTest, RemapKeys) {
  68   MockInputStub mock_stub;
  69   KeyEventMapper event_mapper(&mock_stub);
  70   event_mapper.RemapKey(3, 4);
  71   event_mapper.RemapKey(4, 3);
  72   event_mapper.RemapKey(5, 3);
  73
  74   {
  75     InSequence s;
  76
  77     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
  78     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
  79     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
  80     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
  81     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
  82     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
  83     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
  84     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  85     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
  86     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  87   }
  88
  89   InjectTestSequence(&event_mapper);
  90 }
  91
  92 static void HandleTrappedKey(MockInputStub* stub, uint32 keycode, bool down) {
  93   stub->InjectKeyEvent(NewUsbEvent(keycode, down));
  94 }
  95
  96 // Verify that trapped and mapped USB keys are trapped but not remapped.
  97 TEST(KeyEventMapperTest, TrapKeys) {
  98   MockInputStub mock_stub;
  99   MockInputStub trap_stub;
 100   KeyEventMapper event_mapper(&mock_stub);
 101   KeyEventMapper::KeyTrapCallback callback =
 102       base::Bind(&HandleTrappedKey, base::Unretained(&trap_stub));
 103   event_mapper.SetTrapCallback(callback);
 104   event_mapper.TrapKey(4, true);
 105   event_mapper.TrapKey(5, true);
 106   event_mapper.RemapKey(3, 4);
 107   event_mapper.RemapKey(4, 3);
 108   event_mapper.RemapKey(5, 3);
 109
 110   {
 111     InSequence s;
 112
 113     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
 114     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
 115     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
 116     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
 117     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
 118     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
 119
 120     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
 121     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
 122     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, true)));
 123     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, false)));
 124   }
 125
 126   InjectTestSequence(&event_mapper);
 127 }
 128
 129 }  // namespace remoting