Call RenderViewImpl::SetScreenOrientationForTesting to make sure that events are not sent when orie…

content/renderer/screen_orientation/mock_screen_orientation_controller.cc

 // Copyright 2014 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 "content/renderer/screen_orientation/mock_screen_orientation_controller.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
+#include "content/renderer/render_view_impl.h"
 #include "third_party/WebKit/public/platform/WebScreenOrientationListener.h"
 
 namespace content {
 
 MockScreenOrientationController::MockScreenOrientationController()
     : current_lock_(blink::WebScreenOrientationLockDefault),
       device_orientation_(blink::WebScreenOrientationPortraitPrimary),
       current_orientation_(blink::WebScreenOrientationPortraitPrimary),
-      listener_(NULL) {
+      listener_(NULL),
+      render_view_(NULL) {
   // Since MockScreenOrientationController is held by LazyInstance reference,
   // add this ref for it.
   AddRef();
 }
 
 MockScreenOrientationController::~MockScreenOrientationController() {
 }
 
 void MockScreenOrientationController::SetListener(
     blink::WebScreenOrientationListener* listener) {
@@ skipping 28 matching lines @@
 
 void MockScreenOrientationController::ResetLockSync() {
   bool will_screen_orientation_need_updating =
       !IsOrientationAllowedByCurrentLock(device_orientation_);
   current_lock_ = blink::WebScreenOrientationLockDefault;
   if (will_screen_orientation_need_updating)
     UpdateScreenOrientation(device_orientation_);
 }
 
 void MockScreenOrientationController::UpdateDeviceOrientation(
+    RenderView* render_view,
     blink::WebScreenOrientationType orientation) {
+  // Make sure that render_view_ did not change during test.
+  assert(!render_view_ || render_view_ == render_view);
+  render_view_ = render_view;
+
   if (device_orientation_ == orientation)
     return;
   device_orientation_ = orientation;
   if (!IsOrientationAllowedByCurrentLock(orientation))
     return;
   UpdateScreenOrientation(orientation);
 }
 
 void MockScreenOrientationController::UpdateScreenOrientation(
     blink::WebScreenOrientationType orientation) {
   if (current_orientation_ == orientation)
     return;
   current_orientation_ = orientation;
+  if (render_view_)
+    static_cast<RenderViewImpl*>(render_view_)
+        ->SetScreenOrientationForTesting(orientation);
   if (listener_)
     listener_->didChangeScreenOrientation(current_orientation_);
 }
 
 bool MockScreenOrientationController::IsOrientationAllowedByCurrentLock(
     blink::WebScreenOrientationType orientation) {
   if (current_lock_ == blink::WebScreenOrientationLockDefault ||
       current_lock_ == blink::WebScreenOrientationLockAny) {
     return true;
   }
@@ skipping 26 matching lines @@
     case blink::WebScreenOrientationLockLandscapePrimary:
     case blink::WebScreenOrientationLockLandscape:
       return blink::WebScreenOrientationLandscapePrimary;
     case blink::WebScreenOrientationLockLandscapeSecondary:
       return blink::WebScreenOrientationLandscapePrimary;
     default:
       return blink::WebScreenOrientationPortraitPrimary;
   }
 }
 
+void MockScreenOrientationController::ResetRenderView() {
+  render_view_ = 0;
+}
+
 } // namespace content