Move SetDeviceColorSpaceForTesting into base.

This method can be easily moved into base.

BUG=1097816

Change-Id: Ic10bea81566403eed038a4c93ed54998dc5a1c68
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2542758
Commit-Queue: Dave Tapuska <dtapuska@chromium.org>
Reviewed-by: danakj <danakj@chromium.org>
Cr-Commit-Position: refs/heads/master@{#829008}

third_party/blink/renderer/core/frame/web_frame_widget_base.cc

@@ -1302,6 +1302,18 @@ void WebFrameWidgetBase::UseSynchronousResizeModeForTesting(bool enable) {
   main_data().synchronous_resize_mode_for_testing = enable;
 }
 
+void WebFrameWidgetBase::SetDeviceColorSpaceForTesting(
+    const gfx::ColorSpace& color_space) {
+  DCHECK(ForMainFrame());
+  // We are changing the device color space from the renderer, so allocate a
+  // new viz::LocalSurfaceId to avoid surface invariants violations in tests.
+  widget_base_->LayerTreeHost()->RequestNewLocalSurfaceId();
+
+  blink::ScreenInfo info = widget_base_->GetScreenInfo();
+  info.display_color_spaces = gfx::DisplayColorSpaces(color_space);
+  widget_base_->UpdateScreenInfo(info);
+}
+
 // TODO(665924): Remove direct dispatches of mouse events from
 // PointerLockController, instead passing them through EventHandler.
 void WebFrameWidgetBase::PointerLockMouseEvent(

third_party/blink/renderer/core/frame/web_frame_widget_base.h

@@ -581,6 +581,9 @@ class CORE_EXPORT WebFrameWidgetBase
   // still need the resize to happen in a synchronous fashion.
   void UseSynchronousResizeModeForTesting(bool enable);
 
+  // Sets the device color space for testing.
+  void SetDeviceColorSpaceForTesting(const gfx::ColorSpace& color_space);
+
   // Converts from DIPs to Blink coordinate space (ie. Viewport/Physical
   // pixels).
   gfx::Size DIPsToCeiledBlinkSpace(const gfx::Size& size);

third_party/blink/renderer/core/frame/web_view_frame_widget.cc

@@ -298,17 +298,6 @@ WebInputEventResult WebViewFrameWidget::HandleGestureEvent(
   return event_result;
 }
 
-void WebViewFrameWidget::SetDeviceColorSpaceForTesting(
-    const gfx::ColorSpace& color_space) {
-  // We are changing the device color space from the renderer, so allocate a
-  // new viz::LocalSurfaceId to avoid surface invariants violations in tests.
-  widget_base_->LayerTreeHost()->RequestNewLocalSurfaceId();
-
-  blink::ScreenInfo info = widget_base_->GetScreenInfo();
-  info.display_color_spaces = gfx::DisplayColorSpaces(color_space);
-  widget_base_->UpdateScreenInfo(info);
-}
-
 void WebViewFrameWidget::SetWindowRectSynchronouslyForTesting(
     const gfx::Rect& new_window_rect) {
   SetWindowRectSynchronously(new_window_rect);

third_party/blink/renderer/core/frame/web_view_frame_widget.h

@@ -78,7 +78,6 @@ class CORE_EXPORT WebViewFrameWidget : public WebFrameWidgetBase {
   // WidgetBaseClient overrides:
   void FocusChanged(bool enabled) override;
 
-  void SetDeviceColorSpaceForTesting(const gfx::ColorSpace& color_space);
   void SetWindowRect(const gfx::Rect& window_rect);
   void SetWindowRectSynchronouslyForTesting(const gfx::Rect& new_window_rect);