Converts test_wm to use aura

ui/aura/test/aura_test_base.cc

Index: ui/aura/test/aura_test_base.cc
diff --git a/ui/aura/test/aura_test_base.cc b/ui/aura/test/aura_test_base.cc
index f571fb3cbc6543c545485bcfb5f05bce88f33e45..5182d73d742ce210390f59794a15c066797cc15f 100644
--- a/ui/aura/test/aura_test_base.cc
+++ b/ui/aura/test/aura_test_base.cc
@@ -180,7 +180,10 @@ void AuraTestBase::OnWmNewDisplay(
   window_tree_host_mus_ = std::move(window_tree_host);
 }
 
-void AuraTestBase::OnWmDisplayRemoved(Window* window) {}
+void AuraTestBase::OnWmDisplayRemoved(WindowTreeHostMus* window_tree_host) {
+  if (window_tree_host_mus_.get() == window_tree_host)
+    window_tree_host_mus_.reset();
+}
 
 void AuraTestBase::OnWmDisplayModified(const display::Display& display) {}