| Index: media/video/capture/screen/screen_capture_device_unittest.cc
|
| diff --git a/media/video/capture/screen/screen_capture_device_unittest.cc b/media/video/capture/screen/screen_capture_device_unittest.cc
|
| index 6b7446554d0d1ab95726199ac57a20880b88e1f0..0247bb96bff8303fa3ae836e0141b13027f3d232 100644
|
| --- a/media/video/capture/screen/screen_capture_device_unittest.cc
|
| +++ b/media/video/capture/screen/screen_capture_device_unittest.cc
|
| @@ -10,9 +10,10 @@
|
| #include "base/test/test_timeouts.h"
|
| #include "base/threading/sequenced_worker_pool.h"
|
| #include "base/time.h"
|
| -#include "media/video/capture/screen/screen_capture_data.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/desktop_geometry.h"
|
|
|
| using ::testing::_;
|
| using ::testing::DoAll;
|
| @@ -51,33 +52,33 @@ class MockFrameObserver : public VideoCaptureDevice::EventHandler {
|
| class FakeScreenCapturer : public ScreenCapturer {
|
| public:
|
| FakeScreenCapturer()
|
| - : delegate_(NULL),
|
| + : callback_(NULL),
|
| frame_index_(0) {
|
| - buffer_.reset(new uint8[kBufferSize]);
|
| - frames_[0] = new ScreenCaptureData(
|
| - buffer_.get(), kTestFrameWidth1 * ScreenCaptureData::kBytesPerPixel,
|
| - SkISize::Make(kTestFrameWidth1, kTestFrameHeight1));
|
| - frames_[1] = new ScreenCaptureData(
|
| - buffer_.get(), kTestFrameWidth2 * ScreenCaptureData::kBytesPerPixel,
|
| - SkISize::Make(kTestFrameWidth2, kTestFrameHeight2));
|
| }
|
| virtual ~FakeScreenCapturer() {}
|
|
|
| // VideoFrameCapturer interface.
|
| - virtual void Start(Delegate* delegate) OVERRIDE {
|
| - delegate_ = delegate;
|
| + virtual void Start(Callback* callback) OVERRIDE {
|
| + callback_ = callback;
|
| }
|
| - virtual void CaptureFrame() OVERRIDE {
|
| - scoped_refptr<ScreenCaptureData> frame =
|
| - frames_[frame_index_ % arraysize(frames_)];
|
| +
|
| + virtual void Capture(const webrtc::DesktopRegion& region) OVERRIDE {
|
| + webrtc::DesktopSize size;
|
| + if (frame_index_ % 2 == 0) {
|
| + size = webrtc::DesktopSize(kTestFrameWidth1, kTestFrameHeight1);
|
| + } else {
|
| + size = webrtc::DesktopSize(kTestFrameWidth2, kTestFrameHeight2);
|
| + }
|
| frame_index_++;
|
| - delegate_->OnCaptureCompleted(frame);
|
| + callback_->OnCaptureCompleted(new webrtc::BasicDesktopFrame(size));
|
| + }
|
| +
|
| + virtual void SetMouseShapeObserver(
|
| + MouseShapeObserver* mouse_shape_observer) OVERRIDE {
|
| }
|
|
|
| private:
|
| - Delegate* delegate_;
|
| - scoped_ptr<uint8[]> buffer_;
|
| - scoped_refptr<ScreenCaptureData> frames_[2];
|
| + Callback* callback_;
|
| int frame_index_;
|
| };
|
|
|
|
|