// 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/plugin/mac_key_event_processor.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::InSequence;
using remoting::protocol::InputStub;
using remoting::protocol::KeyEvent;
using remoting::protocol::MockInputStub;
using remoting::protocol::MouseEvent;

namespace remoting {

namespace {

const unsigned int kUsbLeftShift  = 0x0700e1;
const unsigned int kUsbLeftOption = 0x0700e2;
const unsigned int kUsbLeftCmd    = 0x0700e3;
const unsigned int kUsbRightCmd   = 0x0700e7;

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

KeyEvent MakeKeyEvent(int keycode, bool pressed) {
  KeyEvent event;
  event.set_usb_keycode(keycode);
  event.set_pressed(pressed);
  return event;
}

}  // namespace

// Test without pressing command key.
TEST(MacKeyEventProcessorTest, NoInjection) {
  MockInputStub stub;
  MacKeyEventProcessor processor(&stub);

  {
    InSequence s;

    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
  }

  // C Down and C Up.
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent('C', false));
}

// Test pressing command key and other normal keys.
TEST(MacKeyEventProcessorTest, CmdKey) {
  MockInputStub stub;
  MacKeyEventProcessor processor(&stub);

  {
    InSequence s;

    // Left command key.
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, false)));

    // Right command key.
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, false)));

    // More than one keys after CMD.
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('V', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('V', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, false)));
  }

  // Left command key.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, false));

  // Right command key.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, false));

  // More than one keys after CMD.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent('V', true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, false));
}

// Test pressing command and special keys.
TEST(MacKeyEventProcessorTest, SpecialKeys) {
  MockInputStub stub;
  MacKeyEventProcessor processor(&stub);

  {
    InSequence s;

    // Command + Shift.
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftShift, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftShift, false)));

    // Command + Option.
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftOption, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftOption, false)));
  }

  // Command + Shift.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftShift, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, false));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftShift, false));

  // Command + Option.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftOption, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, false));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftOption, false));
}

// Test pressing multiple command keys.
TEST(MacKeyEventProcessorTest, MultipleCmdKeys) {
  MockInputStub stub;
  MacKeyEventProcessor processor(&stub);

  {
    InSequence s;

    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbLeftCmd, false)));
  }

  // Test multiple CMD keys at the same time.
  // L CMD Down, C Down, R CMD Down, L CMD Up.
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbLeftCmd, false));
}

// Test press C key before command key.
TEST(MacKeyEventProcessorTest, BeforeCmdKey) {
  MockInputStub stub;
  MacKeyEventProcessor processor(&stub);

  {
    InSequence s;

    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, true)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent(kUsbRightCmd, false)));
    EXPECT_CALL(stub, InjectKeyEvent(EqualsUsbEvent('C', false)));
  }

  // Press C before command key.
  processor.InjectKeyEvent(MakeKeyEvent('C', true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, true));
  processor.InjectKeyEvent(MakeKeyEvent(kUsbRightCmd, false));
  processor.InjectKeyEvent(MakeKeyEvent('C', false));
}

}  // namespace remoting