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),
+    : RenderViewObserver(NULL),
+      current_lock_(blink::WebScreenOrientationLockDefault),
       device_orientation_(blink::WebScreenOrientationPortraitPrimary),
       current_orientation_(blink::WebScreenOrientationPortraitPrimary),
       listener_(NULL) {
   // Since MockScreenOrientationController is held by LazyInstance reference,
   // add this ref for it.
   AddRef();
 }
 
 MockScreenOrientationController::~MockScreenOrientationController() {
 }
 
 void MockScreenOrientationController::SetListener(
     blink::WebScreenOrientationListener* listener) {
   listener_ = listener;
 }
 
 void MockScreenOrientationController::ResetData() {
+  if (render_view_impl())
+    render_view_impl()->RemoveObserver(this);
+
   current_lock_ = blink::WebScreenOrientationLockDefault;
   device_orientation_ = blink::WebScreenOrientationPortraitPrimary;
   current_orientation_ = blink::WebScreenOrientationPortraitPrimary;
 }
 
 void MockScreenOrientationController::UpdateLock(
     blink::WebScreenOrientationLockType lock) {
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&MockScreenOrientationController::UpdateLockSync, this, lock));
@@ skipping 14 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) {
+  if (this->render_view()) {
+    // Make sure that render_view_ did not change during test.
+    DCHECK_EQ(this->render_view(), render_view);
+  } else {
+    Observe(render_view);
+  }
+
   if (device_orientation_ == orientation)
     return;
   device_orientation_ = orientation;
   if (!IsOrientationAllowedByCurrentLock(orientation))
     return;
   UpdateScreenOrientation(orientation);
 }
 
+RenderViewImpl* MockScreenOrientationController::render_view_impl() const {
+  return static_cast<RenderViewImpl*>(render_view());
+}
+
 void MockScreenOrientationController::UpdateScreenOrientation(
     blink::WebScreenOrientationType orientation) {
   if (current_orientation_ == orientation)
     return;
   current_orientation_ = orientation;
+  if (render_view_impl())
+    render_view_impl()->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::OnDestruct() {
+}
+
 } // namespace content