Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(148)

Side by Side Diff: webrtc/modules/desktop_capture/screen_capturer_unittest.cc

Issue 2030333003: Revert of Use std::unique_ptr<> to pass frame ownership in DesktopCapturer impls. (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source 5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include <memory> 11 #include <memory>
12 12
13 #include "webrtc/modules/desktop_capture/screen_capturer.h" 13 #include "webrtc/modules/desktop_capture/screen_capturer.h"
14 14
15 #include "testing/gmock/include/gmock/gmock.h" 15 #include "testing/gmock/include/gmock/gmock.h"
16 #include "testing/gtest/include/gtest/gtest.h" 16 #include "testing/gtest/include/gtest/gtest.h"
17 #include "webrtc/base/constructormagic.h" 17 #include "webrtc/base/constructormagic.h"
18 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" 18 #include "webrtc/modules/desktop_capture/desktop_capture_options.h"
19 #include "webrtc/modules/desktop_capture/desktop_frame.h" 19 #include "webrtc/modules/desktop_capture/desktop_frame.h"
20 #include "webrtc/modules/desktop_capture/desktop_region.h" 20 #include "webrtc/modules/desktop_capture/desktop_region.h"
21 #include "webrtc/modules/desktop_capture/screen_capturer_mock_objects.h" 21 #include "webrtc/modules/desktop_capture/screen_capturer_mock_objects.h"
22 22
23 using ::testing::_; 23 using ::testing::_;
24 using ::testing::AnyNumber; 24 using ::testing::AnyNumber;
25 using ::testing::Return; 25 using ::testing::Return;
26 using ::testing::SaveArg;
26 27
27 const int kTestSharedMemoryId = 123; 28 const int kTestSharedMemoryId = 123;
28 29
29 namespace webrtc { 30 namespace webrtc {
30 31
31 class ScreenCapturerTest : public testing::Test { 32 class ScreenCapturerTest : public testing::Test {
32 public: 33 public:
33 void SetUp() override { 34 void SetUp() override {
34 capturer_.reset( 35 capturer_.reset(
35 ScreenCapturer::Create(DesktopCaptureOptions::CreateDefault())); 36 ScreenCapturer::Create(DesktopCaptureOptions::CreateDefault()));
(...skipping 25 matching lines...) Expand all
61 62
62 std::unique_ptr<SharedMemory> CreateSharedMemory(size_t size) override { 63 std::unique_ptr<SharedMemory> CreateSharedMemory(size_t size) override {
63 return std::unique_ptr<SharedMemory>( 64 return std::unique_ptr<SharedMemory>(
64 new FakeSharedMemory(new char[size], size)); 65 new FakeSharedMemory(new char[size], size));
65 } 66 }
66 67
67 private: 68 private:
68 RTC_DISALLOW_COPY_AND_ASSIGN(FakeSharedMemoryFactory); 69 RTC_DISALLOW_COPY_AND_ASSIGN(FakeSharedMemoryFactory);
69 }; 70 };
70 71
71 ACTION_P(SaveUniquePtrArg, dest) {
72 *dest = std::move(*arg1);
73 }
74
75 TEST_F(ScreenCapturerTest, GetScreenListAndSelectScreen) { 72 TEST_F(ScreenCapturerTest, GetScreenListAndSelectScreen) {
76 webrtc::ScreenCapturer::ScreenList screens; 73 webrtc::ScreenCapturer::ScreenList screens;
77 EXPECT_TRUE(capturer_->GetScreenList(&screens)); 74 EXPECT_TRUE(capturer_->GetScreenList(&screens));
78 for(webrtc::ScreenCapturer::ScreenList::iterator it = screens.begin(); 75 for(webrtc::ScreenCapturer::ScreenList::iterator it = screens.begin();
79 it != screens.end(); ++it) { 76 it != screens.end(); ++it) {
80 EXPECT_TRUE(capturer_->SelectScreen(it->id)); 77 EXPECT_TRUE(capturer_->SelectScreen(it->id));
81 } 78 }
82 } 79 }
83 80
84 TEST_F(ScreenCapturerTest, StartCapturer) { 81 TEST_F(ScreenCapturerTest, StartCapturer) {
85 capturer_->Start(&callback_); 82 capturer_->Start(&callback_);
86 } 83 }
87 84
88 TEST_F(ScreenCapturerTest, Capture) { 85 TEST_F(ScreenCapturerTest, Capture) {
89 // Assume that Start() treats the screen as invalid initially. 86 // Assume that Start() treats the screen as invalid initially.
90 std::unique_ptr<DesktopFrame> frame; 87 DesktopFrame* frame = NULL;
91 EXPECT_CALL(callback_, 88 EXPECT_CALL(callback_, OnCaptureCompleted(_))
92 OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, _)) 89 .WillOnce(SaveArg<0>(&frame));
93 .WillOnce(SaveUniquePtrArg(&frame));
94 90
95 capturer_->Start(&callback_); 91 capturer_->Start(&callback_);
96 capturer_->Capture(DesktopRegion()); 92 capturer_->Capture(DesktopRegion());
97 93
98 ASSERT_TRUE(frame); 94 ASSERT_TRUE(frame);
99 EXPECT_GT(frame->size().width(), 0); 95 EXPECT_GT(frame->size().width(), 0);
100 EXPECT_GT(frame->size().height(), 0); 96 EXPECT_GT(frame->size().height(), 0);
101 EXPECT_GE(frame->stride(), 97 EXPECT_GE(frame->stride(),
102 frame->size().width() * DesktopFrame::kBytesPerPixel); 98 frame->size().width() * DesktopFrame::kBytesPerPixel);
103 EXPECT_TRUE(frame->shared_memory() == NULL); 99 EXPECT_TRUE(frame->shared_memory() == NULL);
104 100
105 // Verify that the region contains whole screen. 101 // Verify that the region contains whole screen.
106 EXPECT_FALSE(frame->updated_region().is_empty()); 102 EXPECT_FALSE(frame->updated_region().is_empty());
107 DesktopRegion::Iterator it(frame->updated_region()); 103 DesktopRegion::Iterator it(frame->updated_region());
108 ASSERT_TRUE(!it.IsAtEnd()); 104 ASSERT_TRUE(!it.IsAtEnd());
109 EXPECT_TRUE(it.rect().equals(DesktopRect::MakeSize(frame->size()))); 105 EXPECT_TRUE(it.rect().equals(DesktopRect::MakeSize(frame->size())));
110 it.Advance(); 106 it.Advance();
111 EXPECT_TRUE(it.IsAtEnd()); 107 EXPECT_TRUE(it.IsAtEnd());
108
109 delete frame;
112 } 110 }
113 111
114 #if defined(WEBRTC_WIN) 112 #if defined(WEBRTC_WIN)
115 113
116 TEST_F(ScreenCapturerTest, UseSharedBuffers) { 114 TEST_F(ScreenCapturerTest, UseSharedBuffers) {
117 std::unique_ptr<DesktopFrame> frame; 115 DesktopFrame* frame = NULL;
118 EXPECT_CALL(callback_, 116 EXPECT_CALL(callback_, OnCaptureCompleted(_))
119 OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, _)) 117 .WillOnce(SaveArg<0>(&frame));
120 .WillOnce(SaveUniquePtrArg(&frame));
121 118
122 capturer_->Start(&callback_); 119 capturer_->Start(&callback_);
123 capturer_->SetSharedMemoryFactory( 120 capturer_->SetSharedMemoryFactory(
124 std::unique_ptr<SharedMemoryFactory>(new FakeSharedMemoryFactory())); 121 std::unique_ptr<SharedMemoryFactory>(new FakeSharedMemoryFactory()));
125 capturer_->Capture(DesktopRegion()); 122 capturer_->Capture(DesktopRegion());
126 123
127 ASSERT_TRUE(frame); 124 ASSERT_TRUE(frame);
128 ASSERT_TRUE(frame->shared_memory()); 125 ASSERT_TRUE(frame->shared_memory());
129 EXPECT_EQ(frame->shared_memory()->id(), kTestSharedMemoryId); 126 EXPECT_EQ(frame->shared_memory()->id(), kTestSharedMemoryId);
127
128 delete frame;
130 } 129 }
131 130
132 TEST_F(ScreenCapturerTest, UseMagnifier) { 131 TEST_F(ScreenCapturerTest, UseMagnifier) {
133 DesktopCaptureOptions options(DesktopCaptureOptions::CreateDefault()); 132 DesktopCaptureOptions options(DesktopCaptureOptions::CreateDefault());
134 options.set_allow_use_magnification_api(true); 133 options.set_allow_use_magnification_api(true);
135 capturer_.reset(ScreenCapturer::Create(options)); 134 capturer_.reset(ScreenCapturer::Create(options));
136 135
137 std::unique_ptr<DesktopFrame> frame; 136 DesktopFrame* frame = NULL;
138 EXPECT_CALL(callback_, 137 EXPECT_CALL(callback_, OnCaptureCompleted(_)).WillOnce(SaveArg<0>(&frame));
139 OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, _))
140 .WillOnce(SaveUniquePtrArg(&frame));
141 138
142 capturer_->Start(&callback_); 139 capturer_->Start(&callback_);
143 capturer_->Capture(DesktopRegion()); 140 capturer_->Capture(DesktopRegion());
144 ASSERT_TRUE(frame); 141 ASSERT_TRUE(frame);
142 delete frame;
145 } 143 }
146 144
147 #endif // defined(WEBRTC_WIN) 145 #endif // defined(WEBRTC_WIN)
148 146
149 } // namespace webrtc 147 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/modules/desktop_capture/screen_capturer_mock_objects.h ('k') | webrtc/modules/desktop_capture/screen_capturer_x11.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698