Refators display related functionality into own class

components/mus/ws/test_utils.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 "components/mus/ws/test_utils.h"
 
 #include "cc/output/copy_output_request.h"
 #include "components/mus/surfaces/surfaces_state.h"
 #include "components/mus/ws/display_binding.h"
 #include "components/mus/ws/window_manager_factory_service.h"
@@ skipping 229 matching lines @@
 }
 void TestWindowTreeBinding::SetIncomingMethodCallProcessingPaused(bool paused) {
   is_paused_ = paused;
 }
 
 // TestConnectionManagerDelegate ----------------------------------------------
 
 TestConnectionManagerDelegate::TestConnectionManagerDelegate() {}
 TestConnectionManagerDelegate::~TestConnectionManagerDelegate() {}
 
-void TestConnectionManagerDelegate::OnNoMoreRootConnections() {
-  got_on_no_more_connections_ = true;
+void TestConnectionManagerDelegate::OnNoMoreDisplays() {
+  got_on_no_more_displays_ = true;
 }
 
 scoped_ptr<WindowTreeBinding>
 TestConnectionManagerDelegate::CreateWindowTreeBindingForEmbedAtWindow(
     ws::ConnectionManager* connection_manager,
     ws::WindowTree* tree,
     mojom::WindowTreeRequest tree_request,
     mojom::WindowTreeClientPtr client) {
   scoped_ptr<TestWindowTreeBinding> binding(new TestWindowTreeBinding);
   last_binding_ = binding.get();
   return std::move(binding);
 }
 
 void TestConnectionManagerDelegate::CreateDefaultDisplays() {
   DCHECK(num_displays_to_create_);
   DCHECK(connection_manager_);
 
   for (int i = 0; i < num_displays_to_create_; ++i) {
     // Display manages its own lifetime.
     Display* display =
         new Display(connection_manager_, nullptr, scoped_refptr<GpuState>(),
                     scoped_refptr<mus::SurfacesState>());
     display->Init(nullptr);
   }
 }
 
 }  // namespace test
 }  // namespace ws
 }  // namespace mus