mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00

Move FrameGeneratorCapturerConfig and Create family methods from frame_generator_capturer.h to the create_frame_generator_capturer.h Bug: b/272350185 Change-Id: I95674f5238ac0d0a5e395840bbab7f205b160c37 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/301220 Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Commit-Queue: Artem Titov <titovartem@webrtc.org> Cr-Commit-Position: refs/heads/main@{#39850}
93 lines
3.4 KiB
C++
93 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2019 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "test/frame_generator_capturer.h"
|
|
|
|
#include "test/create_frame_generator_capturer.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
#include "test/time_controller/simulated_time_controller.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
namespace {
|
|
using ::testing::Eq;
|
|
using ::testing::Property;
|
|
|
|
constexpr int kWidth = 640;
|
|
constexpr int kHeight = 360;
|
|
|
|
class MockVideoSinkInterfaceVideoFrame
|
|
: public rtc::VideoSinkInterface<VideoFrame> {
|
|
public:
|
|
MOCK_METHOD(void, OnFrame, (const VideoFrame& frame), (override));
|
|
MOCK_METHOD(void, OnDiscardedFrame, (), (override));
|
|
};
|
|
} // namespace
|
|
|
|
TEST(FrameGeneratorCapturerTest, CreateFromConfig) {
|
|
GlobalSimulatedTimeController time(Timestamp::Seconds(1000));
|
|
FrameGeneratorCapturerConfig config;
|
|
config.squares_video->width = 300;
|
|
config.squares_video->height = 200;
|
|
config.squares_video->framerate = 20;
|
|
auto capturer = CreateFrameGeneratorCapturer(
|
|
time.GetClock(), *time.GetTaskQueueFactory(), config);
|
|
testing::StrictMock<MockVideoSinkInterfaceVideoFrame> mock_sink;
|
|
capturer->AddOrUpdateSink(&mock_sink, rtc::VideoSinkWants());
|
|
capturer->Start();
|
|
EXPECT_CALL(mock_sink, OnFrame(Property(&VideoFrame::width, Eq(300))))
|
|
.Times(21);
|
|
time.AdvanceTime(TimeDelta::Seconds(1));
|
|
}
|
|
|
|
TEST(FrameGeneratorCapturerTest, OnOutputFormatRequest) {
|
|
GlobalSimulatedTimeController time(Timestamp::Seconds(1000));
|
|
FrameGeneratorCapturerConfig config;
|
|
config.squares_video->width = kWidth;
|
|
config.squares_video->height = kHeight;
|
|
config.squares_video->framerate = 20;
|
|
auto capturer = CreateFrameGeneratorCapturer(
|
|
time.GetClock(), *time.GetTaskQueueFactory(), config);
|
|
testing::StrictMock<MockVideoSinkInterfaceVideoFrame> mock_sink;
|
|
capturer->AddOrUpdateSink(&mock_sink, rtc::VideoSinkWants());
|
|
capturer->OnOutputFormatRequest(kWidth / 2, kHeight / 2, /*max_fps=*/10);
|
|
capturer->Start();
|
|
EXPECT_CALL(mock_sink, OnFrame(Property(&VideoFrame::width, Eq(kWidth / 2))))
|
|
.Times(11);
|
|
time.AdvanceTime(TimeDelta::Seconds(1));
|
|
}
|
|
|
|
TEST(FrameGeneratorCapturerTest, ChangeResolution) {
|
|
GlobalSimulatedTimeController time(Timestamp::Seconds(1000));
|
|
FrameGeneratorCapturerConfig config;
|
|
config.squares_video->width = kWidth;
|
|
config.squares_video->height = kHeight;
|
|
config.squares_video->framerate = 20;
|
|
auto capturer = CreateFrameGeneratorCapturer(
|
|
time.GetClock(), *time.GetTaskQueueFactory(), config);
|
|
EXPECT_TRUE(capturer->GetResolution());
|
|
EXPECT_EQ(kWidth, capturer->GetResolution()->width);
|
|
EXPECT_EQ(kHeight, capturer->GetResolution()->height);
|
|
capturer->Start();
|
|
time.AdvanceTime(TimeDelta::Seconds(1));
|
|
ASSERT_TRUE(capturer->GetResolution());
|
|
EXPECT_EQ(kWidth, capturer->GetResolution()->width);
|
|
EXPECT_EQ(kHeight, capturer->GetResolution()->height);
|
|
|
|
capturer->ChangeResolution(kWidth / 2, kHeight / 2);
|
|
time.AdvanceTime(TimeDelta::Seconds(1));
|
|
ASSERT_TRUE(capturer->GetResolution());
|
|
EXPECT_EQ(kWidth / 2, capturer->GetResolution()->width);
|
|
EXPECT_EQ(kHeight / 2, capturer->GetResolution()->height);
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|