Index: remoting/protocol/video_frame_pump_unittest.cc |
diff --git a/remoting/host/video_frame_pump_unittest.cc b/remoting/protocol/video_frame_pump_unittest.cc |
similarity index 87% |
rename from remoting/host/video_frame_pump_unittest.cc |
rename to remoting/protocol/video_frame_pump_unittest.cc |
index d6490f41a43948c7e19861b40dadd81db13c9a11..119ad9ba084b127a367fc8e1091b12886755f759 100644 |
--- a/remoting/host/video_frame_pump_unittest.cc |
+++ b/remoting/protocol/video_frame_pump_unittest.cc |
@@ -2,7 +2,7 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
-#include "remoting/host/video_frame_pump.h" |
+#include "remoting/protocol/video_frame_pump.h" |
#include "base/bind.h" |
#include "base/message_loop/message_loop.h" |
@@ -12,19 +12,15 @@ |
#include "remoting/base/auto_thread_task_runner.h" |
#include "remoting/codec/video_encoder.h" |
#include "remoting/codec/video_encoder_verbatim.h" |
-#include "remoting/host/desktop_capturer_proxy.h" |
-#include "remoting/host/fake_desktop_capturer.h" |
-#include "remoting/host/host_mock_objects.h" |
#include "remoting/proto/control.pb.h" |
#include "remoting/proto/video.pb.h" |
+#include "remoting/protocol/fake_desktop_capturer.h" |
#include "remoting/protocol/protocol_mock_objects.h" |
#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
#include "third_party/webrtc/modules/desktop_capture/screen_capturer_mock_objects.h" |
-using ::remoting::protocol::MockVideoStub; |
- |
using ::testing::_; |
using ::testing::AtLeast; |
using ::testing::DoAll; |
@@ -33,6 +29,7 @@ using ::testing::InvokeWithoutArgs; |
using ::testing::Return; |
namespace remoting { |
+namespace protocol { |
namespace { |
@@ -52,6 +49,20 @@ scoped_ptr<webrtc::DesktopFrame> CreateUnchangedFrame( |
return make_scoped_ptr(new webrtc::BasicDesktopFrame(kSize)); |
} |
+class MockVideoEncoder : public VideoEncoder { |
+ public: |
+ MockVideoEncoder() {} |
+ ~MockVideoEncoder() {} |
+ |
+ MOCK_METHOD1(SetLosslessEncode, void(bool)); |
+ MOCK_METHOD1(SetLosslessColor, void(bool)); |
+ MOCK_METHOD1(EncodePtr, VideoPacket*(const webrtc::DesktopFrame&)); |
+ |
+ scoped_ptr<VideoPacket> Encode(const webrtc::DesktopFrame& frame) { |
+ return make_scoped_ptr(EncodePtr(frame)); |
+ } |
+}; |
+ |
} // namespace |
static const int kWidth = 640; |
@@ -162,15 +173,12 @@ TEST_F(VideoFramePumpTest, StartAndStop) { |
// When the first ProcessVideoPacket is received we stop the VideoFramePump. |
EXPECT_CALL(video_stub_, ProcessVideoPacketPtr(_, _)) |
- .WillOnce(DoAll( |
- FinishSend(), |
- InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit))) |
+ .WillOnce(DoAll(FinishSend(), |
+ InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit))) |
.RetiresOnSaturation(); |
// Start video frame capture. |
- pump_.reset(new VideoFramePump(encode_task_runner_, |
- make_scoped_ptr(new DesktopCapturerProxy( |
- capture_task_runner_, capturer.Pass())), |
+ pump_.reset(new VideoFramePump(encode_task_runner_, capturer.Pass(), |
encoder.Pass(), &video_stub_)); |
// Run MessageLoop until the first frame is received. |
@@ -197,9 +205,7 @@ TEST_F(VideoFramePumpTest, NullFrame) { |
.RetiresOnSaturation(); |
// Start video frame capture. |
- pump_.reset(new VideoFramePump(encode_task_runner_, |
- make_scoped_ptr(new DesktopCapturerProxy( |
- capture_task_runner_, capturer.Pass())), |
+ pump_.reset(new VideoFramePump(encode_task_runner_, capturer.Pass(), |
encoder.Pass(), &video_stub_)); |
// Run MessageLoop until the first frame is received.. |
@@ -227,13 +233,12 @@ TEST_F(VideoFramePumpTest, UnchangedFrame) { |
.RetiresOnSaturation(); |
// Start video frame capture. |
- pump_.reset(new VideoFramePump(encode_task_runner_, |
- make_scoped_ptr(new DesktopCapturerProxy( |
- capture_task_runner_, capturer.Pass())), |
+ pump_.reset(new VideoFramePump(encode_task_runner_, capturer.Pass(), |
encoder.Pass(), &video_stub_)); |
- // Run MessageLoop until the first frame is received.. |
+ // Run MessageLoop until the first frame is received. |
run_loop.Run(); |
} |
+} // namespace protocol |
} // namespace remoting |