[Mojo Video Capture] Replace RESOURCE_UTILIZATION with interface ReceiverLoadObserver

content/browser/renderer_host/media/video_capture_device_client_unittest.cc

 // Copyright 2015 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 "media/capture/video/video_capture_device_client.h"
 
 #include <stddef.h>
 
 #include <memory>
 
@@ skipping 19 matching lines @@
 namespace content {
 
 namespace {
 
 class MockVideoCaptureController : public VideoCaptureController {
  public:
   explicit MockVideoCaptureController(int max_buffers)
       : VideoCaptureController(max_buffers) {}
   ~MockVideoCaptureController() override {}
 
-  MOCK_METHOD1(MockOnIncomingCapturedVideoFrame, void(const gfx::Size&));
+  MOCK_METHOD2(MockOnIncomingCapturedVideoFrame,
+               void(const gfx::Size& coded_size, int buffer_id));
   MOCK_METHOD0(OnError, void());
   MOCK_METHOD1(OnLog, void(const std::string& message));
   MOCK_METHOD1(OnBufferDestroyed, void(int buffer_id_to_drop));
 
   void OnIncomingCapturedVideoFrame(
       std::unique_ptr<media::VideoCaptureDevice::Client::Buffer> buffer,
       scoped_refptr<media::VideoFrame> frame) override {
-    MockOnIncomingCapturedVideoFrame(frame->coded_size());
+    MockOnIncomingCapturedVideoFrame(frame->coded_size(), buffer->id());
   }
 };
 
+class MockConsumerLoadObserver : public media::ConsumerLoadObserver {
+ public:
+  MOCK_METHOD2(OnConsumerReportingUtilization,
+               void(int frame_id, double utilization));
+};
+
 // Note that this test does not exercise the class VideoCaptureDeviceClient
 // in isolation. The "unit under test" is an instance of
 // VideoCaptureDeviceClient with some context that is specific to
 // renderer_host/media, and therefore this test must live here and not in
 // media/capture/video.
 class VideoCaptureDeviceClientTest : public ::testing::Test {
  public:
   VideoCaptureDeviceClientTest()
       : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP),
         controller_(new MockVideoCaptureController(1)),
         device_client_(controller_->NewDeviceClient()) {}
   ~VideoCaptureDeviceClientTest() override {}
 
   void TearDown() override { base::RunLoop().RunUntilIdle(); }
 
  protected:
   const content::TestBrowserThreadBundle thread_bundle_;
   const std::unique_ptr<MockVideoCaptureController> controller_;
-  const std::unique_ptr<media::VideoCaptureDevice::Client> device_client_;
+  const std::unique_ptr<media::VideoCaptureDeviceClient> device_client_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(VideoCaptureDeviceClientTest);
 };
 
 }  // namespace
 
 // A small test for reference and to verify VideoCaptureDeviceClient is
 // minimally functional.
 TEST_F(VideoCaptureDeviceClientTest, Minimal) {
   const size_t kScratchpadSizeInBytes = 400;
   unsigned char data[kScratchpadSizeInBytes] = {};
   const media::VideoCaptureFormat kFrameFormat(
       gfx::Size(10, 10), 30.0f /*frame_rate*/,
       media::PIXEL_FORMAT_I420,
       media::PIXEL_STORAGE_CPU);
   DCHECK(device_client_.get());
   EXPECT_CALL(*controller_, OnLog(_)).Times(1);
-  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_)).Times(1);
+  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _)).Times(1);
   device_client_->OnIncomingCapturedData(data, kScratchpadSizeInBytes,
                                          kFrameFormat, 0 /*clockwise rotation*/,
                                          base::TimeTicks(), base::TimeDelta());
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(controller_.get());
 }
 
 // Tests that we don't try to pass on frames with an invalid frame format.
 TEST_F(VideoCaptureDeviceClientTest, FailsSilentlyGivenInvalidFrameFormat) {
   const size_t kScratchpadSizeInBytes = 400;
   unsigned char data[kScratchpadSizeInBytes] = {};
   // kFrameFormat is invalid in a number of ways.
   const media::VideoCaptureFormat kFrameFormat(
       gfx::Size(media::limits::kMaxDimension + 1, media::limits::kMaxDimension),
       media::limits::kMaxFramesPerSecond + 1,
       media::VideoPixelFormat::PIXEL_FORMAT_I420,
       media::VideoPixelStorage::PIXEL_STORAGE_CPU);
   DCHECK(device_client_.get());
   // Expect the the call to fail silently inside the VideoCaptureDeviceClient.
   EXPECT_CALL(*controller_, OnLog(_)).Times(1);
-  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_)).Times(0);
+  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _)).Times(0);
   device_client_->OnIncomingCapturedData(data, kScratchpadSizeInBytes,
                                          kFrameFormat, 0 /*clockwise rotation*/,
                                          base::TimeTicks(), base::TimeDelta());
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(controller_.get());
 }
 
 // Tests that we fail silently if no available buffers to use.
 TEST_F(VideoCaptureDeviceClientTest, DropsFrameIfNoBuffer) {
   const size_t kScratchpadSizeInBytes = 400;
   unsigned char data[kScratchpadSizeInBytes] = {};
   const media::VideoCaptureFormat kFrameFormat(
       gfx::Size(10, 10), 30.0f /*frame_rate*/,
       media::PIXEL_FORMAT_I420,
       media::PIXEL_STORAGE_CPU);
   // We expect the second frame to be silently dropped, so these should
   // only be called once despite the two frames.
   EXPECT_CALL(*controller_, OnLog(_)).Times(1);
-  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_)).Times(1);
+  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _)).Times(1);
   // Pass two frames. The second will be dropped.
   device_client_->OnIncomingCapturedData(data, kScratchpadSizeInBytes,
                                          kFrameFormat, 0 /*clockwise rotation*/,
                                          base::TimeTicks(), base::TimeDelta());
   device_client_->OnIncomingCapturedData(data, kScratchpadSizeInBytes,
                                          kFrameFormat, 0 /*clockwise rotation*/,
                                          base::TimeTicks(), base::TimeDelta());
   base::RunLoop().RunUntilIdle();
   Mock::VerifyAndClearExpectations(controller_.get());
 }
@@ skipping 28 matching lines @@
 #endif
     media::PIXEL_FORMAT_RGB32,
     media::PIXEL_FORMAT_ARGB,
     media::PIXEL_FORMAT_Y16,
   };
 
   for (media::VideoPixelFormat format : kSupportedFormats) {
     params.requested_format.pixel_format = format;
 
     EXPECT_CALL(*controller_, OnLog(_)).Times(1);
-    EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_)).Times(1);
+    EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _)).Times(1);
     device_client_->OnIncomingCapturedData(
         data, params.requested_format.ImageAllocationSize(),
         params.requested_format, 0 /* clockwise_rotation */, base::TimeTicks(),
         base::TimeDelta());
     base::RunLoop().RunUntilIdle();
     Mock::VerifyAndClearExpectations(controller_.get());
   }
 }
 
 // Test that we receive the expected resolution for a given captured frame
@@ skipping 23 matching lines @@
 
   media::VideoCaptureParams params;
   for (const auto& size_and_rotation : kSizeAndRotations) {
     ASSERT_GE(kScratchpadSizeInBytes,
               size_and_rotation.input_resolution.GetArea() * 4u)
         << "Scratchpad is too small to hold the largest pixel format (ARGB).";
     params.requested_format =
         media::VideoCaptureFormat(size_and_rotation.input_resolution, 30.0f,
                                   media::PIXEL_FORMAT_ARGB);
     gfx::Size coded_size;
-    EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_))
+    EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _))
         .Times(1)
         .WillOnce(SaveArg<0>(&coded_size));
     device_client_->OnIncomingCapturedData(
         data, params.requested_format.ImageAllocationSize(),
         params.requested_format, size_and_rotation.rotation, base::TimeTicks(),
         base::TimeDelta());
     base::RunLoop().RunUntilIdle();
 
     EXPECT_EQ(coded_size.width(), size_and_rotation.output_resolution.width());
     EXPECT_EQ(coded_size.height(),
               size_and_rotation.output_resolution.height());
 
     Mock::VerifyAndClearExpectations(controller_.get());
   }
 }
 
+// Tests that consumer load reports arriving at VideoCaptureDeviceClient are
+// correctly forwarded to its ConsumerLoadObserver with the buffer_id being
+// translated to the frame_id.
+TEST_F(VideoCaptureDeviceClientTest, ReportConsumerLoad) {
+  const size_t kScratchpadSizeInBytes = 400;
+  unsigned char data[kScratchpadSizeInBytes] = {};
+  const media::VideoCaptureFormat kFrameFormat(
+      gfx::Size(10, 10), 30.0f /*frame_rate*/,
+      media::PIXEL_FORMAT_I420,
+      media::PIXEL_STORAGE_CPU);
+  const int arbitrary_frame_id = 123;
+  const double arbitrary_utilization = 0.789;
+  DCHECK(device_client_.get());
+  MockConsumerLoadObserver load_observer;
+  device_client_->SetConsumerLoadObserver(&load_observer);
+
+  int buffer_id;
+  EXPECT_CALL(*controller_, MockOnIncomingCapturedVideoFrame(_, _))
+      .Times(1)
+      .WillOnce(SaveArg<1>(&buffer_id));
+  EXPECT_CALL(load_observer, OnConsumerReportingUtilization(
+                                 arbitrary_frame_id, arbitrary_utilization))
+      .Times(1);
+
+  device_client_->OnIncomingCapturedData(data, kScratchpadSizeInBytes,
+                                         kFrameFormat, 0 /*clockwise rotation*/,
+                                         base::TimeTicks(), base::TimeDelta(),
+                                         arbitrary_frame_id);
+  base::RunLoop().RunUntilIdle();
+
+  device_client_->OnReceiverReportingUtilization(buffer_id,
+                                                 arbitrary_utilization);
+
+  base::RunLoop().RunUntilIdle();
+  Mock::VerifyAndClearExpectations(controller_.get());
+}
+
 }  // namespace content