Index: media/cast/video_sender/external_video_encoder_unittest.cc |
diff --git a/media/cast/video_sender/external_video_encoder_unittest.cc b/media/cast/video_sender/external_video_encoder_unittest.cc |
index 1f2e4dd057ed6e2046e8ac9b6c5ffda7b9bf18dc..20c97562d128e6a4fd1944f091e11f907698039a 100644 |
--- a/media/cast/video_sender/external_video_encoder_unittest.cc |
+++ b/media/cast/video_sender/external_video_encoder_unittest.cc |
@@ -45,24 +45,26 @@ class TestVideoEncoderCallback |
public: |
TestVideoEncoderCallback() {} |
- void SetExpectedResult(bool expected_key_frame, |
- uint8 expected_frame_id, |
- uint8 expected_last_referenced_frame_id, |
+ void SetExpectedResult(uint32 expected_frame_id, |
+ uint32 expected_last_referenced_frame_id, |
const base::TimeTicks& expected_capture_time) { |
- expected_key_frame_ = expected_key_frame; |
expected_frame_id_ = expected_frame_id; |
expected_last_referenced_frame_id_ = expected_last_referenced_frame_id; |
expected_capture_time_ = expected_capture_time; |
} |
void DeliverEncodedVideoFrame( |
- scoped_ptr<transport::EncodedVideoFrame> encoded_frame, |
- const base::TimeTicks& capture_time) { |
- EXPECT_EQ(expected_key_frame_, encoded_frame->key_frame); |
+ scoped_ptr<transport::EncodedFrame> encoded_frame) { |
+ if (expected_frame_id_ == expected_last_referenced_frame_id_) { |
+ EXPECT_EQ(transport::EncodedFrame::KEY, encoded_frame->dependency); |
+ } else { |
+ EXPECT_EQ(transport::EncodedFrame::DEPENDENT, |
+ encoded_frame->dependency); |
+ } |
EXPECT_EQ(expected_frame_id_, encoded_frame->frame_id); |
EXPECT_EQ(expected_last_referenced_frame_id_, |
- encoded_frame->last_referenced_frame_id); |
- EXPECT_EQ(expected_capture_time_, capture_time); |
+ encoded_frame->referenced_frame_id); |
+ EXPECT_EQ(expected_capture_time_, encoded_frame->reference_time); |
} |
protected: |
@@ -72,8 +74,8 @@ class TestVideoEncoderCallback |
friend class base::RefCountedThreadSafe<TestVideoEncoderCallback>; |
bool expected_key_frame_; |
- uint8 expected_frame_id_; |
- uint8 expected_last_referenced_frame_id_; |
+ uint32 expected_frame_id_; |
+ uint32 expected_last_referenced_frame_id_; |
base::TimeTicks expected_capture_time_; |
DISALLOW_COPY_AND_ASSIGN(TestVideoEncoderCallback); |
@@ -145,15 +147,14 @@ TEST_F(ExternalVideoEncoderTest, EncodePattern30fpsRunningOutOfAck) { |
base::TimeTicks capture_time; |
capture_time += base::TimeDelta::FromMilliseconds(33); |
- test_video_encoder_callback_->SetExpectedResult(true, 0, 0, capture_time); |
+ test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); |
EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
video_frame_, capture_time, frame_encoded_callback)); |
task_runner_->RunTasks(); |
for (int i = 0; i < 6; ++i) { |
capture_time += base::TimeDelta::FromMilliseconds(33); |
- test_video_encoder_callback_->SetExpectedResult( |
- false, i + 1, i, capture_time); |
+ test_video_encoder_callback_->SetExpectedResult(i + 1, i, capture_time); |
EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
video_frame_, capture_time, frame_encoded_callback)); |
task_runner_->RunTasks(); |
@@ -172,7 +173,7 @@ TEST_F(ExternalVideoEncoderTest, SkipNextFrame) { |
base::TimeTicks capture_time; |
capture_time += base::TimeDelta::FromMilliseconds(33); |
- test_video_encoder_callback_->SetExpectedResult(true, 0, 0, capture_time); |
+ test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); |
EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
video_frame_, capture_time, frame_encoded_callback)); |
task_runner_->RunTasks(); |
@@ -188,8 +189,7 @@ TEST_F(ExternalVideoEncoderTest, SkipNextFrame) { |
video_encoder_->SkipNextFrame(false); |
for (int i = 0; i < 2; ++i) { |
capture_time += base::TimeDelta::FromMilliseconds(33); |
- test_video_encoder_callback_->SetExpectedResult( |
- false, i + 1, i, capture_time); |
+ test_video_encoder_callback_->SetExpectedResult(i + 1, i, capture_time); |
EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
video_frame_, capture_time, frame_encoded_callback)); |
task_runner_->RunTasks(); |
@@ -212,7 +212,7 @@ TEST_F(ExternalVideoEncoderTest, StreamHeader) { |
// Verify the first returned bitstream buffer is still a key frame. |
base::TimeTicks capture_time; |
capture_time += base::TimeDelta::FromMilliseconds(33); |
- test_video_encoder_callback_->SetExpectedResult(true, 0, 0, capture_time); |
+ test_video_encoder_callback_->SetExpectedResult(0, 0, capture_time); |
EXPECT_TRUE(video_encoder_->EncodeVideoFrame( |
video_frame_, capture_time, frame_encoded_callback)); |
task_runner_->RunTasks(); |