Makes windowserver not send messages during shutdown

services/ui/ws/test_utils.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_WS_TEST_UTILS_H_
 #define SERVICES_UI_WS_TEST_UTILS_H_
 
 #include <stdint.h>
 
 #include <memory>
 #include <vector>
 
+#include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "services/ui/public/interfaces/display.mojom.h"
 #include "services/ui/public/interfaces/window_tree.mojom.h"
 #include "services/ui/ws/display.h"
 #include "services/ui/ws/display_binding.h"
 #include "services/ui/ws/drag_controller.h"
 #include "services/ui/ws/event_dispatcher.h"
 #include "services/ui/ws/platform_display.h"
 #include "services/ui/ws/platform_display_factory.h"
 #include "services/ui/ws/test_change_tracker.h"
@@ skipping 443 matching lines @@
   bool record_on_change_completed_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(TestWindowTreeClient);
 };
 
 // -----------------------------------------------------------------------------
 
 // WindowTreeBinding implementation that vends TestWindowTreeBinding.
 class TestWindowTreeBinding : public WindowTreeBinding {
  public:
-  explicit TestWindowTreeBinding(WindowTree* tree);
+  TestWindowTreeBinding(WindowTree* tree,
+                        std::unique_ptr<TestWindowTreeClient> client =
+                            base::MakeUnique<TestWindowTreeClient>());
   ~TestWindowTreeBinding() override;
 
+  std::unique_ptr<TestWindowTreeClient> ReleaseClient() {
+    return std::move(client_);
+  }
+
   WindowTree* tree() { return tree_; }
-  TestWindowTreeClient* client() { return &client_; }
+  TestWindowTreeClient* client() { return client_.get(); }
   TestWindowManager* window_manager() { return window_manager_.get(); }
 
   bool is_paused() const { return is_paused_; }
 
   // WindowTreeBinding:
   mojom::WindowManager* GetWindowManager() override;
   void SetIncomingMethodCallProcessingPaused(bool paused) override;
 
+ protected:
+  // WindowTreeBinding:
+  mojom::WindowTreeClient* CreateClientForShutdown() override;
+
  private:
   WindowTree* tree_;
-  TestWindowTreeClient client_;
+  std::unique_ptr<TestWindowTreeClient> client_;
+  // This is the client created once ResetClientForShutdown() is called.
+  std::unique_ptr<TestWindowTreeClient> client_after_reset_;
   bool is_paused_ = false;
   std::unique_ptr<TestWindowManager> window_manager_;
 
   DISALLOW_COPY_AND_ASSIGN(TestWindowTreeBinding);
 };
 
 // -----------------------------------------------------------------------------
 
 // WindowServerDelegate that creates TestWindowTreeClients.
 class TestWindowServerDelegate : public WindowServerDelegate {
@@ skipping 143 matching lines @@
 ServerWindow* NewWindowInTree(WindowTree* tree, ClientWindowId* client_id);
 ServerWindow* NewWindowInTreeWithParent(WindowTree* tree,
                                         ServerWindow* parent,
                                         ClientWindowId* client_id);
 
 }  // namespace test
 }  // namespace ws
 }  // namespace ui
 
 #endif  // SERVICES_UI_WS_TEST_UTILS_H_