| Index: content/renderer/media/video_source_handler_unittest.cc
|
| ===================================================================
|
| --- content/renderer/media/video_source_handler_unittest.cc (revision 252031)
|
| +++ content/renderer/media/video_source_handler_unittest.cc (working copy)
|
| @@ -14,6 +14,7 @@
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "third_party/WebKit/public/platform/WebMediaStreamTrack.h"
|
| #include "third_party/WebKit/public/platform/WebString.h"
|
| +#include "third_party/libjingle/source/talk/media/base/videocapturer.h"
|
| #include "third_party/libjingle/source/talk/media/base/videorenderer.h"
|
| #include "third_party/libjingle/source/talk/media/webrtc/webrtcvideoframe.h"
|
|
|
| @@ -59,28 +60,59 @@
|
| // Unknow url will return false.
|
| EXPECT_FALSE(handler_->Open(kUnknownStreamUrl, &reader));
|
| EXPECT_TRUE(handler_->Open(kTestStreamUrl, &reader));
|
| - cricket::WebRtcVideoFrame test_frame;
|
| - int width = 640;
|
| - int height = 360;
|
| +
|
| + size_t width = 640;
|
| + size_t height = 360;
|
| int64 et = 123456;
|
| int64 ts = 789012;
|
| - test_frame.InitToBlack(width, height, 1, 1, et, ts);
|
| + size_t size = VideoFrame::SizeOf(width, height);
|
| + std::vector<uint8_t> test_buffer(size);
|
| + for (size_t i = 0; i < size; ++i) {
|
| + test_buffer[i] = (i & 0xFF);
|
| + }
|
| +
|
| + // A new frame is captured.
|
| + cricket::CapturedFrame captured_frame;
|
| + captured_frame.width = width;
|
| + captured_frame.height = height;
|
| + captured_frame.fourcc = cricket::FOURCC_I420;
|
| + // cricket::CapturedFrame time is in nanoseconds.
|
| + captured_frame.elapsed_time = et;
|
| + captured_frame.time_stamp = ts;
|
| + captured_frame.data = &test_buffer[0];
|
| + captured_frame.data_size = size;
|
| + captured_frame.pixel_height = 1;
|
| + captured_frame.pixel_width = 1;
|
| +
|
| + // The frame is delivered to VideoSourceHandler as cricket::VideoFrame.
|
| + cricket::WebRtcVideoFrame i420_frame;
|
| + EXPECT_TRUE(i420_frame.Alias(&captured_frame, width, height));
|
| cricket::VideoRenderer* receiver = handler_->GetReceiver(&reader);
|
| ASSERT(receiver != NULL);
|
| - receiver->RenderFrame(&test_frame);
|
| + receiver->RenderFrame(&i420_frame);
|
|
|
| + // Compare |frame| to |captured_frame|.
|
| const VideoFrame* frame = reader.last_frame();
|
| ASSERT_TRUE(frame != NULL);
|
| + EXPECT_EQ(width, frame->GetWidth());
|
| + EXPECT_EQ(height, frame->GetHeight());
|
| + EXPECT_EQ(et, frame->GetElapsedTime());
|
| + EXPECT_EQ(ts, frame->GetTimeStamp());
|
| + std::vector<uint8_t> tmp_buffer1(size);
|
| + EXPECT_EQ(size, frame->CopyToBuffer(&tmp_buffer1[0], size));
|
| + EXPECT_TRUE(std::equal(test_buffer.begin(), test_buffer.end(),
|
| + tmp_buffer1.begin()));
|
|
|
| - // Compare |frame| to |test_frame|.
|
| - EXPECT_EQ(test_frame.GetWidth(), frame->GetWidth());
|
| - EXPECT_EQ(test_frame.GetHeight(), frame->GetHeight());
|
| - EXPECT_EQ(test_frame.GetElapsedTime(), frame->GetElapsedTime());
|
| - EXPECT_EQ(test_frame.GetTimeStamp(), frame->GetTimeStamp());
|
| - EXPECT_EQ(test_frame.GetYPlane(), frame->GetYPlane());
|
| - EXPECT_EQ(test_frame.GetUPlane(), frame->GetUPlane());
|
| - EXPECT_EQ(test_frame.GetVPlane(), frame->GetVPlane());
|
| + // Invalid the original frame
|
| + memset(&test_buffer[0], 0, size);
|
| + test_buffer.clear();
|
|
|
| + // We should still have the same |frame|.
|
| + std::vector<uint8_t> tmp_buffer2(size);
|
| + EXPECT_EQ(size, frame->CopyToBuffer(&tmp_buffer2[0], size));
|
| + EXPECT_TRUE(std::equal(tmp_buffer1.begin(), tmp_buffer1.end(),
|
| + tmp_buffer2.begin()));
|
| +
|
| EXPECT_FALSE(handler_->Close(NULL));
|
| EXPECT_TRUE(handler_->Close(&reader));
|
| EXPECT_TRUE(handler_->GetReceiver(&reader) == NULL);
|
|
|