Changes ownership of WindowTreeClient

services/ui/public/cpp/tests/test_window_tree_client_setup.h

 // 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.
 
 #ifndef SERVICES_UI_PUBLIC_CPP_TESTS_TEST_WINDOW_TREE_CLIENT_SETUP_H_
 #define SERVICES_UI_PUBLIC_CPP_TESTS_TEST_WINDOW_TREE_CLIENT_SETUP_H_
 
 #include <memory>
 
 #include "base/macros.h"
@@ skipping 19 matching lines @@
 
   // Initializes the WindowTreeClient.
   void Init(WindowTreeClientDelegate* window_tree_delegate);
   void InitForWindowManager(WindowTreeClientDelegate* window_tree_delegate,
                             WindowManagerDelegate* window_manager_delegate,
                             const display::Display& display);
 
   // The WindowTree that WindowTreeClient talks to.
   TestWindowTree* window_tree() { return window_tree_.get(); }
 
+  // Returns ownership of WindowTreeClient to the caller.
+  std::unique_ptr<WindowTreeClient> OwnWindowTreeClient();
+
   WindowTreeClient* window_tree_client();
 
  private:
   // Called by both implementations of init to perform common initialization.
   void CommonInit(WindowTreeClientDelegate* window_tree_delegate,
                   WindowManagerDelegate* window_manager_delegate);
 
-  // ui::WindowTreeClientObserver:
-  void OnDidDestroyClient(ui::WindowTreeClient* client) override;
-
   std::unique_ptr<TestWindowTree> window_tree_;
 
-  // See description of WindowTreeClientDelegate for details on ownership. The
-  // WindowTreeClient may be deleted during shutdown by the test. If not,
-  // we own it and must delete it.
   std::unique_ptr<WindowTreeClient> window_tree_client_;
 
   DISALLOW_COPY_AND_ASSIGN(TestWindowTreeClientSetup);
 };
 
 }  // namespace ui
 
 #endif  // SERVICES_UI_PUBLIC_CPP_TESTS_TEST_WINDOW_TREE_CLIENT_SETUP_H_