path: root/components/mus/ws/display_unittest.cc

// Copyright 2016 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 <stdint.h>

#include <string>

#include "base/macros.h"
#include "base/message_loop/message_loop.h"
#include "components/mus/common/types.h"
#include "components/mus/common/util.h"
#include "components/mus/public/interfaces/window_tree.mojom.h"
#include "components/mus/surfaces/surfaces_state.h"
#include "components/mus/ws/display_manager.h"
#include "components/mus/ws/ids.h"
#include "components/mus/ws/platform_display.h"
#include "components/mus/ws/platform_display_factory.h"
#include "components/mus/ws/server_window.h"
#include "components/mus/ws/test_utils.h"
#include "components/mus/ws/window_manager_state.h"
#include "components/mus/ws/window_server.h"
#include "components/mus/ws/window_server_delegate.h"
#include "components/mus/ws/window_tree.h"
#include "components/mus/ws/window_tree_binding.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/events/event.h"
#include "ui/gfx/geometry/rect.h"

namespace mus {
namespace ws {
namespace test {
namespace {

class TestWindowManagerFactory : public mojom::WindowManagerFactory {
 public:
  TestWindowManagerFactory() {}
  ~TestWindowManagerFactory() override {}

  // mojom::WindowManagerFactory:
  void CreateWindowManager(
      mus::mojom::DisplayPtr display,
      mus::mojom::WindowTreeClientRequest client) override {}

 private:
  DISALLOW_COPY_AND_ASSIGN(TestWindowManagerFactory);
};

ClientWindowId ClientWindowIdForFirstRoot(WindowTree* tree) {
  if (tree->roots().empty())
    return ClientWindowId();
  return ClientWindowIdForWindow(tree, *tree->roots().begin());
}

}  // namespace

// -----------------------------------------------------------------------------

class DisplayTest : public testing::Test {
 public:
  DisplayTest() : cursor_id_(0), platform_display_factory_(&cursor_id_) {}
  ~DisplayTest() override {}

 protected:
  // testing::Test:
  void SetUp() override {
    PlatformDisplay::set_factory_for_testing(&platform_display_factory_);
    window_server_.reset(new WindowServer(&window_server_delegate_,
                                          scoped_refptr<SurfacesState>()));
    window_server_delegate_.set_window_server(window_server_.get());
  }

 protected:
  int32_t cursor_id_;
  TestPlatformDisplayFactory platform_display_factory_;
  TestWindowServerDelegate window_server_delegate_;
  scoped_ptr<WindowServer> window_server_;
  base::MessageLoop message_loop_;
  TestWindowManagerFactory test_window_manager_factory_;

  DISALLOW_COPY_AND_ASSIGN(DisplayTest);
};

TEST_F(DisplayTest, CallsCreateDefaultDisplays) {
  const int kNumHostsToCreate = 2;
  window_server_delegate_.set_num_displays_to_create(kNumHostsToCreate);

  const UserId kTestId1 = "2";
  const UserId kTestId2 = "21";
  DisplayManager* display_manager = window_server_->display_manager();
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);
  // The first register should trigger creation of the default
  // Displays. There should be kNumHostsToCreate Displays.
  EXPECT_EQ(static_cast<size_t>(kNumHostsToCreate),
            display_manager->displays().size());

  // Each host should have a WindowManagerState for kTestId1.
  for (Display* display : display_manager->displays()) {
    EXPECT_EQ(1u, display->num_window_manger_states());
    EXPECT_TRUE(display->GetWindowManagerStateForUser(kTestId1));
    EXPECT_FALSE(display->GetWindowManagerStateForUser(kTestId2));
  }

  // Add another registry, should trigger creation of another wm.
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId2, &test_window_manager_factory_);
  for (Display* display : display_manager->displays()) {
    ASSERT_EQ(2u, display->num_window_manger_states());
    WindowManagerState* state1 =
        display->GetWindowManagerStateForUser(kTestId1);
    ASSERT_TRUE(state1);
    WindowManagerState* state2 =
        display->GetWindowManagerStateForUser(kTestId2);
    ASSERT_TRUE(state2);
    // Verify the two states have different roots.
    EXPECT_NE(state1, state2);
    EXPECT_NE(state1->root(), state2->root());
  }
}

TEST_F(DisplayTest, Destruction) {
  window_server_delegate_.set_num_displays_to_create(1);

  const UserId kTestId1 = "2";
  const UserId kTestId2 = "21";
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);

  // Add another registry, should trigger creation of another wm.
  DisplayManager* display_manager = window_server_->display_manager();
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId2, &test_window_manager_factory_);
  ASSERT_EQ(1u, display_manager->displays().size());
  Display* display = *display_manager->displays().begin();
  ASSERT_EQ(2u, display->num_window_manger_states());
  // There should be two trees, one for each windowmanager.
  EXPECT_EQ(2u, window_server_->num_trees());

  {
    WindowManagerState* state = display->GetWindowManagerStateForUser(kTestId1);
    // Destroy the tree associated with |state|. Should result in deleting
    // |state|.
    window_server_->DestroyTree(state->tree());
    ASSERT_EQ(1u, display->num_window_manger_states());
    EXPECT_FALSE(display->GetWindowManagerStateForUser(kTestId1));
    EXPECT_EQ(1u, display_manager->displays().size());
    EXPECT_EQ(1u, window_server_->num_trees());
  }

  EXPECT_FALSE(window_server_delegate_.got_on_no_more_displays());
  // Destroy the Display, which should shutdown the trees.
  window_server_->display_manager()->DestroyDisplay(display);
  EXPECT_EQ(0u, window_server_->num_trees());
  EXPECT_TRUE(window_server_delegate_.got_on_no_more_displays());
}

TEST_F(DisplayTest, EventStateResetOnUserSwitch) {
  window_server_delegate_.set_num_displays_to_create(1);

  const UserId kTestId1 = "20";
  const UserId kTestId2 = "201";
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId2, &test_window_manager_factory_);

  window_server_->user_id_tracker()->SetActiveUserId(kTestId1);

  DisplayManager* display_manager = window_server_->display_manager();
  ASSERT_EQ(1u, display_manager->displays().size());
  Display* display = *display_manager->displays().begin();
  WindowManagerState* active_wms = display->GetActiveWindowManagerState();
  ASSERT_TRUE(active_wms);
  EXPECT_EQ(kTestId1, active_wms->user_id());

  static_cast<PlatformDisplayDelegate*>(display)->OnEvent(ui::PointerEvent(
      ui::MouseEvent(ui::ET_MOUSE_PRESSED, gfx::Point(20, 25),
                     gfx::Point(20, 25), base::TimeDelta(),
                     ui::EF_LEFT_MOUSE_BUTTON, ui::EF_LEFT_MOUSE_BUTTON)));

  EXPECT_TRUE(EventDispatcherTestApi(active_wms->event_dispatcher())
                  .AreAnyPointersDown());
  EXPECT_EQ(gfx::Point(20, 25),
            active_wms->event_dispatcher()->mouse_pointer_last_location());

  // Switch the user. Should trigger resetting state in old event dispatcher
  // and update state in new event dispatcher.
  window_server_->user_id_tracker()->SetActiveUserId(kTestId2);
  EXPECT_NE(active_wms, display->GetActiveWindowManagerState());
  EXPECT_FALSE(EventDispatcherTestApi(active_wms->event_dispatcher())
                   .AreAnyPointersDown());
  active_wms = display->GetActiveWindowManagerState();
  EXPECT_EQ(kTestId2, active_wms->user_id());
  EXPECT_EQ(gfx::Point(20, 25),
            active_wms->event_dispatcher()->mouse_pointer_last_location());
  EXPECT_FALSE(EventDispatcherTestApi(active_wms->event_dispatcher())
                   .AreAnyPointersDown());
}

// Verifies capture fails when wm is inactive and succeeds when wm is active.
TEST_F(DisplayTest, SetCaptureFromWindowManager) {
  window_server_delegate_.set_num_displays_to_create(1);
  const UserId kTestId1 = "20";
  const UserId kTestId2 = "201";
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId2, &test_window_manager_factory_);
  window_server_->user_id_tracker()->SetActiveUserId(kTestId1);
  DisplayManager* display_manager = window_server_->display_manager();
  ASSERT_EQ(1u, display_manager->displays().size());
  Display* display = *display_manager->displays().begin();
  WindowManagerState* wms_for_id2 =
      display->GetWindowManagerStateForUser(kTestId2);
  ASSERT_TRUE(wms_for_id2);
  EXPECT_FALSE(wms_for_id2->IsActive());

  // Create a child of the root that we can set capture on.
  WindowTree* tree = wms_for_id2->tree();
  ClientWindowId child_window_id;
  ASSERT_TRUE(NewWindowInTree(tree, &child_window_id));

  WindowTreeTestApi(tree).EnableCapture();

  // SetCapture() should fail for user id2 as it is inactive.
  EXPECT_FALSE(tree->SetCapture(child_window_id));

  // Make the second user active and verify capture works.
  window_server_->user_id_tracker()->SetActiveUserId(kTestId2);
  EXPECT_TRUE(wms_for_id2->IsActive());
  EXPECT_TRUE(tree->SetCapture(child_window_id));
}

TEST_F(DisplayTest, FocusFailsForInactiveUser) {
  window_server_delegate_.set_num_displays_to_create(1);
  const UserId kTestId1 = "20";
  const UserId kTestId2 = "201";
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);
  TestWindowTreeClient* window_tree_client1 =
      window_server_delegate_.last_client();
  ASSERT_TRUE(window_tree_client1);
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId2, &test_window_manager_factory_);
  window_server_->user_id_tracker()->SetActiveUserId(kTestId1);
  DisplayManager* display_manager = window_server_->display_manager();
  ASSERT_EQ(1u, display_manager->displays().size());
  Display* display = *display_manager->displays().begin();
  WindowManagerState* wms_for_id2 =
      display->GetWindowManagerStateForUser(kTestId2);
  ASSERT_TRUE(wms_for_id2);
  EXPECT_FALSE(wms_for_id2->IsActive());

  // Focus should fail for windows in inactive window managers.
  EXPECT_FALSE(wms_for_id2->tree()->SetFocus(
      ClientWindowIdForFirstRoot(wms_for_id2->tree())));

  // Focus should succeed for the active window manager.
  WindowManagerState* wms_for_id1 =
      display->GetWindowManagerStateForUser(kTestId1);
  ASSERT_TRUE(wms_for_id1);
  EXPECT_TRUE(wms_for_id1->IsActive());
  EXPECT_TRUE(wms_for_id1->tree()->SetFocus(
      ClientWindowIdForFirstRoot(wms_for_id1->tree())));
}

// Verifies clients are notified of focus changes in different displays.
TEST_F(DisplayTest, CrossDisplayFocus) {
  window_server_delegate_.set_num_displays_to_create(2);
  const UserId kTestId1 = "20";
  WindowManagerFactoryRegistryTestApi(
      window_server_->window_manager_factory_registry())
      .AddService(kTestId1, &test_window_manager_factory_);
  window_server_->user_id_tracker()->SetActiveUserId(kTestId1);
  ASSERT_EQ(2u, window_server_delegate_.bindings()->size());
  TestWindowTreeBinding* window_tree_binding1 =
      (*window_server_delegate_.bindings())[0];
  Display* display1 = window_tree_binding1->tree()->GetDisplay(
      FirstRoot(window_tree_binding1->tree()));
  WindowManagerState* display1_wms =
      display1->GetWindowManagerStateForUser(kTestId1);
  TestWindowTreeBinding* window_tree_binding2 =
      (*window_server_delegate_.bindings())[1];
  Display* display2 = window_tree_binding2->tree()->GetDisplay(
      FirstRoot(window_tree_binding2->tree()));
  WindowManagerState* display2_wms =
      display2->GetWindowManagerStateForUser(kTestId1);

  // Create children in both displays.
  ClientWindowId child1_id;
  ServerWindow* child1 = NewWindowInTree(display1_wms->tree(), &child1_id);
  ASSERT_TRUE(child1);
  child1->set_can_focus(true);
  ClientWindowId child2_id;
  ServerWindow* child2 = NewWindowInTree(display2_wms->tree(), &child2_id);
  ASSERT_TRUE(child2);
  child2->set_can_focus(true);

  display1->AddActivationParent(FirstRoot(display1_wms->tree()));
  display2->AddActivationParent(FirstRoot(display2_wms->tree()));
  FirstRoot(display1_wms->tree())->set_can_focus(true);
  FirstRoot(display2_wms->tree())->set_can_focus(true);
  EXPECT_TRUE(display1_wms->tree()->SetFocus(child1_id));
  EXPECT_EQ(child1, display1->GetFocusedWindow());
  EXPECT_FALSE(display2->GetFocusedWindow());
  window_tree_binding1->client()->tracker()->changes()->clear();
  window_tree_binding2->client()->tracker()->changes()->clear();
  // Moving focus to display2 should result in notifying display1.
  EXPECT_TRUE(display2_wms->tree()->SetFocus(child2_id));
  EXPECT_EQ("Focused id=null",
            SingleChangeToDescription(
                *window_tree_binding1->client()->tracker()->changes()));
  EXPECT_EQ("", SingleChangeToDescription(
                    *window_tree_binding2->client()->tracker()->changes()));
  EXPECT_TRUE(window_tree_binding2->client()->tracker()->changes()->empty());
  window_tree_binding1->client()->tracker()->changes()->clear();
  window_tree_binding2->client()->tracker()->changes()->clear();
  EXPECT_FALSE(display1->GetFocusedWindow());
  EXPECT_EQ(child2, display2->GetFocusedWindow());
}

}  // namespace test
}  // namespace ws
}  // namespace mus