src/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 // A hardcoded value used to verify |lock_states| is preserved.
  23 static const uint32 kTestLockStates = protocol::KeyEvent::LOCK_STATES_CAPSLOCK;
  24
  25 MATCHER_P2(EqualsUsbEvent, usb_keycode, pressed, "") {
  26   return arg.usb_keycode() == static_cast<uint32>(usb_keycode) &&
  27          arg.pressed() == pressed &&
  28          // |lock_states| is hardcoded to kTestLockStates in all key events.
  29          arg.lock_states() == kTestLockStates;
  30 }
  31
  32 MATCHER_P2(EqualsUsbEventLockStates, usb_keycode, pressed, "") {
  33   return arg.usb_keycode() == static_cast<uint32>(usb_keycode) &&
  34          arg.pressed() == pressed &&
  35          arg.lock_states() == kTestLockStates;
  36 }
  37
  38 static KeyEvent NewUsbEvent(uint32 usb_keycode,
  39                             bool pressed,
  40                             uint32 lock_states) {
  41   KeyEvent event;
  42   event.set_usb_keycode(usb_keycode);
  43   event.set_pressed(pressed);
  44   event.set_lock_states(lock_states);
  45
  46   return event;
  47 }
  48
  49 static void PressAndReleaseUsb(InputStub* input_stub, uint32 usb_keycode) {
  50   input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, true, kTestLockStates));
  51   input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, false, kTestLockStates));
  52 }
  53
  54 static void InjectTestSequence(InputStub* input_stub) {
  55   for (int i = 1; i <= 5; ++i)
  56     PressAndReleaseUsb(input_stub, i);
  57 }
  58
  59 // Verify that keys are passed through the KeyEventMapper by default.
  60 TEST(KeyEventMapperTest, NoMappingOrTrapping) {
  61   MockInputStub mock_stub;
  62   KeyEventMapper event_mapper(&mock_stub);
  63
  64   {
  65     InSequence s;
  66
  67     for (int i = 1; i <= 5; ++i) {
  68       EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, true)));
  69       EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, false)));
  70     }
  71
  72     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
  73     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  74   }
  75
  76   InjectTestSequence(&event_mapper);
  77   PressAndReleaseUsb(&event_mapper, 3);
  78 }
  79
  80 // Verify that USB keys are remapped at most once.
  81 TEST(KeyEventMapperTest, RemapKeys) {
  82   MockInputStub mock_stub;
  83   KeyEventMapper event_mapper(&mock_stub);
  84   event_mapper.RemapKey(3, 4);
  85   event_mapper.RemapKey(4, 3);
  86   event_mapper.RemapKey(5, 3);
  87
  88   {
  89     InSequence s;
  90
  91     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
  92     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
  93     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
  94     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
  95     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
  96     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
  97     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
  98     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  99     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
 100     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
 101   }
 102
 103   InjectTestSequence(&event_mapper);
 104 }
 105
 106 static void HandleTrappedKey(MockInputStub* stub, const KeyEvent& event) {
 107   stub->InjectKeyEvent(event);
 108 }
 109
 110 // Verify that trapped and mapped USB keys are trapped but not remapped.
 111 TEST(KeyEventMapperTest, TrapKeys) {
 112   MockInputStub mock_stub;
 113   MockInputStub trap_stub;
 114   KeyEventMapper event_mapper(&mock_stub);
 115   KeyEventMapper::KeyTrapCallback callback =
 116       base::Bind(&HandleTrappedKey, base::Unretained(&trap_stub));
 117   event_mapper.SetTrapCallback(callback);
 118   event_mapper.TrapKey(4, true);
 119   event_mapper.TrapKey(5, true);
 120   event_mapper.RemapKey(3, 4);
 121   event_mapper.RemapKey(4, 3);
 122   event_mapper.RemapKey(5, 3);
 123
 124   {
 125     InSequence s;
 126
 127     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
 128     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
 129     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
 130     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
 131     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
 132     EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
 133
 134     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
 135     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
 136     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, true)));
 137     EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, false)));
 138   }
 139
 140   InjectTestSequence(&event_mapper);
 141 }
 142
 143 }  // namespace remoting