src/ash/accelerators/nested_dispatcher_controller_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 "ash/accelerators/accelerator_controller.h"
   6 #include "ash/session_state_delegate.h"
   7 #include "ash/shell.h"
   8 #include "ash/shell_window_ids.h"
   9 #include "ash/test/ash_test_base.h"
  10 #include "base/bind.h"
  11 #include "base/event_types.h"
  12 #include "base/message_loop/message_loop.h"
  13 #include "ui/aura/client/dispatcher_client.h"
  14 #include "ui/aura/root_window.h"
  15 #include "ui/aura/test/test_windows.h"
  16 #include "ui/aura/window.h"
  17 #include "ui/base/accelerators/accelerator.h"
  18 #include "ui/events/event_constants.h"
  19 #include "ui/events/event_utils.h"
  20
  21 #if defined(USE_X11)
  22 #include <X11/Xlib.h>
  23 #include "ui/events/x/events_x_utils.h"
  24 #endif  // USE_X11
  25
  26 namespace ash {
  27 namespace test {
  28
  29 namespace {
  30
  31 class MockDispatcher : public base::MessageLoop::Dispatcher {
  32  public:
  33   MockDispatcher() : num_key_events_dispatched_(0) {
  34   }
  35
  36   int num_key_events_dispatched() { return num_key_events_dispatched_; }
  37
  38 #if defined(OS_WIN) || defined(USE_X11) || defined(USE_OZONE)
  39   virtual bool Dispatch(const base::NativeEvent& event) OVERRIDE {
  40     if (ui::EventTypeFromNative(event) == ui::ET_KEY_RELEASED)
  41       num_key_events_dispatched_++;
  42     return !ui::IsNoopEvent(event);
  43   }
  44 #endif
  45
  46  private:
  47   int num_key_events_dispatched_;
  48 };
  49
  50 class TestTarget : public ui::AcceleratorTarget {
  51  public:
  52   TestTarget() : accelerator_pressed_count_(0) {
  53   }
  54   virtual ~TestTarget() {
  55   }
  56
  57   int accelerator_pressed_count() const {
  58     return accelerator_pressed_count_;
  59   }
  60
  61   // Overridden from ui::AcceleratorTarget:
  62   virtual bool AcceleratorPressed(const ui::Accelerator& accelerator) OVERRIDE {
  63     accelerator_pressed_count_++;
  64     return true;
  65   }
  66   virtual bool CanHandleAccelerators() const OVERRIDE {
  67     return true;
  68   }
  69
  70  private:
  71   int accelerator_pressed_count_;
  72
  73   DISALLOW_COPY_AND_ASSIGN(TestTarget);
  74 };
  75
  76 void DispatchKeyReleaseA() {
  77   // Sending both keydown and keyup is necessary here because the accelerator
  78   // manager only checks a keyup event following a keydown event. See
  79   // ShouldHandle() in ui/base/accelerators/accelerator_manager.cc for details.
  80 #if defined(OS_WIN)
  81   MSG native_event_down = { NULL, WM_KEYDOWN, ui::VKEY_A, 0 };
  82   ash::Shell::GetPrimaryRootWindow()->PostNativeEvent(native_event_down);
  83   MSG native_event_up = { NULL, WM_KEYUP, ui::VKEY_A, 0 };
  84   ash::Shell::GetPrimaryRootWindow()->PostNativeEvent(native_event_up);
  85 #elif defined(USE_X11)
  86   XEvent native_event;
  87   ui::InitXKeyEventForTesting(ui::ET_KEY_PRESSED,
  88                               ui::VKEY_A,
  89                               0,
  90                               &native_event);
  91   aura::WindowEventDispatcher* dispatcher =
  92       ash::Shell::GetPrimaryRootWindow()->GetDispatcher();
  93   dispatcher->PostNativeEvent(&native_event);
  94   ui::InitXKeyEventForTesting(ui::ET_KEY_RELEASED,
  95                               ui::VKEY_A,
  96                               0,
  97                               &native_event);
  98   dispatcher->PostNativeEvent(&native_event);
  99 #endif
 100
 101   // Send noop event to signal dispatcher to exit.
 102   dispatcher->PostNativeEvent(ui::CreateNoopEvent());
 103 }
 104
 105 }  // namespace
 106
 107 typedef AshTestBase NestedDispatcherTest;
 108
 109 // Aura window below lock screen in z order.
 110 TEST_F(NestedDispatcherTest, AssociatedWindowBelowLockScreen) {
 111   MockDispatcher inner_dispatcher;
 112   scoped_ptr<aura::Window> associated_window(CreateTestWindowInShellWithId(0));
 113
 114   Shell::GetInstance()->session_state_delegate()->LockScreen();
 115   DispatchKeyReleaseA();
 116   aura::Window* root_window = ash::Shell::GetPrimaryRootWindow();
 117   aura::client::GetDispatcherClient(root_window)->RunWithDispatcher(
 118       &inner_dispatcher,
 119       associated_window.get(),
 120       true /* nestable_tasks_allowed */);
 121   EXPECT_EQ(0, inner_dispatcher.num_key_events_dispatched());
 122   Shell::GetInstance()->session_state_delegate()->UnlockScreen();
 123 }
 124
 125 // Aura window above lock screen in z order.
 126 TEST_F(NestedDispatcherTest, AssociatedWindowAboveLockScreen) {
 127   MockDispatcher inner_dispatcher;
 128
 129   scoped_ptr<aura::Window>mock_lock_container(
 130       CreateTestWindowInShellWithId(0));
 131   aura::test::CreateTestWindowWithId(0, mock_lock_container.get());
 132   scoped_ptr<aura::Window> associated_window(CreateTestWindowInShellWithId(0));
 133   EXPECT_TRUE(aura::test::WindowIsAbove(associated_window.get(),
 134       mock_lock_container.get()));
 135
 136   DispatchKeyReleaseA();
 137   aura::Window* root_window = ash::Shell::GetPrimaryRootWindow();
 138   aura::client::GetDispatcherClient(root_window)->RunWithDispatcher(
 139       &inner_dispatcher,
 140       associated_window.get(),
 141       true /* nestable_tasks_allowed */);
 142   EXPECT_EQ(1, inner_dispatcher.num_key_events_dispatched());
 143 }
 144
 145 // Test that the nested dispatcher handles accelerators.
 146 TEST_F(NestedDispatcherTest, AcceleratorsHandled) {
 147   MockDispatcher inner_dispatcher;
 148   aura::Window* root_window = ash::Shell::GetPrimaryRootWindow();
 149
 150   ui::Accelerator accelerator(ui::VKEY_A, ui::EF_NONE);
 151   accelerator.set_type(ui::ET_KEY_RELEASED);
 152   TestTarget target;
 153   Shell::GetInstance()->accelerator_controller()->Register(accelerator,
 154                                                            &target);
 155
 156   DispatchKeyReleaseA();
 157   aura::client::GetDispatcherClient(root_window)->RunWithDispatcher(
 158       &inner_dispatcher,
 159       root_window,
 160       true /* nestable_tasks_allowed */);
 161   EXPECT_EQ(0, inner_dispatcher.num_key_events_dispatched());
 162   EXPECT_EQ(1, target.accelerator_pressed_count());
 163 }
 164
 165 } //  namespace test
 166 } //  namespace ash