| Index: remoting/host/video_scheduler_unittest.cc
|
| diff --git a/remoting/host/video_scheduler_unittest.cc b/remoting/host/video_scheduler_unittest.cc
|
| index 2dd7afbf195f2561d50a32c88c0230e7b7527954..25567d27baa35a34df4a6c2ddc99a2ea0766def8 100644
|
| --- a/remoting/host/video_scheduler_unittest.cc
|
| +++ b/remoting/host/video_scheduler_unittest.cc
|
| @@ -37,8 +37,7 @@ namespace {
|
|
|
| ACTION(FinishEncode) {
|
| scoped_ptr<VideoPacket> packet(new VideoPacket());
|
| - packet->set_flags(VideoPacket::LAST_PACKET | VideoPacket::LAST_PARTITION);
|
| - arg1.Run(packet.Pass());
|
| + return packet.release();
|
| }
|
|
|
| ACTION(FinishSend) {
|
| @@ -55,9 +54,11 @@ class MockVideoEncoder : public VideoEncoder {
|
| MockVideoEncoder();
|
| virtual ~MockVideoEncoder();
|
|
|
| - MOCK_METHOD2(Encode, void(
|
| - const webrtc::DesktopFrame* frame,
|
| - const DataAvailableCallback& data_available_callback));
|
| + scoped_ptr<VideoPacket> Encode(
|
| + const webrtc::DesktopFrame& frame) {
|
| + return scoped_ptr<VideoPacket>(EncodePtr(frame));
|
| + }
|
| + MOCK_METHOD1(EncodePtr, VideoPacket*(const webrtc::DesktopFrame& frame));
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(MockVideoEncoder);
|
| @@ -158,7 +159,6 @@ TEST_F(VideoSchedulerTest, StartAndStop) {
|
|
|
| frame_.reset(new webrtc::BasicDesktopFrame(
|
| webrtc::DesktopSize(kWidth, kHeight)));
|
| - webrtc::DesktopFrame* frame_ptr = frame_.get();
|
|
|
| // First the capturer is called.
|
| Expectation capturer_capture = EXPECT_CALL(*capturer, Capture(_))
|
| @@ -166,7 +166,7 @@ TEST_F(VideoSchedulerTest, StartAndStop) {
|
| .WillRepeatedly(Invoke(this, &VideoSchedulerTest::OnCaptureFrame));
|
|
|
| // Expect the encoder be called.
|
| - EXPECT_CALL(*encoder_, Encode(frame_ptr, _))
|
| + EXPECT_CALL(*encoder_, EncodePtr(_))
|
| .WillRepeatedly(FinishEncode());
|
|
|
| // By default delete the arguments when ProcessVideoPacket is received.
|
|
|