Index: webrtc/modules/desktop_capture/screen_capturer_integration_test.cc |
diff --git a/webrtc/modules/desktop_capture/screen_capturer_integration_test.cc b/webrtc/modules/desktop_capture/screen_capturer_integration_test.cc |
index f208f14f595b719259045c552a3c431723cdfc96..5aa1f12f4d2de6e5e066e6e4caff326f4f9f5e04 100644 |
--- a/webrtc/modules/desktop_capture/screen_capturer_integration_test.cc |
+++ b/webrtc/modules/desktop_capture/screen_capturer_integration_test.cc |
@@ -26,7 +26,7 @@ |
#include "webrtc/modules/desktop_capture/desktop_capture_options.h" |
#include "webrtc/modules/desktop_capture/desktop_frame.h" |
#include "webrtc/modules/desktop_capture/desktop_region.h" |
-#include "webrtc/modules/desktop_capture/screen_capturer_mock_objects.h" |
+#include "webrtc/modules/desktop_capture/mock_desktop_capturer_callback.h" |
#include "webrtc/modules/desktop_capture/screen_drawer.h" |
#if defined(WEBRTC_WIN) |
@@ -77,16 +77,16 @@ bool ArePixelsColoredBy(const DesktopFrame& frame, |
class ScreenCapturerIntegrationTest : public testing::Test { |
public: |
void SetUp() override { |
- capturer_.reset( |
- ScreenCapturer::Create(DesktopCaptureOptions::CreateDefault())); |
+ capturer_ = DesktopCapturer::CreateScreenCapturer( |
+ DesktopCaptureOptions::CreateDefault()); |
} |
protected: |
void TestCaptureUpdatedRegion( |
- std::initializer_list<ScreenCapturer*> capturers) { |
+ std::initializer_list<DesktopCapturer*> capturers) { |
RTC_DCHECK(capturers.size() > 0); |
- // A large enough area for the tests, which should be able to fulfill by |
- // most of systems. |
+ // A large enough area for the tests, which should be able to be fulfilled |
+ // by most systems. |
const int kTestArea = 512; |
const int kRectSize = 32; |
std::unique_ptr<ScreenDrawer> drawer = ScreenDrawer::Create(); |
@@ -101,7 +101,7 @@ class ScreenCapturerIntegrationTest : public testing::Test { |
return; |
} |
- for (ScreenCapturer* capturer : capturers) { |
+ for (DesktopCapturer* capturer : capturers) { |
capturer->Start(&callback_); |
} |
@@ -138,11 +138,12 @@ class ScreenCapturerIntegrationTest : public testing::Test { |
#if defined(WEBRTC_WIN) |
// Enable allow_directx_capturer in DesktopCaptureOptions, but let |
- // ScreenCapturer::Create to decide whether a DirectX capturer should be used. |
+ // DesktopCapturer::CreateScreenCapturer() to decide whether a DirectX |
+ // capturer should be used. |
void MaybeCreateDirectxCapturer() { |
DesktopCaptureOptions options(DesktopCaptureOptions::CreateDefault()); |
options.set_allow_directx_capturer(true); |
- capturer_.reset(ScreenCapturer::Create(options)); |
+ capturer_ = DesktopCapturer::CreateScreenCapturer(options); |
} |
bool CreateDirectxCapturer() { |
@@ -158,19 +159,19 @@ class ScreenCapturerIntegrationTest : public testing::Test { |
void CreateMagnifierCapturer() { |
DesktopCaptureOptions options(DesktopCaptureOptions::CreateDefault()); |
options.set_allow_use_magnification_api(true); |
- capturer_.reset(ScreenCapturer::Create(options)); |
+ capturer_ = DesktopCapturer::CreateScreenCapturer(options); |
} |
#endif // defined(WEBRTC_WIN) |
- std::unique_ptr<ScreenCapturer> capturer_; |
- MockScreenCapturerCallback callback_; |
+ std::unique_ptr<DesktopCapturer> capturer_; |
+ MockDesktopCapturerCallback callback_; |
private: |
// Repeats capturing the frame by using |capturers| one-by-one for 600 times, |
// typically 30 seconds, until they succeeded captured a |color| rectangle at |
// |rect|. This function uses |drawer|->WaitForPendingDraws() between two |
// attempts to wait for the screen to update. |
- void TestCaptureOneFrame(std::vector<ScreenCapturer*> capturers, |
+ void TestCaptureOneFrame(std::vector<DesktopCapturer*> capturers, |
ScreenDrawer* drawer, |
DesktopRect rect, |
RgbaColor color) { |
@@ -182,7 +183,7 @@ class ScreenCapturerIntegrationTest : public testing::Test { |
drawer->WaitForPendingDraws(); |
for (size_t j = 0; j < capturers.size(); j++) { |
if (capturers[j] == nullptr) { |
- // ScreenCapturer should return an empty updated_region() if no |
+ // DesktopCapturer should return an empty updated_region() if no |
// update detected. So we won't test it again if it has captured |
// the rectangle we drew. |
continue; |
@@ -210,7 +211,7 @@ class ScreenCapturerIntegrationTest : public testing::Test { |
} |
// Expects |capturer| to successfully capture a frame, and returns it. |
- std::unique_ptr<DesktopFrame> CaptureFrame(ScreenCapturer* capturer) { |
+ std::unique_ptr<DesktopFrame> CaptureFrame(DesktopCapturer* capturer) { |
std::unique_ptr<DesktopFrame> frame; |
EXPECT_CALL(callback_, |
OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, _)) |
@@ -226,7 +227,7 @@ TEST_F(ScreenCapturerIntegrationTest, CaptureUpdatedRegion) { |
} |
TEST_F(ScreenCapturerIntegrationTest, TwoCapturers) { |
- std::unique_ptr<ScreenCapturer> capturer2 = std::move(capturer_); |
+ std::unique_ptr<DesktopCapturer> capturer2 = std::move(capturer_); |
SetUp(); |
TestCaptureUpdatedRegion({capturer_.get(), capturer2.get()}); |
} |
@@ -246,7 +247,7 @@ TEST_F(ScreenCapturerIntegrationTest, TwoDirectxCapturers) { |
return; |
} |
- std::unique_ptr<ScreenCapturer> capturer2 = std::move(capturer_); |
+ std::unique_ptr<DesktopCapturer> capturer2 = std::move(capturer_); |
RTC_CHECK(CreateDirectxCapturer()); |
TestCaptureUpdatedRegion({capturer_.get(), capturer2.get()}); |
} |
@@ -259,7 +260,7 @@ TEST_F(ScreenCapturerIntegrationTest, |
TEST_F(ScreenCapturerIntegrationTest, TwoMagnifierCapturers) { |
CreateMagnifierCapturer(); |
- std::unique_ptr<ScreenCapturer> capturer2 = std::move(capturer_); |
+ std::unique_ptr<DesktopCapturer> capturer2 = std::move(capturer_); |
CreateMagnifierCapturer(); |
TestCaptureUpdatedRegion({capturer_.get(), capturer2.get()}); |
} |