content/renderer/media/webrtc/webrtc_video_capturer_adapter_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <algorithm>
   6
   7 #include "content/renderer/media/webrtc/webrtc_video_capturer_adapter.h"
   8 #include "media/base/video_frame.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace content {
  12
  13 class WebRtcVideoCapturerAdapterTest
  14     : public sigslot::has_slots<>,
  15       public ::testing::Test {
  16  public:
  17   WebRtcVideoCapturerAdapterTest()
  18       : adapter_(false),
  19         output_frame_width_(0),
  20         output_frame_height_(0) {
  21     adapter_.SignalFrameCaptured.connect(
  22         this, &WebRtcVideoCapturerAdapterTest::OnFrameCaptured);
  23   }
  24   ~WebRtcVideoCapturerAdapterTest() override {}
  25
  26   void TestSourceCropFrame(int capture_width,
  27                            int capture_height,
  28                            int cropped_width,
  29                            int cropped_height,
  30                            int natural_width,
  31                            int natural_height) {
  32     const int horiz_crop = ((capture_width - cropped_width) / 2);
  33     const int vert_crop = ((capture_height - cropped_height) / 2);
  34
  35     gfx::Size coded_size(capture_width, capture_height);
  36     gfx::Size natural_size(natural_width, natural_height);
  37     gfx::Rect view_rect(horiz_crop, vert_crop, cropped_width, cropped_height);
  38     scoped_refptr<media::VideoFrame> frame =
  39         media::VideoFrame::CreateFrame(media::VideoFrame::I420,
  40                                        coded_size, view_rect, natural_size,
  41                                        base::TimeDelta());
  42     adapter_.OnFrameCaptured(frame);
  43     EXPECT_EQ(natural_width, output_frame_width_);
  44     EXPECT_EQ(natural_height, output_frame_height_);
  45   }
  46  protected:
  47   void OnFrameCaptured(cricket::VideoCapturer* capturer,
  48                        const cricket::CapturedFrame* frame) {
  49     output_frame_width_ = frame->width;
  50     output_frame_height_ = frame->height;
  51   }
  52
  53  private:
  54   WebRtcVideoCapturerAdapter adapter_;
  55   int output_frame_width_;
  56   int output_frame_height_;
  57 };
  58
  59 TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo640360) {
  60   TestSourceCropFrame(640, 480, 640, 360, 640, 360);
  61 }
  62
  63 TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo320320) {
  64   TestSourceCropFrame(640, 480, 480, 480, 320, 320);
  65 }
  66
  67 TEST_F(WebRtcVideoCapturerAdapterTest, Scale720To640360) {
  68   TestSourceCropFrame(1280, 720, 1280, 720, 640, 360);
  69 }
  70
  71 }  // namespace content