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/logging.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) {
 }
 
 void MockScreenOrientationController::SetListener(
     blink::WebScreenOrientationListener* listener) {
   listener_ = listener;
 }
 
 void MockScreenOrientationController::ResetData() {
   current_lock_ = blink::WebScreenOrientationLockDefault;
   device_orientation_ = blink::WebScreenOrientationPortraitPrimary;
@@ skipping 10 matching lines @@
 
 void MockScreenOrientationController::ResetLock() {
   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 (render_view_) {
+    // Make sure that render_view_ did not change during test.
+    assert(render_view_ == render_view);
+  } else {
+    render_view_ = render_view;
+  }

[mlamouri (slow - plz ping), 2014/05/28 13:45:20]

I guess you can do:
  assert(!render_view_ || render_view_ == render_view);
  render_view_ = render_view;

It is easier to read I think.

[ostap, 2014/05/28 19:17:05]

Done.

+
   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_);

[Inactive, 2014/05/28 12:02:20]

BTW, we should get rid of the old listener old in a follow-up, now that the
Blink side no longer relies on it.

 }
 
 bool MockScreenOrientationController::IsOrientationAllowedByCurrentLock(
     blink::WebScreenOrientationType orientation) {
   if (current_lock_ == blink::WebScreenOrientationLockDefault ||
       current_lock_ == blink::WebScreenOrientationLockAny) {
     return true;
   }
 
   switch (orientation) {
@@ skipping 24 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