Fixes to WindowTreeHostMus

ui/aura/mus/in_flight_change.cc

Index: ui/aura/mus/in_flight_change.cc
diff --git a/ui/aura/mus/in_flight_change.cc b/ui/aura/mus/in_flight_change.cc
index aa70fb0413afa783b66bb27ae6ce16f84ef4d65c..67aa2e5e47d0ba6157646aced049d5b55f3a9b91 100644
--- a/ui/aura/mus/in_flight_change.cc
+++ b/ui/aura/mus/in_flight_change.cc
@@ -27,9 +27,11 @@ void InFlightChange::ChangeFailed() {}
 
 // InFlightBoundsChange -------------------------------------------------------
 
-InFlightBoundsChange::InFlightBoundsChange(WindowMus* window,
+InFlightBoundsChange::InFlightBoundsChange(WindowTreeClient* window_tree_client,
+                                           WindowMus* window,
                                            const gfx::Rect& revert_bounds)
     : InFlightChange(window, ChangeType::BOUNDS),
+      window_tree_client_(window_tree_client),
       revert_bounds_(revert_bounds) {}
 
 void InFlightBoundsChange::SetRevertValueFrom(const InFlightChange& change) {
@@ -38,7 +40,7 @@ void InFlightBoundsChange::SetRevertValueFrom(const InFlightChange& change) {
 }
 
 void InFlightBoundsChange::Revert() {
-  window()->SetBoundsFromServer(revert_bounds_);
+  window_tree_client_->SetWindowBoundsFromServer(window(), revert_bounds_);
 }
 
 // InFlightDragChange -----------------------------------------------------
@@ -186,9 +188,11 @@ void InFlightPredefinedCursorChange::Revert() {
 
 // InFlightVisibleChange -------------------------------------------------------
 
-InFlightVisibleChange::InFlightVisibleChange(WindowMus* window,
+InFlightVisibleChange::InFlightVisibleChange(WindowTreeClient* client,
+                                             WindowMus* window,
                                              bool revert_value)
     : InFlightChange(window, ChangeType::VISIBLE),
+      window_tree_client_(client),
       revert_visible_(revert_value) {}
 
 InFlightVisibleChange::~InFlightVisibleChange() {}
@@ -199,7 +203,7 @@ void InFlightVisibleChange::SetRevertValueFrom(const InFlightChange& change) {
 }
 
 void InFlightVisibleChange::Revert() {
-  window()->SetVisibleFromServer(revert_visible_);
+  window_tree_client_->SetWindowVisibleFromServer(window(), revert_visible_);
 }
 
 // InFlightOpacityChange -------------------------------------------------------