Use webrtc::DesktopCapturer for screen capturer implementation.

media/video/capture/screen/screen_capture_device_unittest.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/video/capture/screen/screen_capture_device.h"
 
 #include "base/basictypes.h"
 #include "base/sequenced_task_runner.h"
 #include "base/synchronization/waitable_event.h"
 #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;
 using ::testing::InvokeWithoutArgs;
 using ::testing::SaveArg;
 
 namespace media {
 
 namespace {
 
@@ skipping 18 matching lines @@
                                              bool flip_horiz));
   MOCK_METHOD2(OnIncomingCapturedVideoFrame,
       void(const scoped_refptr<media::VideoFrame>& frame,
            base::Time timestamp));
 };
 
 // TODO(sergeyu): Move this to a separate file where it can be reused.
 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_ == 0) {

[alexeypa (please no reviews), 2013/05/08 22:24:59]

Previously, this method was alternating before 1st and 2nd test frames. Now it
returns the 1st frame only once.

[Sergey Ulanov, 2013/05/09 18:49:02]

Doesn't really matter for the test that uses this class. Fixed it anyway.

+      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_;
 };
 
 class ScreenCaptureDeviceTest : public testing::Test {
  public:
   virtual void SetUp() OVERRIDE {
     worker_pool_ = new base::SequencedWorkerPool(3, "TestCaptureThread");
   }
 
  protected:
@@ skipping 70 matching lines @@
   EXPECT_EQ(kTestFrameWidth1, caps.width);
   EXPECT_EQ(kTestFrameHeight1, caps.height);
   EXPECT_EQ(kFrameRate, caps.frame_rate);
   EXPECT_EQ(VideoCaptureCapability::kARGB, caps.color);
   EXPECT_FALSE(caps.interlaced);
 
   EXPECT_EQ(caps.width * caps.height * 4, frame_size);
 }
 
 }  // namespace media