1 /*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15 #include <fcntl.h>
16 #include <gtest/gtest.h>
17 #include "display.h"
18 #include "display_manager.h"
19 #include "snapshot_utils.h"
20 #include "common_test_utils.h"
21
22 using namespace testing;
23 using namespace testing::ext;
24
25 namespace OHOS {
26 namespace Rosen {
27 namespace {
28 constexpr int BPP = 4;
29 }
30 class SnapshotUtilsTest : public testing::Test {
31 public:
32 static void SetUpTestCase();
33 static void TearDownTestCase();
34 virtual void SetUp() override;
35 virtual void TearDown() override;
36 const std::string defaultFile_ = "/data/snapshot_display_1.jpeg";
37 const int defaultBitDepth_ = 8;
38 };
39
SetUpTestCase()40 void SnapshotUtilsTest::SetUpTestCase()
41 {
42 CommonTestUtils::InjectTokenInfoByHapName(0, "com.ohos.systemui", 0);
43 const char** perms = new const char *[1];
44 perms[0] = "ohos.permission.CAPTURE_SCREEN";
45 CommonTestUtils::SetAceessTokenPermission("DisplayManagerServiceTest", perms, 1);
46 }
47
TearDownTestCase()48 void SnapshotUtilsTest::TearDownTestCase()
49 {
50 }
51
SetUp()52 void SnapshotUtilsTest::SetUp()
53 {
54 }
55
TearDown()56 void SnapshotUtilsTest::TearDown()
57 {
58 }
59
60 namespace {
61 /**
62 * @tc.name: Check01
63 * @tc.desc: Check if default jpeg is valid file names
64 * @tc.type: FUNC
65 */
66 HWTEST_F(SnapshotUtilsTest, Check01, Function | SmallTest | Level3)
67 {
68 ASSERT_EQ(true, SnapShotUtils::CheckFileNameValid(defaultFile_));
69 }
70
71 /**
72 * @tc.name: Check02
73 * @tc.desc: Check custom jpeg is valid file names
74 * @tc.type: FUNC
75 */
76 HWTEST_F(SnapshotUtilsTest, Check02, Function | SmallTest | Level3)
77 {
78 std::string fileName = "/data/test.jpeg";
79 ASSERT_EQ(true, SnapShotUtils::CheckFileNameValid(fileName));
80 }
81
82 /**
83 * @tc.name: Check03
84 * @tc.desc: Check random path is invalid file names
85 * @tc.type: FUNC
86 */
87 HWTEST_F(SnapshotUtilsTest, Check03, Function | SmallTest | Level3)
88 {
89 std::string fileName1 = "/path/to/test/1.jpeg";
90 ASSERT_EQ(false, SnapShotUtils::CheckFileNameValid(fileName1));
91 std::string fileName2 = "";
92 ASSERT_EQ(false, SnapShotUtils::CheckFileNameValid(fileName2));
93 std::string fileName3 = "/data/test.png";
94 ASSERT_EQ(false, SnapShotUtils::CheckFileNameValid(fileName3));
95 }
96
97 /**
98 * @tc.name: RGBA8888ToRGB88801
99 * @tc.desc: RGBA8888 to RGB888 using invalid params
100 * @tc.type: FUNC
101 */
102 HWTEST_F(SnapshotUtilsTest, RGBA8888ToRGB88801, Function | SmallTest | Level3)
103 {
104 ASSERT_FALSE(SnapShotUtils::RGBA8888ToRGB888(nullptr, nullptr, -1));
105 }
106
107 /**
108 * @tc.name: WriteRgb888ToJpeg01
109 * @tc.desc: write rgb888 to jpeg using invalid data
110 * @tc.type: FUNC
111 */
112 HWTEST_F(SnapshotUtilsTest, WriteRgb888ToJpeg01, Function | SmallTest | Level3)
113 {
114 uint8_t *data = nullptr;
115 FILE *file = fopen(defaultFile_.c_str(), "wb");
116 if (file == nullptr) {
117 return;
118 }
119 ASSERT_FALSE(SnapShotUtils::WriteRgb888ToJpeg(file, 100, 100, data));
120 fclose(file);
121 }
122
123 /**
124 * @tc.name: WriteRgb888ToJpeg02
125 * @tc.desc: write rgb888 to jpeg using invalid file
126 * @tc.type: FUNC
127 */
128 HWTEST_F(SnapshotUtilsTest, WriteRgb888ToJpeg02, Function | SmallTest | Level3)
129 {
130 uint8_t *data = new uint8_t;
131 FILE *file = nullptr;
132 ASSERT_FALSE(SnapShotUtils::WriteRgb888ToJpeg(file, 100, 100, data));
133 }
134
135 /**
136 * @tc.name: Write01
137 * @tc.desc: Write default jpeg using valid file names and valid PixelMap
138 * @tc.type: FUNC
139 */
140 HWTEST_F(SnapshotUtilsTest, Write01, Function | MediumTest | Level3)
141 {
142 DisplayId id = DisplayManager::GetInstance().GetDefaultDisplayId();
143 std::shared_ptr<Media::PixelMap> pixelMap = DisplayManager::GetInstance().GetScreenshot(id);
144 ASSERT_NE(nullptr, pixelMap);
145 ASSERT_EQ(true, SnapShotUtils::WriteToJpegWithPixelMap(defaultFile_, *pixelMap));
146 }
147
148 /**
149 * @tc.name: Write02
150 * @tc.desc: Write default jpeg using valid file names and valid WriteToJpegParam
151 * @tc.type: FUNC
152 */
153 HWTEST_F(SnapshotUtilsTest, Write02, Function | MediumTest | Level3)
154 {
155 DisplayId id = DisplayManager::GetInstance().GetDefaultDisplayId();
156 std::shared_ptr<Media::PixelMap> pixelMap = DisplayManager::GetInstance().GetScreenshot(id);
157 ASSERT_NE(nullptr, pixelMap);
158 WriteToJpegParam param = {
159 .width = pixelMap->GetWidth(),
160 .height = pixelMap->GetHeight(),
161 .stride = pixelMap->GetRowBytes(),
162 .format = pixelMap->GetPixelFormat(),
163 .data = pixelMap->GetPixels()
164 };
165 ASSERT_EQ(true, SnapShotUtils::WriteToJpeg(defaultFile_, param));
166 }
167
168 /**
169 * @tc.name: Write03
170 * @tc.desc: Write custom jpeg using valid file names and valid WriteToJpegParam
171 * @tc.type: FUNC
172 */
173 HWTEST_F(SnapshotUtilsTest, Write03, Function | MediumTest | Level3)
174 {
175 DisplayId id = DisplayManager::GetInstance().GetDefaultDisplayId();
176 std::shared_ptr<Media::PixelMap> pixelMap = DisplayManager::GetInstance().GetScreenshot(id);
177 ASSERT_NE(nullptr, pixelMap);
178 WriteToJpegParam param = {
179 .width = (pixelMap->GetWidth() / 2),
180 .height = (pixelMap->GetWidth() / 2),
181 .stride = pixelMap->GetRowBytes(),
182 .format = pixelMap->GetPixelFormat(),
183 .data = pixelMap->GetPixels()
184 };
185 ASSERT_EQ(false, SnapShotUtils::WriteToJpeg(defaultFile_, param));
186 }
187
188 /**
189 * @tc.name: Write04
190 * @tc.desc: Write pixel map with jpeg, using fd
191 * @tc.type: FUNC
192 */
193 HWTEST_F(SnapshotUtilsTest, Write04, Function | MediumTest | Level3)
194 {
195 DisplayId id = DisplayManager::GetInstance().GetDefaultDisplayId();
196 std::shared_ptr<Media::PixelMap> pixelMap = DisplayManager::GetInstance().GetScreenshot(id);
197 int fd = open(defaultFile_.c_str(), O_RDWR | O_CREAT | O_TRUNC, 0666);
198 if (fd == -1) {
199 return;
200 }
201 ASSERT_EQ(true, SnapShotUtils::WriteToJpegWithPixelMap(fd, *pixelMap));
202 close(fd);
203 }
204
205 /**
206 * @tc.name: Write05
207 * @tc.desc: Write custom jpeg using invalid file names and valid WriteToJpegParam
208 * @tc.type: FUNC
209 */
210 HWTEST_F(SnapshotUtilsTest, Write05, Function | MediumTest | Level3)
211 {
212 WriteToJpegParam param = {
213 .width = 256,
214 .height = 256,
215 .stride = 256 * BPP,
216 .format = Media::PixelFormat::UNKNOWN,
217 .data = new uint8_t
218 };
219 ASSERT_FALSE(SnapShotUtils::WriteToJpeg("", param));
220 }
221
222 /**
223 * @tc.name: Write06
224 * @tc.desc: Write custom jpeg using valid file names and invalid WriteToJpegParam
225 * @tc.type: FUNC
226 */
227 HWTEST_F(SnapshotUtilsTest, Write06, Function | MediumTest | Level3)
228 {
229 WriteToJpegParam param = {
230 .width = 256,
231 .height = 256,
232 .stride = 256 * BPP,
233 .format = Media::PixelFormat::UNKNOWN,
234 .data = nullptr
235 };
236 ASSERT_FALSE(SnapShotUtils::WriteToJpeg(defaultFile_, param));
237 }
238
239 /**
240 * @tc.name: Write07
241 * @tc.desc: Write custom jpeg using valid fd and invalid WriteToJpegParam
242 * @tc.type: FUNC
243 */
244 HWTEST_F(SnapshotUtilsTest, Write07, Function | MediumTest | Level3)
245 {
246 WriteToJpegParam param = {
247 .width = 256,
248 .height = 256,
249 .stride = 256 * BPP,
250 .format = Media::PixelFormat::UNKNOWN,
251 .data = nullptr
252 };
253 ASSERT_FALSE(SnapShotUtils::WriteToJpeg(1, param));
254 }
255
256 /**
257 * @tc.name: CheckWHValid
258 * @tc.desc: Check width and height whether valid
259 * @tc.type: FUNC
260 */
261 HWTEST_F(SnapshotUtilsTest, CheckWHValid, Function | SmallTest | Level3)
262 {
263 ASSERT_EQ(false, SnapShotUtils::CheckWHValid(0));
264 ASSERT_EQ(true, SnapShotUtils::CheckWHValid(DisplayManager::MAX_RESOLUTION_SIZE_SCREENSHOT));
265 ASSERT_EQ(false, SnapShotUtils::CheckWHValid(DisplayManager::MAX_RESOLUTION_SIZE_SCREENSHOT + 1));
266 }
267
268 /**
269 * @tc.name: CheckParamValid01
270 * @tc.desc: Check jpeg param whether valid width
271 * @tc.type: FUNC
272 */
273 HWTEST_F(SnapshotUtilsTest, CheckParamValid01, Function | SmallTest | Level3)
274 {
275 WriteToJpegParam paramInvalidWidth = {
276 .width = DisplayManager::MAX_RESOLUTION_SIZE_SCREENSHOT + 1,
277 .height = 0,
278 .stride = 0,
279 .format = Media::PixelFormat::UNKNOWN,
280 .data = nullptr
281 };
282 ASSERT_EQ(false, SnapShotUtils::CheckParamValid(paramInvalidWidth));
283 }
284
285 /**
286 * @tc.name: CheckParamValid02
287 * @tc.desc: Check jpeg param whether valid height
288 * @tc.type: FUNC
289 */
290 HWTEST_F(SnapshotUtilsTest, CheckParamValid02, Function | SmallTest | Level3)
291 {
292 WriteToJpegParam paramInvalidHeight = {
293 .width = DisplayManager::MAX_RESOLUTION_SIZE_SCREENSHOT,
294 .height = 0,
295 .stride = 0,
296 .format = Media::PixelFormat::UNKNOWN,
297 .data = nullptr
298 };
299 ASSERT_EQ(false, SnapShotUtils::CheckParamValid(paramInvalidHeight));
300 }
301
302 /**
303 * @tc.name: CheckParamValid03
304 * @tc.desc: Check jpeg param whether valid stride
305 * @tc.type: FUNC
306 */
307 HWTEST_F(SnapshotUtilsTest, CheckParamValid03, Function | SmallTest | Level3)
308 {
309 WriteToJpegParam paramInvalidStride = {
310 .width = 256,
311 .height = 256,
312 .stride = 1,
313 .format = Media::PixelFormat::UNKNOWN,
314 .data = nullptr
315 };
316 ASSERT_EQ(false, SnapShotUtils::CheckParamValid(paramInvalidStride));
317 }
318
319 /**
320 * @tc.name: CheckParamValid04
321 * @tc.desc: Check jpeg param whether valid data
322 * @tc.type: FUNC
323 */
324 HWTEST_F(SnapshotUtilsTest, CheckParamValid04, Function | SmallTest | Level3)
325 {
326 WriteToJpegParam paramInvalidData = {
327 .width = 256,
328 .height = 256,
329 .stride = 256 * BPP,
330 .format = Media::PixelFormat::UNKNOWN,
331 .data = nullptr
332 };
333 ASSERT_EQ(false, SnapShotUtils::CheckParamValid(paramInvalidData));
334 }
335 }
336 } // namespace Rosen
337 } // namespace OHOS
