Add observers to FrameSinkManagerHost.

content/browser/renderer_host/offscreen_canvas_provider_impl_unittest.cc

+// Copyright 2017 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/browser/renderer_host/offscreen_canvas_provider_impl.h"
+
+#include "base/memory/ptr_util.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "cc/ipc/mojo_compositor_frame_sink.mojom.h"
+#include "content/browser/compositor/test/no_transport_image_transport_factory.h"
+#include "content/browser/renderer_host/offscreen_canvas_surface_impl.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/modules/offscreencanvas/offscreen_canvas_surface.mojom.h"
+
+#if !defined(OS_ANDROID)
+#include "content/browser/compositor/image_transport_factory.h"
+#endif
+
+using testing::ElementsAre;
+using testing::IsEmpty;
+
+namespace content {
+namespace {
+
+constexpr uint32_t kRendererClientId = 3;
+constexpr cc::FrameSinkId kParentFrameSink(kRendererClientId, 1);
+constexpr cc::FrameSinkId kFrameSinkA(kRendererClientId, 3);
+constexpr cc::FrameSinkId kFrameSinkB(kRendererClientId, 4);
+
+class StubOffscreenCanvasSurfaceClient
+    : public blink::mojom::OffscreenCanvasSurfaceClient {
+ public:
+  StubOffscreenCanvasSurfaceClient() : binding_(this) {}
+  ~StubOffscreenCanvasSurfaceClient() override {}
+
+  blink::mojom::OffscreenCanvasSurfaceClientPtr GetInterfacePtr() {
+    return binding_.CreateInterfacePtrAndBind();
+  }
+
+  const cc::SurfaceInfo& GetLastSurfaceInfo() const { return surface_info_; }
+
+  // blink::mojom::OffscreenCanvasSurfaceClient:
+  void OnSurfaceCreated(const cc::SurfaceInfo& surface_info) override {
+    surface_info_ = surface_info;
+  }
+
+ private:
+  mojo::Binding<blink::mojom::OffscreenCanvasSurfaceClient> binding_;
+
+  cc::SurfaceInfo surface_info_;
+
+  DISALLOW_COPY_AND_ASSIGN(StubOffscreenCanvasSurfaceClient);
+};
+
+class StubClientCompositorFrameSink
+    : public cc::mojom::MojoCompositorFrameSinkClient {
+ public:
+  StubClientCompositorFrameSink() : binding_(this) {}
+  ~StubClientCompositorFrameSink() override {}
+
+  cc::mojom::MojoCompositorFrameSinkClientPtr GetInterfacePtr() {
+    return binding_.CreateInterfacePtrAndBind();
+  }
+
+  // cc::mojom::MojoCompositorFrameSinkClient:
+  void DidReceiveCompositorFrameAck(
+      const cc::ReturnedResourceArray& resources) override {}
+  void OnBeginFrame(const cc::BeginFrameArgs& begin_frame_args) override {}
+  void ReclaimResources(const cc::ReturnedResourceArray& resources) override {}
+
+ private:
+  mojo::Binding<cc::mojom::MojoCompositorFrameSinkClient> binding_;
+
+  DISALLOW_COPY_AND_ASSIGN(StubClientCompositorFrameSink);
+};
+
+// Create a CompositorFrame suitable to send over IPC.
+cc::CompositorFrame MakeCompositorFrame() {
+  cc::CompositorFrame frame;
+  frame.metadata.begin_frame_ack.source_id =
+      cc::BeginFrameArgs::kManualSourceId;
+  frame.metadata.begin_frame_ack.sequence_number =
+      cc::BeginFrameArgs::kStartingFrameNumber;
+  frame.metadata.device_scale_factor = 1.0f;
+
+  auto render_pass = cc::RenderPass::Create();
+  render_pass->id = 1;
+  render_pass->output_rect = gfx::Rect(100, 100);
+
+  frame.render_pass_list.push_back(std::move(render_pass));
+  return frame;
+}
+
+void RunUntilIdle() {
+  base::RunLoop().RunUntilIdle();
+}
+
+}  // namespace
+
+class OffscreenCanvasProviderImplTest : public testing::Test {
+ public:
+  OffscreenCanvasProviderImpl* provider() { return provider_.get(); }
+
+  OffscreenCanvasSurfaceImpl* GetOffscreenCanvasSurface(
+      const cc::FrameSinkId& frame_sink_id) {
+    auto iter = provider_->surfaces_.find(frame_sink_id);
+    if (iter == provider_->surfaces_.end())
+      return nullptr;
+    return iter->second.get();
+  }
+
+  std::vector<cc::FrameSinkId> ActiveSurfaces() {
+    std::vector<cc::FrameSinkId> frame_sink_ids;
+    for (auto& map_entry : provider_->surfaces_) {
+      frame_sink_ids.push_back(map_entry.second->frame_sink_id());
+    }
+    std::sort(frame_sink_ids.begin(), frame_sink_ids.end());
+    return frame_sink_ids;
+  }
+
+  void DeleteOffscreenCanvasProviderImpl() { provider_.reset(); }
+
+ protected:
+  void SetUp() override {
+#if !defined(OS_ANDROID)
+    ImageTransportFactory::InitializeForUnitTests(
+        std::unique_ptr<ImageTransportFactory>(
+            new NoTransportImageTransportFactory));
+    ImageTransportFactory::GetInstance()
+        ->GetFrameSinkManagerHost()
+        ->ConnectToFrameSinkManager();
+#endif
+    provider_ =
+        base::MakeUnique<OffscreenCanvasProviderImpl>(kRendererClientId);
+  }
+  void TearDown() override {
+    provider_.reset();
+#if !defined(OS_ANDROID)
+    ImageTransportFactory::Terminate();
+#endif
+  }
+
+ private:
+  base::MessageLoop message_loop_;
+  std::unique_ptr<OffscreenCanvasProviderImpl> provider_;
+};
+
+// Mimics the workflow of OffscreenCanvas.commit() on renderer process.
+TEST_F(OffscreenCanvasProviderImplTest,
+       SingleHTMLCanvasElementTransferToOffscreen) {
+  // Mimic connection from the renderer main thread.
+  StubOffscreenCanvasSurfaceClient surface_client;
+  blink::mojom::OffscreenCanvasSurfacePtr surface;
+  provider()->CreateOffscreenCanvasSurface(kParentFrameSink, kFrameSinkA,
+                                           surface_client.GetInterfacePtr(),
+                                           mojo::MakeRequest(&surface));
+
+  OffscreenCanvasSurfaceImpl* surface_impl =
+      GetOffscreenCanvasSurface(kFrameSinkA);
+
+  // There should be a single OffscreenCanvasSurfaceImpl and it should have the
+  // provided FrameSinkId and parent FrameSinkId.
+  EXPECT_EQ(kFrameSinkA, surface_impl->frame_sink_id());
+  EXPECT_EQ(kParentFrameSink, surface_impl->parent_frame_sink_id());
+  EXPECT_THAT(ActiveSurfaces(), ElementsAre(kFrameSinkA));
+
+  // Mimic connection from the renderer main or worker thread.
+  cc::mojom::MojoCompositorFrameSinkPtr compositor_frame_sink;
+  StubClientCompositorFrameSink compositor_frame_sink_client;
+  provider()->CreateCompositorFrameSink(
+      kFrameSinkA, compositor_frame_sink_client.GetInterfacePtr(),
+      mojo::MakeRequest(&compositor_frame_sink));
+
+  // Submit a CompositorFrame with |local_id|.
+  const cc::LocalSurfaceId local_id(1, base::UnguessableToken::Create());
+  compositor_frame_sink->SubmitCompositorFrame(local_id, MakeCompositorFrame());
+
+  RunUntilIdle();
+
+  // OffscreenCanvasSurfaceImpl should have LocalSurfaceId that was submitted
+  // with the CompositorFrame.
+  EXPECT_EQ(local_id, surface_impl->local_surface_id());
+
+  // OffscreenCanvasSurfaceClient should get the new SurfaceId.
+  const auto& surface_info = surface_client.GetLastSurfaceInfo();
+  EXPECT_EQ(kFrameSinkA, surface_info.id().frame_sink_id());
+  EXPECT_EQ(local_id, surface_info.id().local_surface_id());
+
+  provider()->DestroyOffscreenCanvasSurface(kFrameSinkA);
+
+  EXPECT_THAT(ActiveSurfaces(), IsEmpty());
+}
+
+TEST_F(OffscreenCanvasProviderImplTest, ClientClosesConnection) {
+  StubOffscreenCanvasSurfaceClient surface_client;
+  blink::mojom::OffscreenCanvasSurfacePtr surface;
+  provider()->CreateOffscreenCanvasSurface(kParentFrameSink, kFrameSinkA,
+                                           surface_client.GetInterfacePtr(),
+                                           mojo::MakeRequest(&surface));
+
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_THAT(ActiveSurfaces(), ElementsAre(kFrameSinkA));
+
+  // Mimic closing the connection from the renderer.
+  surface.reset();
+
+  RunUntilIdle();
+
+  // The renderer closing the connection should destroy the
+  // OffscreenCanvasSurfaceImpl.
+  EXPECT_THAT(ActiveSurfaces(), IsEmpty());
+}
+
+// Check that destroying OffscreenCanvasProviderImpl closes all connections.
+TEST_F(OffscreenCanvasProviderImplTest, OffscreenCanvasNotDestroyed) {
+  StubOffscreenCanvasSurfaceClient surface_client;
+  blink::mojom::OffscreenCanvasSurfacePtr surface;
+  provider()->CreateOffscreenCanvasSurface(kParentFrameSink, kFrameSinkA,
+                                           surface_client.GetInterfacePtr(),
+                                           mojo::MakeRequest(&surface));
+
+  bool encountered_error = false;
+  surface.set_connection_error_handler(
+      base::Bind([](bool* encountered_error) { *encountered_error = true; },
+                 &encountered_error));
+
+  RunUntilIdle();
+
+  // There should be OffscreenCanvasSurfaceImpl and |surface| be connected.
+  EXPECT_THAT(ActiveSurfaces(), ElementsAre(kFrameSinkA));
+  EXPECT_TRUE(surface.is_bound());
+  EXPECT_FALSE(encountered_error);
+
+  // Delete OffscreenCanvasProviderImpl before all clients are disconnected.
+  DeleteOffscreenCanvasProviderImpl();
+
+  RunUntilIdle();
+
+  // This should destroy the OffscreenCanvasSurfaceImpl and close the connection
+  // to |surface|.
+  EXPECT_TRUE(encountered_error);
+}
+
+// This test tries to create an OffscreenCanvasSurfaceImpl with a client id that
+// doesn't match the renderer. This should fail.
+TEST_F(OffscreenCanvasProviderImplTest, InvalidClientId) {
+  const cc::FrameSinkId invalid_frame_sink_id(4, 3);
+  EXPECT_NE(kRendererClientId, invalid_frame_sink_id.client_id());
+
+  StubOffscreenCanvasSurfaceClient surface_client;
+  blink::mojom::OffscreenCanvasSurfacePtr surface;
+  provider()->CreateOffscreenCanvasSurface(
+      kParentFrameSink, invalid_frame_sink_id, surface_client.GetInterfacePtr(),
+      mojo::MakeRequest(&surface));
+
+  RunUntilIdle();
+
+  // No OffscreenCanvasSurfaceImpl should have been created.
+  EXPECT_THAT(ActiveSurfaces(), IsEmpty());
+}
+
+// Mimic renderer making two offscreen canvases.
+TEST_F(OffscreenCanvasProviderImplTest,
+       MultiHTMLCanvasElementTransferToOffscreen) {
+  StubOffscreenCanvasSurfaceClient surface_client_a;
+  blink::mojom::OffscreenCanvasSurfacePtr surface_a;
+  provider()->CreateOffscreenCanvasSurface(kParentFrameSink, kFrameSinkA,
+                                           surface_client_a.GetInterfacePtr(),
+                                           mojo::MakeRequest(&surface_a));
+
+  StubOffscreenCanvasSurfaceClient surface_client_b;
+  blink::mojom::OffscreenCanvasSurfacePtr surface_b;
+  provider()->CreateOffscreenCanvasSurface(kParentFrameSink, kFrameSinkB,
+                                           surface_client_b.GetInterfacePtr(),
+                                           mojo::MakeRequest(&surface_b));
+
+  RunUntilIdle();
+
+  // There should be two OffscreenCanvasSurfaceImpls created.
+  EXPECT_THAT(ActiveSurfaces(), ElementsAre(kFrameSinkA, kFrameSinkB));
+
+  // Mimic closing first connection from the renderer.
+  surface_a.reset();
+  RunUntilIdle();
+
+  EXPECT_THAT(ActiveSurfaces(), ElementsAre(kFrameSinkB));
+
+  // Mimic closing second connection from the renderer.
+  surface_b.reset();
+  RunUntilIdle();
+
+  EXPECT_THAT(ActiveSurfaces(), IsEmpty());
+}
+
+}  // namespace content