WIP: Surface Synchronization System

cc/surfaces/surface_unittest.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 "base/memory/ptr_util.h"
+#include "cc/surfaces/display_compositor_lock_manager.h"
 #include "cc/surfaces/surface.h"
 #include "cc/surfaces/surface_factory.h"
 #include "cc/surfaces/surface_factory_client.h"
 #include "cc/surfaces/surface_id_allocator.h"
 #include "cc/surfaces/surface_manager.h"
+#include "cc/test/begin_frame_args_test.h"
+#include "cc/test/fake_external_begin_frame_source.h"
 #include "cc/test/scheduler_test_common.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace cc {
 namespace {
 
 static constexpr FrameSinkId kArbitraryFrameSinkId(1, 1);
 
+CompositorFrame MakeCompositorFrame(
+    std::vector<SurfaceId> referenced_surfaces) {
+  CompositorFrame compositor_frame;
+  compositor_frame.metadata.referenced_surfaces =
+      std::move(referenced_surfaces);
+  return compositor_frame;
+}
+
 class FakeSurfaceFactoryClient : public SurfaceFactoryClient {
  public:
   FakeSurfaceFactoryClient() : begin_frame_source_(nullptr) {}
 
   void ReturnResources(const ReturnedResourceArray& resources) override {}
 
   void SetBeginFrameSource(BeginFrameSource* begin_frame_source) override {
     begin_frame_source_ = begin_frame_source;
   }
 
   BeginFrameSource* begin_frame_source() { return begin_frame_source_; }
 
  private:
   BeginFrameSource* begin_frame_source_;
 };
 
+// Surface 1 is blocked on Surface 2 and Surface 3.
+TEST(SurfaceTest, DisplayCompositorLocking1) {
+  SurfaceManager manager;
+  std::unique_ptr<FakeExternalBeginFrameSource> begin_frame_source(
+      new FakeExternalBeginFrameSource(0.f, false));
+
+  std::unique_ptr<DisplayCompositorLockManager> lock_manager(
+      new DisplayCompositorLockManager(&manager, begin_frame_source.get()));
+  manager.SetLockManager(std::move(lock_manager));
+
+  LocalFrameId local_frame_id1(6, base::UnguessableToken::Create());
+  SurfaceId surface_id1(FrameSinkId(1, 1), local_frame_id1);
+
+  LocalFrameId local_frame_id2(7, base::UnguessableToken::Create());
+  SurfaceId surface_id2(FrameSinkId(2, 2), local_frame_id2);
+
+  LocalFrameId local_frame_id3(8, base::UnguessableToken::Create());
+  SurfaceId surface_id3(FrameSinkId(3, 3), local_frame_id3);
+
+  FakeSurfaceFactoryClient surface_factory_client1;
+  SurfaceFactory factory1(FrameSinkId(1, 1), &manager,
+                          &surface_factory_client1);
+  factory1.SubmitCompositorFrame(
+      local_frame_id1, MakeCompositorFrame({surface_id2, surface_id3}),
+      SurfaceFactory::DrawCallback());
+  EXPECT_TRUE(manager.lock_manager()->observing_begin_frames());
+
+  // |factory1| is blocked on |surface_id2| and |surface_id3|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client2;
+  SurfaceFactory factory2(FrameSinkId(2, 2), &manager,
+                          &surface_factory_client2);
+  factory2.SubmitCompositorFrame(local_frame_id2,
+                                 MakeCompositorFrame(std::vector<SurfaceId>()),
+                                 SurfaceFactory::DrawCallback());
+
+  EXPECT_TRUE(manager.lock_manager()->observing_begin_frames());
+  EXPECT_TRUE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory1| is blocked on just |surface_id3|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client3;
+  SurfaceFactory factory3(FrameSinkId(3, 3), &manager,
+                          &surface_factory_client3);
+  factory3.SubmitCompositorFrame(local_frame_id3,
+                                 MakeCompositorFrame(std::vector<SurfaceId>()),
+                                 SurfaceFactory::DrawCallback());
+  EXPECT_FALSE(manager.lock_manager()->observing_begin_frames());
+
+  // |factory1|'s Frame is now active.
+  EXPECT_TRUE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  factory1.EvictSurface();
+  factory2.EvictSurface();
+  factory3.EvictSurface();
+}
+
+// Surface 1 is blocked on Surface 2 which is blocked on Surface 3.
+TEST(SurfaceTest, DisplayCompositorLocking2) {
+  SurfaceManager manager;
+  std::unique_ptr<FakeExternalBeginFrameSource> begin_frame_source(
+      new FakeExternalBeginFrameSource(0.f, false));
+
+  std::unique_ptr<DisplayCompositorLockManager> lock_manager(
+      new DisplayCompositorLockManager(&manager, begin_frame_source.get()));
+  manager.SetLockManager(std::move(lock_manager));
+
+  LocalFrameId local_frame_id1(6, base::UnguessableToken::Create());
+  SurfaceId surface_id1(FrameSinkId(1, 1), local_frame_id1);
+
+  LocalFrameId local_frame_id2(7, base::UnguessableToken::Create());
+  SurfaceId surface_id2(FrameSinkId(2, 2), local_frame_id2);
+
+  LocalFrameId local_frame_id3(8, base::UnguessableToken::Create());
+  SurfaceId surface_id3(FrameSinkId(3, 3), local_frame_id3);
+
+  FakeSurfaceFactoryClient surface_factory_client1;
+  SurfaceFactory factory1(FrameSinkId(1, 1), &manager,
+                          &surface_factory_client1);
+  factory1.SubmitCompositorFrame(local_frame_id1,
+                                 MakeCompositorFrame({surface_id2}),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory1| is blocked on |surface_id2|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client2;
+  SurfaceFactory factory2(FrameSinkId(2, 2), &manager,
+                          &surface_factory_client2);
+  factory2.SubmitCompositorFrame(local_frame_id2,
+                                 MakeCompositorFrame({surface_id3}),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory2| is blocked on |surface_id3|.
+  EXPECT_FALSE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory1| is still blocked on just |surface_id2|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client3;
+  SurfaceFactory factory3(FrameSinkId(3, 3), &manager,
+                          &surface_factory_client3);
+  CompositorFrame frame3;
+  factory3.SubmitCompositorFrame(local_frame_id3,
+                                 MakeCompositorFrame(std::vector<SurfaceId>()),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory1|'s Frame is now active.
+  EXPECT_TRUE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory2|'s Frame is now active.
+  EXPECT_TRUE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  factory1.EvictSurface();
+  factory2.EvictSurface();
+  factory3.EvictSurface();
+}
+
+// Surface 1 and Surface 2 are blocked on Surface 3.
+TEST(SurfaceTest, DisplayCompositorLocking3) {
+  SurfaceManager manager;
+  std::unique_ptr<FakeExternalBeginFrameSource> begin_frame_source(
+      new FakeExternalBeginFrameSource(0.f, false));
+
+  std::unique_ptr<DisplayCompositorLockManager> lock_manager(
+      new DisplayCompositorLockManager(&manager, begin_frame_source.get()));
+  manager.SetLockManager(std::move(lock_manager));
+
+  LocalFrameId local_frame_id1(6, base::UnguessableToken::Create());
+  SurfaceId surface_id1(FrameSinkId(1, 1), local_frame_id1);
+
+  LocalFrameId local_frame_id2(7, base::UnguessableToken::Create());
+  SurfaceId surface_id2(FrameSinkId(2, 2), local_frame_id2);
+
+  LocalFrameId local_frame_id3(8, base::UnguessableToken::Create());
+  SurfaceId surface_id3(FrameSinkId(3, 3), local_frame_id3);
+
+  FakeSurfaceFactoryClient surface_factory_client1;
+  SurfaceFactory factory1(FrameSinkId(1, 1), &manager,
+                          &surface_factory_client1);
+  factory1.SubmitCompositorFrame(local_frame_id1,
+                                 MakeCompositorFrame({surface_id3}),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory1| is blocked on |surface_id3|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client2;
+  SurfaceFactory factory2(FrameSinkId(2, 2), &manager,
+                          &surface_factory_client2);
+  factory2.SubmitCompositorFrame(local_frame_id2,
+                                 MakeCompositorFrame({surface_id3}),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory2| is blocked on |surface_id3|.
+  EXPECT_FALSE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory1| is still blocked on |surface_id3|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client3;
+  SurfaceFactory factory3(FrameSinkId(3, 3), &manager,
+                          &surface_factory_client3);
+  factory3.SubmitCompositorFrame(local_frame_id3,
+                                 MakeCompositorFrame(std::vector<SurfaceId>()),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory1|'s Frame is now active.
+  EXPECT_TRUE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory2|'s Frame is now active.
+  EXPECT_TRUE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  factory1.EvictSurface();
+  factory2.EvictSurface();
+  factory3.EvictSurface();
+}
+
+// |factory1| is blocked on |surface_id2| but the deadline hits.
+TEST(SurfaceTest, DisplayCompositorLocking4) {
+  SurfaceManager manager;
+  std::unique_ptr<FakeExternalBeginFrameSource> begin_frame_source(
+      new FakeExternalBeginFrameSource(0.f, false));
+
+  std::unique_ptr<DisplayCompositorLockManager> lock_manager(
+      new DisplayCompositorLockManager(&manager, begin_frame_source.get()));
+  manager.SetLockManager(std::move(lock_manager));
+
+  LocalFrameId local_frame_id1(6, base::UnguessableToken::Create());
+  SurfaceId surface_id1(FrameSinkId(1, 1), local_frame_id1);
+
+  LocalFrameId local_frame_id2(7, base::UnguessableToken::Create());
+  SurfaceId surface_id2(FrameSinkId(2, 2), local_frame_id2);
+
+  LocalFrameId local_frame_id3(8, base::UnguessableToken::Create());
+  SurfaceId surface_id3(FrameSinkId(3, 3), local_frame_id3);
+
+  FakeSurfaceFactoryClient surface_factory_client1;
+  SurfaceFactory factory1(FrameSinkId(1, 1), &manager,
+                          &surface_factory_client1);
+  CompositorFrame frame;
+  factory1.SubmitCompositorFrame(local_frame_id1,
+                                 MakeCompositorFrame({surface_id2}),
+                                 SurfaceFactory::DrawCallback());
+  EXPECT_TRUE(manager.lock_manager()->observing_begin_frames());
+
+  // |factory1| is blocked on |surface_id2|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  FakeSurfaceFactoryClient surface_factory_client2;
+  SurfaceFactory factory2(FrameSinkId(2, 2), &manager,
+                          &surface_factory_client2);
+  factory2.SubmitCompositorFrame(local_frame_id2,
+                                 MakeCompositorFrame({surface_id3}),
+                                 SurfaceFactory::DrawCallback());
+  EXPECT_TRUE(manager.lock_manager()->observing_begin_frames());
+
+  // |factory2| is blocked on |surface_id3|.
+  EXPECT_FALSE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  BeginFrameArgs args =
+      CreateBeginFrameArgsForTesting(BEGINFRAME_FROM_HERE, 0, 1);
+  begin_frame_source->TestOnBeginFrame(args);
+
+  // |factory1| is still blocked on |surface_id2|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  // |factory2| is still blcoked on |surface_id3|.
+  EXPECT_FALSE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+
+  begin_frame_source->TestOnBeginFrame(args);
+  begin_frame_source->TestOnBeginFrame(args);
+  begin_frame_source->TestOnBeginFrame(args);
+
+  // |factory1| and |factory2| are no longer blocked.
+  EXPECT_TRUE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+  EXPECT_TRUE(factory2.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory2.current_surface_for_testing()->GetPendingFrame().has_value());
+  EXPECT_FALSE(manager.lock_manager()->observing_begin_frames());
+
+  factory1.EvictSurface();
+  factory2.EvictSurface();
+}
+
+// Frame activates once a new frame is submitted.
+TEST(SurfaceTest, DisplayCompositorLocking5) {
+  SurfaceManager manager;
+  std::unique_ptr<FakeExternalBeginFrameSource> begin_frame_source(
+      new FakeExternalBeginFrameSource(0.f, false));
+
+  manager.SetLockManager(base::MakeUnique<DisplayCompositorLockManager>(
+      &manager, begin_frame_source.get()));
+
+  LocalFrameId local_frame_id1(6, base::UnguessableToken::Create());
+  SurfaceId surface_id1(FrameSinkId(1, 1), local_frame_id1);
+
+  LocalFrameId local_frame_id2(7, base::UnguessableToken::Create());
+  SurfaceId surface_id2(FrameSinkId(2, 2), local_frame_id2);
+
+  FakeSurfaceFactoryClient surface_factory_client1;
+  SurfaceFactory factory1(FrameSinkId(1, 1), &manager,
+                          &surface_factory_client1);
+  factory1.SubmitCompositorFrame(local_frame_id1,
+                                 MakeCompositorFrame({surface_id2}),
+                                 SurfaceFactory::DrawCallback());
+
+  // |factory1|'s Frame is blocked on |surface_id2|.
+  EXPECT_FALSE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_TRUE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+  EXPECT_TRUE(manager.lock_manager()->observing_begin_frames());
+
+  // Another frame is submitted to |factory1| that has no dependencies.
+  factory1.SubmitCompositorFrame(local_frame_id1,
+                                 MakeCompositorFrame(std::vector<SurfaceId>()),
+                                 SurfaceFactory::DrawCallback());
+  EXPECT_TRUE(factory1.current_surface_for_testing()->HasFrame());
+  EXPECT_FALSE(
+      factory1.current_surface_for_testing()->GetPendingFrame().has_value());
+  EXPECT_FALSE(manager.lock_manager()->observing_begin_frames());
+
+  factory1.EvictSurface();
+}
+
 TEST(SurfaceTest, SurfaceLifetime) {
   SurfaceManager manager;
   FakeSurfaceFactoryClient surface_factory_client;
   SurfaceFactory factory(kArbitraryFrameSinkId, &manager,
                          &surface_factory_client);
 
   LocalFrameId local_frame_id(6, base::UnguessableToken::Create());
   SurfaceId surface_id(kArbitraryFrameSinkId, local_frame_id);
   factory.SubmitCompositorFrame(local_frame_id, CompositorFrame(),
                                 SurfaceFactory::DrawCallback());
   EXPECT_TRUE(manager.GetSurfaceForId(surface_id));
   factory.EvictSurface();
 
   EXPECT_EQ(NULL, manager.GetSurfaceForId(surface_id));
 }
 
 TEST(SurfaceTest, SurfaceIds) {
   for (size_t i = 0; i < 3; ++i) {
     SurfaceIdAllocator allocator;
     LocalFrameId id1 = allocator.GenerateId();
     LocalFrameId id2 = allocator.GenerateId();
     EXPECT_NE(id1, id2);
   }
 }
 
 }  // namespace
 }  // namespace cc