Index: content/browser/renderer_host/offscreen_canvas_surface_manager_unittest.cc |
diff --git a/content/browser/renderer_host/offscreen_canvas_surface_manager_unittest.cc b/content/browser/renderer_host/offscreen_canvas_surface_manager_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c685606abccca83cf5b10563402132f99903ac1d |
--- /dev/null |
+++ b/content/browser/renderer_host/offscreen_canvas_surface_manager_unittest.cc |
@@ -0,0 +1,166 @@ |
+// Copyright (c) 2016 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/browser_thread_impl.h" |
+#include "content/browser/compositor/test/no_transport_image_transport_factory.h" |
+#include "content/browser/renderer_host/offscreen_canvas_compositor_frame_sink.h" |
+#include "content/browser/renderer_host/offscreen_canvas_surface_impl.h" |
+#include "content/browser/renderer_host/offscreen_canvas_surface_manager.h" |
+#include "mojo/public/cpp/bindings/binding.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+#if defined(OS_ANDROID) |
+#include "content/test/mock_gpu_channel_establish_factory.h" |
+#else |
+#include "content/browser/compositor/image_transport_factory.h" |
+#endif |
+ |
+namespace content { |
+ |
+class MockMojoCompositorFrameSinkClient |
+ : public cc::mojom::MojoCompositorFrameSinkClient { |
+ public: |
+ MockMojoCompositorFrameSinkClient() : binding_(this) {} |
+ ~MockMojoCompositorFrameSinkClient() override {} |
+ |
+ cc::mojom::MojoCompositorFrameSinkClientPtr GetProxy() { |
+ return binding_.CreateInterfacePtrAndBind(); |
+ } |
+ |
+ void DidReceiveCompositorFrameAck() override {} |
+ void OnBeginFrame(const cc::BeginFrameArgs& args) override {} |
+ void ReclaimResources(const cc::ReturnedResourceArray& resources) override {} |
+ |
+ private: |
+ mojo::Binding<cc::mojom::MojoCompositorFrameSinkClient> binding_; |
+}; |
+ |
+class OffscreenCanvasSurfaceManagerTest : public testing::Test { |
+ public: |
+ int getNumSurfaceImplInstances() { |
+ return OffscreenCanvasSurfaceManager::GetInstance() |
+ ->registered_surface_instances_.size(); |
+ } |
+ const cc::SurfaceId getCurrentSurfaceId() { return current_surface_id_; } |
+ void setSurfaceId(const cc::SurfaceId& surface_id) { |
+ current_surface_id_ = surface_id; |
+ } |
+ |
+ protected: |
+ void SetUp() override; |
+ void TearDown() override; |
+ |
+ private: |
+ cc::SurfaceId current_surface_id_; |
+ std::unique_ptr<BrowserThreadImpl> ui_thread_; |
+ base::MessageLoopForUI message_loop_; |
+#if defined(OS_ANDROID) |
+ MockGpuChannelEstablishFactory gpu_channel_factory_; |
+#endif |
+}; |
+ |
+void OffscreenCanvasSurfaceManagerTest::SetUp() { |
+#if defined(OS_ANDROID) |
+ ContextProviderFactoryImpl::Initialize(&gpu_channel_factory_); |
+ ui::ContextProviderFactory::SetInstance( |
+ ContextProviderFactoryImpl::GetInstance()); |
+#else |
+ ImageTransportFactory::InitializeForUnitTests( |
+ std::unique_ptr<ImageTransportFactory>( |
+ new NoTransportImageTransportFactory)); |
+#endif |
+ ui_thread_.reset(new BrowserThreadImpl(BrowserThread::UI, &message_loop_)); |
+} |
+ |
+void OffscreenCanvasSurfaceManagerTest::TearDown() { |
+#if defined(OS_ANDROID) |
+ ui::ContextProviderFactory::SetInstance(nullptr); |
+ ContextProviderFactoryImpl::Terminate(); |
+#else |
+ ImageTransportFactory::Terminate(); |
+#endif |
+} |
+ |
+// This test mimics the workflow of OffscreenCanvas.commit() on renderer |
+// process. |
+TEST_F(OffscreenCanvasSurfaceManagerTest, |
+ SingleHTMLCanvasElementTransferToOffscreen) { |
+ // Assume that HTMLCanvasElement.transferControlToOffscreen() is triggered and |
+ // it will invoke GetSurfaceId function on OffscreenCanvasSurfaceImpl to |
+ // obtain a unique SurfaceId from browser. |
+ std::unique_ptr<OffscreenCanvasSurfaceImpl> surface_impl = |
+ base::WrapUnique(new OffscreenCanvasSurfaceImpl()); |
+ surface_impl->GetSurfaceId( |
+ base::Bind(&OffscreenCanvasSurfaceManagerTest::setSurfaceId, |
+ base::Unretained(this))); |
+ |
+ EXPECT_TRUE(this->getCurrentSurfaceId().is_valid()); |
+ EXPECT_EQ(1, this->getNumSurfaceImplInstances()); |
+ cc::FrameSinkId frame_sink_id = surface_impl.get()->frame_sink_id(); |
+ EXPECT_EQ(frame_sink_id, this->getCurrentSurfaceId().frame_sink_id()); |
+ EXPECT_EQ(surface_impl.get(), |
+ OffscreenCanvasSurfaceManager::GetInstance()->GetSurfaceInstance( |
+ frame_sink_id)); |
+ |
+ // Next, after the SurfaceId is passed onto OffscreenCanvas on worker, it will |
+ // request browser side to create a OffscreenCanvasCompositorFrameSink with |
+ // the given SurfaceId. |
+ MockMojoCompositorFrameSinkClient client_service; |
+ std::unique_ptr<OffscreenCanvasCompositorFrameSink> compositor_frame_sink = |
+ base::WrapUnique(new OffscreenCanvasCompositorFrameSink( |
dcheng
2016/11/15 07:28:29
auto compositor_frame_sink = base::MakeUnique<Offs
xlai (Olivia)
2016/11/15 17:16:18
Done.
|
+ this->getCurrentSurfaceId(), client_service.GetProxy())); |
+ surface_impl = nullptr; |
+ EXPECT_EQ(0, this->getNumSurfaceImplInstances()); |
+} |
+ |
+TEST_F(OffscreenCanvasSurfaceManagerTest, |
+ MultiHTMLCanvasElementTransferToOffscreen) { |
+ // Same scenario as above test except that now we have two HTMLCanvasElement |
+ // transferControlToOffscreen at the same time. |
+ std::unique_ptr<OffscreenCanvasSurfaceImpl> surface_impl_a = |
+ base::WrapUnique(new OffscreenCanvasSurfaceImpl()); |
+ surface_impl_a->GetSurfaceId( |
+ base::Bind(&OffscreenCanvasSurfaceManagerTest::setSurfaceId, |
+ base::Unretained(this))); |
+ cc::SurfaceId surface_id_a = this->getCurrentSurfaceId(); |
+ |
+ EXPECT_TRUE(surface_id_a.is_valid()); |
+ |
+ std::unique_ptr<OffscreenCanvasSurfaceImpl> surface_impl_b = |
+ base::WrapUnique(new OffscreenCanvasSurfaceImpl()); |
+ surface_impl_b->GetSurfaceId( |
+ base::Bind(&OffscreenCanvasSurfaceManagerTest::setSurfaceId, |
+ base::Unretained(this))); |
+ cc::SurfaceId surface_id_b = this->getCurrentSurfaceId(); |
+ |
+ EXPECT_TRUE(surface_id_b.is_valid()); |
+ EXPECT_NE(surface_id_a, surface_id_b); |
+ |
+ EXPECT_EQ(2, this->getNumSurfaceImplInstances()); |
+ EXPECT_EQ(surface_impl_a.get(), |
+ OffscreenCanvasSurfaceManager::GetInstance()->GetSurfaceInstance( |
+ surface_id_a.frame_sink_id())); |
+ EXPECT_EQ(surface_impl_b.get(), |
+ OffscreenCanvasSurfaceManager::GetInstance()->GetSurfaceInstance( |
+ surface_id_b.frame_sink_id())); |
+ |
+ // Next, one of the transferred OffscreenCanvas does commit(). |
+ MockMojoCompositorFrameSinkClient client_service_a; |
+ std::unique_ptr<OffscreenCanvasCompositorFrameSink> compositor_frame_sink_a = |
+ base::WrapUnique(new OffscreenCanvasCompositorFrameSink( |
+ surface_id_a, client_service_a.GetProxy())); |
+ |
+ // Another transferred OffscreenCanvas also does commit(). |
+ MockMojoCompositorFrameSinkClient client_service_b; |
+ std::unique_ptr<OffscreenCanvasCompositorFrameSink> compositor_frame_sink_b = |
+ base::WrapUnique(new OffscreenCanvasCompositorFrameSink( |
+ surface_id_b, client_service_b.GetProxy())); |
+ |
+ surface_impl_a = nullptr; |
+ EXPECT_EQ(1, this->getNumSurfaceImplInstances()); |
+ surface_impl_b = nullptr; |
+ EXPECT_EQ(0, this->getNumSurfaceImplInstances()); |
+} |
+ |
+} // namespace content |