Add FrameGenerator Unit Tests

services/ui/ws/frame_generator_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 "services/ui/ws/frame_generator.h"
+
+#include "base/macros.h"
+#include "cc/output/compositor_frame_sink.h"
+#include "cc/scheduler/begin_frame_source.h"
+#include "cc/test/begin_frame_args_test.cc"
+#include "cc/test/fake_external_begin_frame_source.h"
+#include "services/ui/ws/frame_generator_delegate.h"
+#include "services/ui/ws/server_window.h"
+#include "services/ui/ws/server_window_delegate.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace ui {
+namespace ws {
+namespace test {
+
+class TestFrameGeneratorDelegate : public FrameGeneratorDelegate {
+ public:
+  TestFrameGeneratorDelegate() {}
+  ~TestFrameGeneratorDelegate() override {}
+
+  // FrameGeneratorDelegate implementation:
+  bool IsInHighContrastMode() override { return false; }
+};
+
+class TestServerWindowDelegate : public ServerWindowDelegate {
+ public:
+  TestServerWindowDelegate() {}
+  ~TestServerWindowDelegate() override {}
+
+  // ServerWindowDelegate implementation:
+  cc::mojom::DisplayCompositor* GetDisplayCompositor() override {
+    return nullptr;
+  }
+
+  ServerWindow* GetRootWindow(const ServerWindow* window) override {
+    return nullptr;
+  }
+};
+
+// FakeCompositorFrameSink observes a FakeExternalBeginFrameSource and receives
+// CompositorFrame from a FrameGenerator.
+class FakeCompositorFrameSink : public cc::CompositorFrameSink,
+                                public cc::BeginFrameObserver,
+                                public cc::ExternalBeginFrameSourceClient {
+ public:
+  FakeCompositorFrameSink()
+      : cc::CompositorFrameSink(nullptr, nullptr, nullptr, nullptr) {}
+
+  // cc::CompositorFrameSink implementation:
+  bool BindToClient(cc::CompositorFrameSinkClient* client) override {
+    if (!cc::CompositorFrameSink::BindToClient(client))
+      return false;
+
+    begin_frame_source_ = base::MakeUnique<cc::ExternalBeginFrameSource>(this);
+    client_->SetBeginFrameSource(begin_frame_source_.get());
+    return true;
+  }
+
+  void DetachFromClient() override {
+    cc::CompositorFrameSink::DetachFromClient();
+  }
+
+  void SubmitCompositorFrame(cc::CompositorFrame frame) override {
+    ++number_frames_received_;
+    last_frame_ = std::move(frame);
+  }
+
+  // cc::BeginFrameObserver implementation.
+  void OnBeginFrame(const cc::BeginFrameArgs& args) override {
+    begin_frame_source_->OnBeginFrame(args);
+    last_begin_frame_args_ = args;
+  }
+
+  const cc::BeginFrameArgs& LastUsedBeginFrameArgs() const override {
+    return last_begin_frame_args_;
+  }
+
+  void OnBeginFrameSourcePausedChanged(bool paused) override {}
+
+  // cc::ExternalBeginFrameSourceClient implementation:
+  void OnNeedsBeginFrames(bool needs_begin_frames) override {}
+
+  void OnDidFinishFrame(const cc::BeginFrameAck& ack) override {}
+
+  const cc::CompositorFrame& last_frame() { return last_frame_; }
+
+  int number_frames_received() { return number_frames_received_; }
+
+ private:
+  int number_frames_received_ = 0;
+  std::unique_ptr<cc::ExternalBeginFrameSource> begin_frame_source_;
+  cc::BeginFrameArgs last_begin_frame_args_;
+  cc::CompositorFrame last_frame_;
+};
+
+class FrameGeneratorTest : public testing::Test {
+ public:
+  FrameGeneratorTest() {}
+  ~FrameGeneratorTest() override {}
+
+  // testing::Test overrides:
+  void SetUp() override {
+    testing::Test::SetUp();
+
+    std::unique_ptr<FakeCompositorFrameSink> compositor_frame_sink =
+        base::MakeUnique<FakeCompositorFrameSink>();
+    compositor_frame_sink_ = compositor_frame_sink.get();
+
+    constexpr float kRefreshRate = 0.f;
+    constexpr bool kTickAutomatically = false;
+    begin_frame_source_ = base::MakeUnique<cc::FakeExternalBeginFrameSource>(
+        kRefreshRate, kTickAutomatically);
+    begin_frame_source_->AddObserver(compositor_frame_sink_);
+    delegate_ = base::MakeUnique<TestFrameGeneratorDelegate>();
+    server_window_delegate_ = base::MakeUnique<TestServerWindowDelegate>();
+    root_window_ = base::MakeUnique<ServerWindow>(server_window_delegate_.get(),
+                                                  WindowId());
+    root_window_->SetVisible(true);
+    frame_generator_ = base::MakeUnique<FrameGenerator>(
+        delegate_.get(), root_window_.get(), std::move(compositor_frame_sink));
+  };
+
+  int NumberOfFramesRecieved() {
+    return compositor_frame_sink_->number_frames_received();
+  }
+
+  void IssueBeginFrame(const cc::BeginFrameArgs& args) {
+    begin_frame_source_->TestOnBeginFrame(args);
+  }
+
+  FrameGenerator* frame_generator() { return frame_generator_.get(); }
+
+  const cc::SurfaceId& id_last_frame() {

[Fady Samuel, 2017/03/15 20:44:58]

Make this a bit more generic.

ReferencedSurfaces()? Also, what if we haven't received a frame?

perhaps referenced_surfaces can return an const std::vector<...>* instead of
const ref? return nullptr if we don't have a CompositorFrame.

[Alex Z., 2017/03/17 13:50:43]

Done.

+    return compositor_frame_sink_->last_frame()
+        .metadata.referenced_surfaces.front();
+  }
+
+ private:
+  FakeCompositorFrameSink* compositor_frame_sink_ = nullptr;
+  std::unique_ptr<cc::FakeExternalBeginFrameSource> begin_frame_source_;
+  std::unique_ptr<TestFrameGeneratorDelegate> delegate_;
+  std::unique_ptr<TestServerWindowDelegate> server_window_delegate_;
+  std::unique_ptr<ServerWindow> root_window_;
+  std::unique_ptr<FrameGenerator> frame_generator_;
+
+  DISALLOW_COPY_AND_ASSIGN(FrameGeneratorTest);
+};
+
+TEST_F(FrameGeneratorTest, OnSurfaceCreated) {
+  DCHECK_EQ(0, NumberOfFramesRecieved());
+
+  // FrameGenerator does not request BeginFrames upon creation.
+  cc::BeginFrameArgs args =
+      cc::CreateBeginFrameArgsForTesting(BEGINFRAME_FROM_HERE, 0, 1);
+  IssueBeginFrame(args);
+  EXPECT_EQ(0, NumberOfFramesRecieved());
+
+  const cc::SurfaceId kArbitrarySurfaceId(
+      cc::FrameSinkId(1, 1),
+      cc::LocalSurfaceId(1, base::UnguessableToken::Create()));
+  const cc::SurfaceInfo kArbitrarySurfaceInfo(kArbitrarySurfaceId, 1.0f,
+                                              gfx::Size(100, 100));
+  frame_generator()->OnSurfaceCreated(kArbitrarySurfaceInfo);
+  IssueBeginFrame(args);
+  EXPECT_EQ(1, NumberOfFramesRecieved());
+
+  // Verify that the CompositorFrame refers to the window manager's surface via
+  // referenced_surfaces.
+  cc::SurfaceId id = id_last_frame();
+  EXPECT_EQ(kArbitrarySurfaceId, id);

[Fady Samuel, 2017/03/15 20:44:58]

Make this a bit more generic?

[Alex Z., 2017/03/17 13:50:43]

Done.

+}
+
+// TODO(staraz): Add OnSurfaceCreated test and SetDeviceScaleFactor test.

[Fady Samuel, 2017/03/15 20:44:58]

A test for SetDeviceScaleFactor should be fairly simple too:

SetDeviceScaleFactor(2);
// Verify we don't yet have a new CompositorFrame.
IssueBeginFrame(..);
// Verify we have a new CompositorFrame.
SetDeviceScaleFactor(2);
// Verify we don't have a new CompositorFrame.
IssueBeginFrame(..);
// Verify we have a new CompositorFrame.

[Alex Z., 2017/03/17 13:50:43]

Done.

+
+}  // namespace test
+}  // namespace ws
+}  // namespace ui