Home | History | Annotate | Download | only in accelerators 
      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 "ash/accelerators/accelerator_filter.h"
      6 
      7 #include "ash/accelerators/accelerator_controller.h"
      8 #include "ash/screenshot_delegate.h"
      9 #include "ash/shell.h"
     10 #include "ash/shell_window_ids.h"
     11 #include "ash/test/ash_test_base.h"
     12 #include "ash/wm/window_util.h"
     13 #include "base/memory/scoped_ptr.h"
     14 #include "testing/gtest/include/gtest/gtest.h"
     15 #include "ui/aura/test/aura_test_base.h"
     16 #include "ui/aura/test/event_generator.h"
     17 #include "ui/aura/test/test_windows.h"
     18 #include "ui/aura/window.h"
     19 #include "ui/gfx/rect.h"
     20 
     21 namespace ash {
     22 namespace test {
     23 
     24 namespace {
     25 
     26 class DummyScreenshotDelegate : public ScreenshotDelegate {
     27  public:
     28   DummyScreenshotDelegate() : handle_take_screenshot_count_(0) {
     29   }
     30   virtual ~DummyScreenshotDelegate() {}
     31 
     32   // Overridden from ScreenshotDelegate:
     33   virtual void HandleTakeScreenshotForAllRootWindows() OVERRIDE {
     34     ++handle_take_screenshot_count_;
     35   }
     36 
     37   virtual void HandleTakePartialScreenshot(
     38       aura::Window* window, const gfx::Rect& rect) OVERRIDE {
     39     // Do nothing because it's not tested yet.
     40   }
     41 
     42   virtual bool CanTakeScreenshot() OVERRIDE {
     43     return true;
     44   }
     45 
     46   int handle_take_screenshot_count() const {
     47     return handle_take_screenshot_count_;
     48   }
     49 
     50  private:
     51   int handle_take_screenshot_count_;
     52 
     53   DISALLOW_COPY_AND_ASSIGN(DummyScreenshotDelegate);
     54 };
     55 
     56 AcceleratorController* GetController() {
     57   return Shell::GetInstance()->accelerator_controller();
     58 }
     59 
     60 }  // namespace
     61 
     62 typedef AshTestBase AcceleratorFilterTest;
     63 
     64 // Tests if AcceleratorFilter works without a focused window.
     65 TEST_F(AcceleratorFilterTest, TestFilterWithoutFocus) {
     66   DummyScreenshotDelegate* delegate = new DummyScreenshotDelegate;
     67   GetController()->SetScreenshotDelegate(
     68       scoped_ptr<ScreenshotDelegate>(delegate).Pass());
     69   EXPECT_EQ(0, delegate->handle_take_screenshot_count());
     70 
     71   aura::test::EventGenerator generator(Shell::GetPrimaryRootWindow());
     72   // AcceleratorController calls ScreenshotDelegate::HandleTakeScreenshot() when
     73   // VKEY_PRINT is pressed. See kAcceleratorData[] in accelerator_controller.cc.
     74   generator.PressKey(ui::VKEY_PRINT, 0);
     75   EXPECT_EQ(1, delegate->handle_take_screenshot_count());
     76   generator.ReleaseKey(ui::VKEY_PRINT, 0);
     77   EXPECT_EQ(1, delegate->handle_take_screenshot_count());
     78 }
     79 
     80 // Tests if AcceleratorFilter works as expected with a focused window.
     81 TEST_F(AcceleratorFilterTest, TestFilterWithFocus) {
     82   aura::test::TestWindowDelegate test_delegate;
     83   scoped_ptr<aura::Window> window(CreateTestWindowInShellWithDelegate(
     84       &test_delegate,
     85       -1,
     86       gfx::Rect()));
     87   wm::ActivateWindow(window.get());
     88 
     89   DummyScreenshotDelegate* delegate = new DummyScreenshotDelegate;
     90   GetController()->SetScreenshotDelegate(
     91       scoped_ptr<ScreenshotDelegate>(delegate).Pass());
     92   EXPECT_EQ(0, delegate->handle_take_screenshot_count());
     93 
     94   // AcceleratorFilter should ignore the key events since the root window is
     95   // not focused.
     96   aura::test::EventGenerator generator(Shell::GetPrimaryRootWindow());
     97   generator.PressKey(ui::VKEY_PRINT, 0);
     98   EXPECT_EQ(0, delegate->handle_take_screenshot_count());
     99   generator.ReleaseKey(ui::VKEY_PRINT, 0);
    100   EXPECT_EQ(0, delegate->handle_take_screenshot_count());
    101 
    102   // Reset window before |test_delegate| gets deleted.
    103   window.reset();
    104 }
    105 
    106 // Tests if AcceleratorFilter ignores the flag for Caps Lock.
    107 TEST_F(AcceleratorFilterTest, TestCapsLockMask) {
    108   DummyScreenshotDelegate* delegate = new DummyScreenshotDelegate;
    109   GetController()->SetScreenshotDelegate(
    110       scoped_ptr<ScreenshotDelegate>(delegate).Pass());
    111   EXPECT_EQ(0, delegate->handle_take_screenshot_count());
    112 
    113   aura::test::EventGenerator generator(Shell::GetPrimaryRootWindow());
    114   generator.PressKey(ui::VKEY_PRINT, 0);
    115   EXPECT_EQ(1, delegate->handle_take_screenshot_count());
    116   generator.ReleaseKey(ui::VKEY_PRINT, 0);
    117   EXPECT_EQ(1, delegate->handle_take_screenshot_count());
    118 
    119   // Check if AcceleratorFilter ignores the mask for Caps Lock. Note that there
    120   // is no ui::EF_ mask for Num Lock.
    121   generator.PressKey(ui::VKEY_PRINT, ui::EF_CAPS_LOCK_DOWN);
    122   EXPECT_EQ(2, delegate->handle_take_screenshot_count());
    123   generator.ReleaseKey(ui::VKEY_PRINT, ui::EF_CAPS_LOCK_DOWN);
    124   EXPECT_EQ(2, delegate->handle_take_screenshot_count());
    125 }
    126 
    127 }  // namespace test
    128 }  // namespace ash
    129