summaryrefslogtreecommitdiffstats
path: root/remoting/client/key_event_mapper_unittest.cc
blob: 0f5a91d2c999a7263ba870026044448f7eccce4c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// 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 "remoting/client/key_event_mapper.h"

#include "remoting/proto/event.pb.h"
#include "remoting/protocol/protocol_mock_objects.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using ::testing::_;
using ::testing::ExpectationSet;
using ::testing::InSequence;

namespace remoting {

using protocol::InputStub;
using protocol::KeyEvent;
using protocol::MockInputStub;

MATCHER_P2(EqualsUsbEvent, usb_keycode, pressed, "") {
  return arg.usb_keycode() == static_cast<uint32>(usb_keycode) &&
         arg.pressed() == pressed;
}

static KeyEvent NewUsbEvent(uint32 usb_keycode, bool pressed) {
  KeyEvent event;
  event.set_usb_keycode(usb_keycode);
  event.set_pressed(pressed);
  return event;
}

static void PressAndReleaseUsb(InputStub* input_stub,
                               uint32 usb_keycode) {
  input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, true));
  input_stub->InjectKeyEvent(NewUsbEvent(usb_keycode, false));
}

static void InjectTestSequence(InputStub* input_stub) {
  for (int i = 1; i <= 5; ++i)
    PressAndReleaseUsb(input_stub, i);
}

// Verify that keys are passed through the KeyEventMapper by default.
TEST(KeyEventMapperTest, NoMappingOrTrapping) {
  MockInputStub mock_stub;
  KeyEventMapper event_mapper(&mock_stub);

  {
    InSequence s;

    for (int i = 1; i <= 5; ++i) {
      EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, true)));
      EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(i, false)));
    }

    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  }

  InjectTestSequence(&event_mapper);
  PressAndReleaseUsb(&event_mapper, 3);
}

// Verify that USB keys are remapped at most once.
TEST(KeyEventMapperTest, RemapKeys) {
  MockInputStub mock_stub;
  KeyEventMapper event_mapper(&mock_stub);
  event_mapper.RemapKey(3, 4);
  event_mapper.RemapKey(4, 3);
  event_mapper.RemapKey(5, 3);

  {
    InSequence s;

    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(3, false)));
  }

  InjectTestSequence(&event_mapper);
}

static void HandleTrappedKey(MockInputStub* stub, uint32 keycode, bool down) {
  stub->InjectKeyEvent(NewUsbEvent(keycode, down));
}

// Verify that trapped and mapped USB keys are trapped but not remapped.
TEST(KeyEventMapperTest, TrapKeys) {
  MockInputStub mock_stub;
  MockInputStub trap_stub;
  KeyEventMapper event_mapper(&mock_stub);
  KeyEventMapper::KeyTrapCallback callback =
      base::Bind(&HandleTrappedKey, base::Unretained(&trap_stub));
  event_mapper.SetTrapCallback(callback);
  event_mapper.TrapKey(4, true);
  event_mapper.TrapKey(5, true);
  event_mapper.RemapKey(3, 4);
  event_mapper.RemapKey(4, 3);
  event_mapper.RemapKey(5, 3);

  {
    InSequence s;

    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(1, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(2, false)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
    EXPECT_CALL(mock_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));

    EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, true)));
    EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(4, false)));
    EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, true)));
    EXPECT_CALL(trap_stub, InjectKeyEvent(EqualsUsbEvent(5, false)));
  }

  InjectTestSequence(&event_mapper);
}

}  // namespace remoting