Handles users rotating screen too early

ash/test/shell_test_api.cc

 // Copyright (c) 2012 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.
 
 #include "ash/test/shell_test_api.h"
 
 #include "ash/common/session/session_state_delegate.h"
 #include "ash/display/display_configuration_controller.h"
 #include "ash/root_window_controller.h"
 #include "ash/shell.h"
@@ skipping 20 matching lines @@
 }
 
 DragDropController* ShellTestApi::drag_drop_controller() {
   return shell_->drag_drop_controller_.get();
 }
 
 void ShellTestApi::DisableDisplayAnimator() {
   shell_->display_configuration_controller()->ResetAnimatorForTest();
 }
 
+ScreenRotationAnimator* ShellTestApi::GetOrMakeAnimatorByDisplayId(
+    int64_t display_id) {
+  return shell_->display_configuration_controller()
+      ->GetOrMakeAnimatorByDisplayId(display_id);
+}
+
+int ShellTestApi::DisplayIdAnimatorMapSize() {
+  return shell_->display_configuration_controller()
+      ->display_animator_map_.size();
+}
+
 void ShellTestApi::SetSessionStateDelegate(
     SessionStateDelegate* session_state_delegate) {
   shell_->session_state_delegate_.reset(session_state_delegate);
 }
 
 }  // namespace test
 }  // namespace ash