Unify and change logout/sleep/lock shortcuts

ash/accelerators/accelerator_controller_unittest.cc

 // 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 "ash/accelerators/accelerator_controller.h"
 #include "ash/accelerators/accelerator_table.h"
 #include "ash/caps_lock_delegate.h"
 #include "ash/display/display_manager.h"
 #include "ash/ime_control_delegate.h"
 #include "ash/screenshot_delegate.h"
@@ skipping 303 matching lines @@
 
  protected:
   void EnableInternalDisplay() {
     test::DisplayManagerTestApi(Shell::GetInstance()->display_manager()).
         SetFirstDisplayAsInternalDisplay();
   }
 
   static AcceleratorController* GetController();
   static bool ProcessWithContext(const ui::Accelerator& accelerator);
 
+  // Several functions to access ExitWarningHandler (as friend).
+  static void StubForTest(ExitWarningHandler& ewh) {
+    ewh.stub_timers_for_test_ = true;
+  }
+  static void SimulateTimer1Expired(ExitWarningHandler& ewh) {
+    ewh.Timer1Action();
+  }
+  static void SimulateTimer2Expired(ExitWarningHandler& ewh) {
+    ewh.Timer2Action();
+  }
+  static bool is_ui_shown(ExitWarningHandler& ewh) {
+    return !!ewh.widget_;
+  }
+  static bool is_idle(ExitWarningHandler& ewh) {
+    return ewh.state_ == ExitWarningHandler::IDLE;
+  }
+  static bool is_exiting(ExitWarningHandler& ewh) {
+    return ewh.state_ == ExitWarningHandler::EXITING;
+  }
+
  private:
   DISALLOW_COPY_AND_ASSIGN(AcceleratorControllerTest);
 };
 
 AcceleratorController* AcceleratorControllerTest::GetController() {
   return Shell::GetInstance()->accelerator_controller();
 }
 
 bool AcceleratorControllerTest::ProcessWithContext(
     const ui::Accelerator& accelerator) {
   AcceleratorController* controller = GetController();
   controller->context()->UpdateContext(accelerator);
   return controller->Process(accelerator);
 }
 
+// Quick double press of exit key => exiting
+TEST_F(AcceleratorControllerTest, ExitWarningHandlerTestDoublePress) {
+  ExitWarningHandler ewh;
+  StubForTest(ewh);
+  EXPECT_TRUE(is_idle(ewh));
+  EXPECT_FALSE(is_ui_shown(ewh));
+
+  ewh.HandleExitKey(true);
+  EXPECT_TRUE(is_ui_shown(ewh));
+  ewh.HandleExitKey(false);
+  ewh.HandleExitKey(true); // double press
+  SimulateTimer1Expired(ewh); // simulate double press timer expired
+  SimulateTimer2Expired(ewh); // simulate long hold timer expired
+  ewh.HandleExitKey(false);
+  EXPECT_FALSE(is_ui_shown(ewh));
+  EXPECT_TRUE(is_exiting(ewh));
+}
+
+// Long hold of exit key => exiting
+TEST_F(AcceleratorControllerTest, ExitWarningHandlerTestLongHold) {
+  ExitWarningHandler ewh;
+  StubForTest(ewh);
+  EXPECT_TRUE(is_idle(ewh));
+  EXPECT_FALSE(is_ui_shown(ewh));
+
+  ewh.HandleExitKey(true);
+  EXPECT_TRUE(is_ui_shown(ewh));
+  SimulateTimer1Expired(ewh); // simulate double press timer expired
+  SimulateTimer2Expired(ewh); // simulate long hold timer expired
+  ewh.HandleExitKey(false); // release after long hold
+  EXPECT_FALSE(is_ui_shown(ewh));
+  EXPECT_TRUE(is_exiting(ewh));
+}
+
+// Release of exit key before hold time limit => cancel
+TEST_F(AcceleratorControllerTest, ExitWarningHandlerTestEarlyRelease) {
+  ExitWarningHandler ewh;
+  StubForTest(ewh);
+  EXPECT_TRUE(is_idle(ewh));
+  EXPECT_FALSE(is_ui_shown(ewh));
+
+  ewh.HandleExitKey(true);
+  EXPECT_TRUE(is_ui_shown(ewh));
+  SimulateTimer1Expired(ewh); // simulate double press timer expired
+  ewh.HandleExitKey(false); // release before long hold limit
+  SimulateTimer2Expired(ewh); // simulate long hold timer expired
+  EXPECT_FALSE(is_ui_shown(ewh));
+  EXPECT_TRUE(is_idle(ewh));
+}
+
+// Release of exit key before double press limit => cancel.
+TEST_F(AcceleratorControllerTest, ExitWarningHandlerTestQuickRelease) {
+  ExitWarningHandler ewh;
+  StubForTest(ewh);
+  EXPECT_TRUE(is_idle(ewh));
+  EXPECT_FALSE(is_ui_shown(ewh));
+
+  ewh.HandleExitKey(true);
+  EXPECT_TRUE(is_ui_shown(ewh));
+  ewh.HandleExitKey(false); // release before double press limit
+  SimulateTimer1Expired(ewh); // simulate double press timer expired
+  SimulateTimer2Expired(ewh); // simulate long hold timer expired
+  EXPECT_FALSE(is_ui_shown(ewh));
+  EXPECT_TRUE(is_idle(ewh));
+}
+
 TEST_F(AcceleratorControllerTest, Register) {
   const ui::Accelerator accelerator_a(ui::VKEY_A, ui::EF_NONE);
   TestTarget target;
   GetController()->Register(accelerator_a, &target);
 
   // The registered accelerator is processed.
   EXPECT_TRUE(ProcessWithContext(accelerator_a));
   EXPECT_EQ(1, target.accelerator_pressed_count());
 }
 
@@ skipping 504 matching lines @@
       ui::Accelerator(ui::VKEY_B, ui::EF_CONTROL_DOWN | ui::EF_ALT_DOWN)));
 #if !defined(OS_LINUX)
   // ToggleDesktopFullScreen (not implemented yet on Linux)
   EXPECT_TRUE(ProcessWithContext(
       ui::Accelerator(ui::VKEY_F11, ui::EF_CONTROL_DOWN)));
 #endif //  OS_LINUX
 #endif //  !NDEBUG
 
 #if !defined(OS_WIN)
   // Exit
+  ExitWarningHandler* ewh = GetController()->GetExitWarningHandlerForTest();
+  ASSERT_TRUE(!!ewh);
+  StubForTest(*ewh);
+  EXPECT_TRUE(is_idle(*ewh));
+  EXPECT_FALSE(is_ui_shown(*ewh));
   EXPECT_TRUE(ProcessWithContext(
       ui::Accelerator(ui::VKEY_Q, ui::EF_SHIFT_DOWN | ui::EF_CONTROL_DOWN)));
+  EXPECT_FALSE(is_idle(*ewh));
+  EXPECT_TRUE(is_ui_shown(*ewh));
+  SimulateTimer1Expired(*ewh);
+  SimulateTimer2Expired(*ewh);
+  EXPECT_FALSE(is_ui_shown(*ewh));
+  EXPECT_TRUE(is_exiting(*ewh));
 #endif
 
   // New tab
   EXPECT_TRUE(ProcessWithContext(
       ui::Accelerator(ui::VKEY_T, ui::EF_CONTROL_DOWN)));
 
   // New incognito window
   EXPECT_TRUE(ProcessWithContext(
       ui::Accelerator(ui::VKEY_N, ui::EF_SHIFT_DOWN | ui::EF_CONTROL_DOWN)));
 
@@ skipping 369 matching lines @@
     EXPECT_EQ(volume_down, delegate->last_accelerator());
     EXPECT_EQ(0, delegate->handle_volume_up_count());
     EXPECT_TRUE(ProcessWithContext(volume_up));
     EXPECT_EQ(1, delegate->handle_volume_up_count());
     EXPECT_EQ(volume_up, delegate->last_accelerator());
   }
 }
 #endif
 
 }  // namespace ash