[2/4]Allow hotplugging of primary DRM device

ui/ozone/platform/drm/gpu/screen_manager_unittest.cc

 // Copyright 2014 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 "testing/gtest/include/gtest/gtest.h"
 #include "ui/ozone/platform/drm/gpu/crtc_controller.h"
 #include "ui/ozone/platform/drm/gpu/drm_buffer.h"
+#include "ui/ozone/platform/drm/gpu/drm_device_generator.h"
 #include "ui/ozone/platform/drm/gpu/drm_device_manager.h"
 #include "ui/ozone/platform/drm/gpu/drm_window.h"
 #include "ui/ozone/platform/drm/gpu/hardware_display_controller.h"
 #include "ui/ozone/platform/drm/gpu/screen_manager.h"
 #include "ui/ozone/platform/drm/test/mock_drm_device.h"
 
 namespace {
 
 // Create a basic mode for a 6x4 screen.
 const drmModeModeInfo kDefaultMode =
@@ skipping 16 matching lines @@
   }
 
   // Secondary is in extended mode, right-of primary.
   gfx::Rect GetSecondaryBounds() const {
     return gfx::Rect(kDefaultMode.hdisplay, 0, kDefaultMode.hdisplay,
                      kDefaultMode.vdisplay);
   }
 
   void SetUp() override {
     drm_ = new ui::MockDrmDevice();
+    device_manager_.reset(new ui::DrmDeviceManager(nullptr));
     buffer_generator_.reset(new ui::DrmBufferGenerator());
     screen_manager_.reset(new ui::ScreenManager(buffer_generator_.get()));
   }
   void TearDown() override {
     screen_manager_.reset();
     drm_ = nullptr;
   }
 
  protected:
   scoped_refptr<ui::MockDrmDevice> drm_;
+  scoped_ptr<ui::DrmDeviceManager> device_manager_;
   scoped_ptr<ui::DrmBufferGenerator> buffer_generator_;
   scoped_ptr<ui::ScreenManager> screen_manager_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ScreenManagerTest);
 };
 
 TEST_F(ScreenManagerTest, CheckWithNoControllers) {
   EXPECT_FALSE(screen_manager_->GetDisplayController(GetPrimaryBounds()));
 }
@@ skipping 237 matching lines @@
       screen_manager_->GetDisplayController(GetPrimaryBounds());
   ui::HardwareDisplayController* controller2 =
       screen_manager_->GetDisplayController(GetSecondaryBounds());
 
   EXPECT_NE(controller1, controller2);
   EXPECT_EQ(drm_, controller1->crtc_controllers()[0]->drm());
   EXPECT_EQ(drm2, controller2->crtc_controllers()[0]->drm());
 }
 
 TEST_F(ScreenManagerTest, CheckControllerToWindowMappingWithSameBounds) {
-  ui::DrmDeviceManager device_manager(drm_);
   scoped_ptr<ui::DrmWindow> window(
-      new ui::DrmWindow(1, &device_manager, screen_manager_.get()));
+      new ui::DrmWindow(1, device_manager_.get(), screen_manager_.get()));
   window->Initialize();
   window->OnBoundsChanged(GetPrimaryBounds());
   screen_manager_->AddWindow(1, window.Pass());
 
   screen_manager_->AddDisplayController(drm_, kPrimaryCrtc, kPrimaryConnector);
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   EXPECT_TRUE(screen_manager_->GetWindow(1)->GetController());
 
   window = screen_manager_->RemoveWindow(1);
   window->Shutdown();
 }
 
 TEST_F(ScreenManagerTest, CheckControllerToWindowMappingWithDifferentBounds) {
-  ui::DrmDeviceManager device_manager(drm_);
   scoped_ptr<ui::DrmWindow> window(
-      new ui::DrmWindow(1, &device_manager, screen_manager_.get()));
+      new ui::DrmWindow(1, device_manager_.get(), screen_manager_.get()));
   window->Initialize();
   gfx::Rect new_bounds = GetPrimaryBounds();
   new_bounds.Inset(0, 0, 1, 1);
   window->OnBoundsChanged(new_bounds);
   screen_manager_->AddWindow(1, window.Pass());
 
   screen_manager_->AddDisplayController(drm_, kPrimaryCrtc, kPrimaryConnector);
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   EXPECT_FALSE(screen_manager_->GetWindow(1)->GetController());
 
   window = screen_manager_->RemoveWindow(1);
   window->Shutdown();
 }
 
 TEST_F(ScreenManagerTest,
        CheckControllerToWindowMappingWithOverlappingWindows) {
-  ui::DrmDeviceManager device_manager(drm_);
   const size_t kWindowCount = 2;
   for (size_t i = 1; i < kWindowCount + 1; ++i) {
     scoped_ptr<ui::DrmWindow> window(
-        new ui::DrmWindow(1, &device_manager, screen_manager_.get()));
+        new ui::DrmWindow(i, device_manager_.get(), screen_manager_.get()));
     window->Initialize();
     window->OnBoundsChanged(GetPrimaryBounds());
     screen_manager_->AddWindow(i, window.Pass());
   }
 
   screen_manager_->AddDisplayController(drm_, kPrimaryCrtc, kPrimaryConnector);
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   bool window1_has_controller = screen_manager_->GetWindow(1)->GetController();
   bool window2_has_controller = screen_manager_->GetWindow(2)->GetController();
   // Only one of the windows can have a controller.
   EXPECT_TRUE(window1_has_controller ^ window2_has_controller);
 
   for (size_t i = 1; i < kWindowCount + 1; ++i) {
     scoped_ptr<ui::DrmWindow> window = screen_manager_->RemoveWindow(i);
     window->Shutdown();
   }
 }
 
 TEST_F(ScreenManagerTest, ShouldDissociateWindowOnControllerRemoval) {
   gfx::AcceleratedWidget window_id = 1;
-  ui::DrmDeviceManager device_manager(drm_);
-  scoped_ptr<ui::DrmWindow> window(
-      new ui::DrmWindow(window_id, &device_manager, screen_manager_.get()));
+  scoped_ptr<ui::DrmWindow> window(new ui::DrmWindow(
+      window_id, device_manager_.get(), screen_manager_.get()));
   window->Initialize();
   window->OnBoundsChanged(GetPrimaryBounds());
   screen_manager_->AddWindow(window_id, window.Pass());
 
   screen_manager_->AddDisplayController(drm_, kPrimaryCrtc, kPrimaryConnector);
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   EXPECT_TRUE(screen_manager_->GetWindow(window_id)->GetController());
 
   screen_manager_->RemoveDisplayController(drm_, kPrimaryCrtc);
 
   EXPECT_FALSE(screen_manager_->GetWindow(window_id)->GetController());
 
   window = screen_manager_->RemoveWindow(1);
   window->Shutdown();
 }
 
 TEST_F(ScreenManagerTest, EnableControllerWhenWindowHasNoBuffer) {
-  ui::DrmDeviceManager device_manager(drm_);
   scoped_ptr<ui::DrmWindow> window(
-      new ui::DrmWindow(1, &device_manager, screen_manager_.get()));
+      new ui::DrmWindow(1, device_manager_.get(), screen_manager_.get()));
   window->Initialize();
   window->OnBoundsChanged(GetPrimaryBounds());
   screen_manager_->AddWindow(1, window.Pass());
 
   screen_manager_->AddDisplayController(drm_, kPrimaryCrtc, kPrimaryConnector);
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   EXPECT_TRUE(screen_manager_->GetWindow(1)->GetController());
   // There is a buffer after initial config.
   uint32_t framebuffer = drm_->current_framebuffer();
   EXPECT_NE(0U, framebuffer);
 
   screen_manager_->ConfigureDisplayController(
       drm_, kPrimaryCrtc, kPrimaryConnector, GetPrimaryBounds().origin(),
       kDefaultMode);
 
   // There is a new buffer after we configured with the same mode but no
   // pending frames on the window.
   EXPECT_NE(framebuffer, drm_->current_framebuffer());
 
   window = screen_manager_->RemoveWindow(1);
   window->Shutdown();
 }