chromeos: fix focus for mushrome

ash/test/ash_test_helper.h

 // Copyright 2013 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 ASH_TEST_ASH_TEST_HELPER_H_
 #define ASH_TEST_ASH_TEST_HELPER_H_
 
 #include <stdint.h>
 
 #include <memory>
@@ skipping 91 matching lines @@
 
   // Version of DisplayManagerTestApi::UpdateDisplay() for mash.
   void UpdateDisplayForMash(const std::string& display_spec);
 
   display::Display GetSecondaryDisplay();
 
   // Null in classic ash.
   mus::WindowManagerApplication* window_manager_app() {
     return window_manager_app_.get();
   }
+  aura::TestWindowTreeClientSetup* window_tree_client_setup() {
+    return &window_tree_client_setup_;
+  }
 
   TestSessionControllerClient* test_session_controller_client() {
     return session_controller_client_.get();
   }
   void set_test_session_controller_client(
       std::unique_ptr<TestSessionControllerClient> session_controller_client) {
     session_controller_client_ = std::move(session_controller_client);
   }
 
   void reset_commandline() { command_line_.reset(); }
@@ skipping 51 matching lines @@
 
   std::unique_ptr<base::test::ScopedCommandLine> command_line_;
 
   DISALLOW_COPY_AND_ASSIGN(AshTestHelper);
 };
 
 }  // namespace test
 }  // namespace ash
 
 #endif  // ASH_TEST_ASH_TEST_HELPER_H_