content/blimp: Set up hooks for enabling LTHRemote in the renderer.

blimp/engine/renderer/scheduler_unittest.cc

+// Copyright 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 "blimp/engine/renderer/scheduler.h"
+
+#include "base/run_loop.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "blimp/engine/renderer/scheduler_client.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace blimp {
+namespace engine {
+namespace {
+
+class SchedulerForTesting : public Scheduler {
+ public:
+  explicit SchedulerForTesting(SchedulerClient* client)
+      // Use a zero time delta to let tests run main frames back-to-back.
+      : Scheduler(base::TimeDelta::FromSeconds(0),
+                  base::ThreadTaskRunnerHandle::Get(),
+                  client) {}
+  ~SchedulerForTesting() override = default;
+};
+
+class SchedulerTest : public testing::Test, public SchedulerClient {
+ public:
+  SchedulerTest() : scheduler_(this) {}
+  ~SchedulerTest() override {}
+
+  // SchedulerClient implementation.
+  void StartMainFrameUpdate() override {
+    num_main_frames_++;
+    if (send_client_update_during_main_frame_) {
+      scheduler_.DidSendFrameUpdateToClient();
+      send_client_update_during_main_frame_ = false;
+    }
+  }
+
+ protected:
+  base::MessageLoop loop_;
+  SchedulerForTesting scheduler_;
+  int num_main_frames_ = 0;
+
+  bool send_client_update_during_main_frame_ = false;
+};
+
+TEST_F(SchedulerTest, FirstMainFrameRequest) {
+  // The request for the first main frame should be responded to right away.
+  scheduler_.SetNeedsMainFrame();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, num_main_frames_);
+}
+
+TEST_F(SchedulerTest, MultipleMainFrameRequestsResultInSingleFrame) {
+  // Multiple main frame requests result in a single callback to the client.
+  scheduler_.SetNeedsMainFrame();
+  scheduler_.SetNeedsMainFrame();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, num_main_frames_);
+}
+
+TEST_F(SchedulerTest, MainFrameBlockedOnAckFromClient) {
+  // Request a main frame, start it and send an update to the client. The second
+  // frame should remain blocked till an ack is received.
+  scheduler_.SetNeedsMainFrame();
+  send_client_update_during_main_frame_ = true;
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, num_main_frames_);
+
+  // Request a frame and ensure that the client is not asked to start it.
+  scheduler_.SetNeedsMainFrame();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, num_main_frames_);
+
+  // Now we have an ack from the client. This should start another frame.
+  scheduler_.DidReceiveFrameUpdateAck();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(2, num_main_frames_);
+}
+
+}  // namespace
+}  // namespace engine
+}  // namespace blimp