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 <gtest/gtest.h>
16 #include "iconsumer_surface.h"
17 #include <iservice_registry.h>
18 #include <ctime>
19 #include "native_buffer.h"
20 #include "native_buffer_inner.h"
21 #include "native_window.h"
22 #include "surface_type.h"
23 #include "graphic_common_c.h"
24 
25 using namespace std;
26 using namespace testing;
27 using namespace testing::ext;
28 
29 namespace OHOS::Rosen {
30 class BufferConsumerListener : public IBufferConsumerListener {
31 public:
OnBufferAvailable()32     void OnBufferAvailable() override
33     {
34     }
35 };
36 
37 class NativeBufferTest : public testing::Test {
38 public:
39     static void SetUpTestCase();
40     static void TearDownTestCase();
41 
42     static inline OH_NativeBuffer_Config config = {
43         .width = 0x100,
44         .height = 0x100,
45         .format = GRAPHIC_PIXEL_FMT_RGBA_8888,
46         .usage = BUFFER_USAGE_CPU_READ | BUFFER_USAGE_CPU_WRITE | BUFFER_USAGE_MEM_DMA,
47     };
48     static inline OH_NativeBuffer_Config checkConfig = {};
49     static inline OH_NativeBuffer* buffer = nullptr;
50 };
51 
SetUpTestCase()52 void NativeBufferTest::SetUpTestCase()
53 {
54     buffer = nullptr;
55 }
56 
TearDownTestCase()57 void NativeBufferTest::TearDownTestCase()
58 {
59     buffer = nullptr;
60 }
61 
62 /*
63 * Function: OH_NativeBufferFromSurfaceBuffer
64 * Type: Function
65 * Rank: Important(2)
66 * EnvConditions: N/A
67 * CaseDescription: 1. call OH_NativeBufferFromSurfaceBuffer by abnormal input
68 *                  2. check ret
69 */
70 HWTEST_F(NativeBufferTest, OH_NativeBufferFromSurfaceBuffer001, Function | MediumTest | Level2)
71 {
72     sptr<OHOS::SurfaceBuffer> surfaceBuffer = OHOS::SurfaceBuffer::Create();
73     NativeWindowBuffer* nativeWindowBuffer = OH_NativeWindow_CreateNativeWindowBufferFromSurfaceBuffer(&surfaceBuffer);
74     ASSERT_NE(nativeWindowBuffer, nullptr);
75     nativeWindowBuffer->sfbuffer = nullptr;
76     OH_NativeBuffer* nativeBuffer = OH_NativeBufferFromNativeWindowBuffer(nativeWindowBuffer);
77     ASSERT_EQ(nativeBuffer, nullptr);
78 }
79 
80 /*
81 * Function: OH_NativeBuffer_Alloc
82 * Type: Function
83 * Rank: Important(2)
84 * EnvConditions: N/A
85 * CaseDescription: 1. call OH_NativeBuffer_Alloc by abnormal input
86 *                  2. check ret
87 */
88 HWTEST_F(NativeBufferTest, OHNativeBufferAlloc001, Function | MediumTest | Level2)
89 {
90     buffer = OH_NativeBuffer_Alloc(nullptr);
91     ASSERT_EQ(buffer, nullptr);
92 }
93 
94 /*
95 * Function: OH_NativeBuffer_Alloc
96 * Type: Function
97 * Rank: Important(2)
98 * EnvConditions: N/A
99 * CaseDescription: 1. call OH_NativeBuffer_Alloc
100 *                  2. check ret
101 */
102 HWTEST_F(NativeBufferTest, OHNativeBufferAlloc002, Function | MediumTest | Level2)
103 {
104     buffer = OH_NativeBuffer_Alloc(&config);
105     ASSERT_NE(buffer, nullptr);
106 }
107 
108 /*
109 * Function: OH_NativeBuffer_GetSeqNum
110 * Type: Function
111 * Rank: Important(2)
112 * EnvConditions: N/A
113 * CaseDescription: 1. call OH_NativeBuffer_GetSeqNum by abnormal input
114 *                  2. check ret
115 */
116 HWTEST_F(NativeBufferTest, OHNativeBufferGetSeqNum001, Function | MediumTest | Level2)
117 {
118     uint32_t id = OH_NativeBuffer_GetSeqNum(nullptr);
119     ASSERT_EQ(id, UINT_MAX);
120 }
121 
122 /*
123 * Function: OH_NativeBuffer_GetSeqNum
124 * Type: Function
125 * Rank: Important(2)
126 * EnvConditions: N/A
127 * CaseDescription: 1. call OH_NativeBuffer_GetSeqNum
128 *                  2. check ret
129 */
130 HWTEST_F(NativeBufferTest, OHNativeBufferGetSeqNum002, Function | MediumTest | Level2)
131 {
132     uint32_t id = OH_NativeBuffer_GetSeqNum(buffer);
133     ASSERT_NE(id, GSERROR_INVALID_ARGUMENTS);
134 }
135 
136 /*
137 * Function: OH_NativeBuffer_GetConfig
138 * Type: Function
139 * Rank: Important(2)
140 * EnvConditions: N/A
141 * CaseDescription: 1. call OH_NativeBuffer_GetConfig
142 *                  2. check ret
143  */
144 HWTEST_F(NativeBufferTest, OHNativeBufferGetConfig001, Function | MediumTest | Level2)
145 {
146     OH_NativeBuffer_GetConfig(buffer, &checkConfig);
147     ASSERT_NE(&checkConfig, nullptr);
148 }
149 
150 /*
151 * Function: OH_NativeBuffer_GetConfig
152 * Type: Function
153 * Rank: Important(2)
154 * EnvConditions: N/A
155 * CaseDescription: 1. call OH_NativeBuffer_GetConfig by abnormal input
156 *                  2. check ret
157 */
158 HWTEST_F(NativeBufferTest, OHNativeBufferGetConfig002, Function | MediumTest | Level2)
159 {
160     checkConfig.width = 0x0;
161     checkConfig.height = 0x0;
162     checkConfig.format = 0x0;
163     checkConfig.usage = 0x0;
164     OH_NativeBuffer_GetConfig(nullptr, &checkConfig);
165     ASSERT_EQ(checkConfig.width, 0x0);
166     ASSERT_EQ(checkConfig.height, 0x0);
167     ASSERT_EQ(checkConfig.format, 0x0);
168     ASSERT_EQ(checkConfig.usage, 0x0);
169 }
170 
171 /*
172 * Function: OH_NativeBuffer_Reference
173 * Type: Function
174 * Rank: Important(2)
175 * EnvConditions: N/A
176 * CaseDescription: 1. call OH_NativeBuffer_Reference by abnormal input
177 *                  2. check ret
178 */
179 HWTEST_F(NativeBufferTest, OHNativeBufferReference001, Function | MediumTest | Level2)
180 {
181     int32_t ret = OH_NativeBuffer_Reference(nullptr);
182     ASSERT_NE(ret, GSERROR_OK);
183 }
184 
185 /*
186 * Function: OH_NativeBuffer_Reference
187 * Type: Function
188 * Rank: Important(2)
189 * EnvConditions: N/A
190 * CaseDescription: 1. call OH_NativeBuffer_Reference
191 *                  2. check ret
192 */
193 HWTEST_F(NativeBufferTest, OHNativeBufferReference002, Function | MediumTest | Level2)
194 {
195     int32_t ret = OH_NativeBuffer_Reference(buffer);
196     ASSERT_EQ(ret, GSERROR_OK);
197 }
198 
199 /*
200 * Function: OH_NativeBuffer_Unreference
201 * Type: Function
202 * Rank: Important(2)
203 * EnvConditions: N/A
204 * CaseDescription: 1. call OH_NativeBuffer_Unreference by abnormal input
205 *                  2. check ret
206 */
207 HWTEST_F(NativeBufferTest, OHNativeBufferUnreference001, Function | MediumTest | Level2)
208 {
209     int32_t ret = OH_NativeBuffer_Unreference(nullptr);
210     ASSERT_NE(ret, GSERROR_OK);
211 }
212 
213 /*
214 * Function: OH_NativeBuffer_Unreference
215 * Type: Function
216 * Rank: Important(2)
217 * EnvConditions: N/A
218 * CaseDescription: 1. call OH_NativeBuffer_Unreference
219 *                  2. check ret
220 */
221 HWTEST_F(NativeBufferTest, OHNativeBufferUnreference002, Function | MediumTest | Level2)
222 {
223     int32_t ret = OH_NativeBuffer_Unreference(buffer);
224     ASSERT_EQ(ret, GSERROR_OK);
225 }
226 
227 /*
228 * Function: OH_NativeBuffer_GetSeqNum and OH_NativeBuffer_Unreference
229 * Type: Function
230 * Rank: Important(2)
231 * EnvConditions: N/A
232 * CaseDescription: 1. call OH_NativeBuffer_GetSeqNum
233 *                  2. call OH_NativeBuffer_Unreference
234 *                  3. OH_NativeBuffer_Alloc again
235 *                  4. check OH_NativeBuffer_GetSeqNum = oldSeq + 1
236 */
237 HWTEST_F(NativeBufferTest, OHNativeBufferGetSeqNum003, Function | MediumTest | Level2)
238 {
239     uint32_t oldSeq = OH_NativeBuffer_GetSeqNum(buffer);
240     int32_t ret = OH_NativeBuffer_Unreference(buffer);
241     ASSERT_EQ(ret, GSERROR_OK);
242     buffer = OH_NativeBuffer_Alloc(&config);
243     ASSERT_EQ(oldSeq + 1, OH_NativeBuffer_GetSeqNum(buffer));
244 }
245 
246 /*
247 * Function: OH_NativeBuffer_GetNativeBufferConfig
248 * Type: Function
249 * Rank: Important(2)
250 * EnvConditions: N/A
251 * CaseDescription: 1. call OH_NativeBuffer_GetNativeBufferConfig
252 *                  2. check result
253 */
254 HWTEST_F(NativeBufferTest, OHNativeBufferGetNativeBufferConfig001, Function | MediumTest | Level2)
255 {
256     OH_NativeBuffer_Config testConfig = {};
257     OH_NativeBuffer_GetNativeBufferConfig(buffer, &testConfig);
258     ASSERT_EQ(testConfig.width, config.width);
259     ASSERT_EQ(testConfig.height, config.height);
260     ASSERT_EQ(testConfig.format, config.format);
261     ASSERT_EQ(testConfig.usage, config.usage);
262 }
263 
264 /*
265 * Function: OH_NativeBuffer_SetColorSpace
266 * Type: Function
267 * Rank: Important(2)
268 * EnvConditions: N/A
269 * CaseDescription: 1. call OH_NativeBuffer_SetColorSpace by abnormal input
270 *                  2. check ret
271 */
272 HWTEST_F(NativeBufferTest, OHNativeBufferSetColorSpace001, Function | MediumTest | Level2)
273 {
274     int32_t ret = OH_NativeBuffer_SetColorSpace(nullptr, OH_COLORSPACE_DISPLAY_BT2020_PQ);
275     ASSERT_NE(ret, GSERROR_OK);
276 }
277 
278 /*
279 * Function: OH_NativeBuffer_SetColorSpace
280 * Type: Function
281 * Rank: Important(2)
282 * EnvConditions: N/A
283 * CaseDescription: 1. call OH_NativeBuffer_SetColorSpace
284 *                  2. check ret
285 */
286 HWTEST_F(NativeBufferTest, OHNativeBufferSetColorSpace002, Function | MediumTest | Level2)
287 {
288     if (buffer == nullptr) {
289         buffer = OH_NativeBuffer_Alloc(&config);
290         ASSERT_NE(buffer, nullptr);
291     }
292 
293     int32_t ret = OH_NativeBuffer_SetColorSpace(buffer, OH_COLORSPACE_BT709_LIMIT);
294     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
295         ASSERT_EQ(ret, GSERROR_OK);
296     }
297 }
298 
299 /*
300 * Function: OH_NativeBuffer_GetColorSpace
301 * Type: Function
302 * Rank: Important(2)
303 * EnvConditions: N/A
304 * CaseDescription: 1. call OH_NativeBuffer_GetColorSpace by abnormal input
305 *                  2. check ret
306 */
307 HWTEST_F(NativeBufferTest, OHNativeBufferGetColorSpace001, Function | MediumTest | Level2)
308 {
309     OH_NativeBuffer_ColorSpace *colorSpace = nullptr;
310     int32_t ret = OH_NativeBuffer_GetColorSpace(nullptr, colorSpace);
311     ASSERT_NE(ret, GSERROR_OK);
312 }
313 
314 /*
315 * Function: OH_NativeBuffer_GetColorSpace
316 * Type: Function
317 * Rank: Important(2)
318 * EnvConditions: N/A
319 * CaseDescription: 1. call OH_NativeBuffer_GetColorSpace
320 *                  2. check ret
321 */
322 HWTEST_F(NativeBufferTest, OHNativeBufferGetColorSpace002, Function | MediumTest | Level2)
323 {
324     if (buffer == nullptr) {
325         buffer = OH_NativeBuffer_Alloc(&config);
326         ASSERT_NE(buffer, nullptr);
327     }
328     OH_NativeBuffer_ColorSpace colorSpace = OH_COLORSPACE_NONE;
329     int32_t ret = OH_NativeBuffer_SetColorSpace(buffer, OH_COLORSPACE_BT709_LIMIT);
330     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
331         ASSERT_EQ(ret, GSERROR_OK);
332     }
333     ret = OH_NativeBuffer_GetColorSpace(buffer, &colorSpace);
334     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
335         ASSERT_EQ(colorSpace, OH_COLORSPACE_BT709_LIMIT);
336         ASSERT_EQ(ret, GSERROR_OK);
337     }
338 }
339 
340 /*
341 * Function: OH_NativeBuffer_GetColorSpace
342 * Type: Function
343 * Rank: Important(2)
344 * EnvConditions: N/A
345 * CaseDescription: 1. call OH_NativeBuffer_GetColorSpace
346 *                  2. check ret
347 */
348 HWTEST_F(NativeBufferTest, OHNativeBufferGetColorSpace003, Function | MediumTest | Level2)
349 {
350     if (buffer == nullptr) {
351         buffer = OH_NativeBuffer_Alloc(&config);
352         ASSERT_NE(buffer, nullptr);
353     }
354     OH_NativeBuffer_ColorSpace colorSpace = OH_COLORSPACE_NONE;
355     int32_t ret = OH_NativeBuffer_GetColorSpace(buffer, &colorSpace);
356     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
357         ASSERT_EQ(ret, GSERROR_OK);
358     }
359     ret = OH_NativeBuffer_SetColorSpace(buffer, OH_COLORSPACE_BT709_LIMIT);
360     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
361         ASSERT_EQ(colorSpace, OH_COLORSPACE_BT709_LIMIT);
362         ASSERT_EQ(ret, GSERROR_OK);
363     }
364 }
365 
366 /*
367 * Function: OH_NativeBuffer_SetMetadataValue
368 * Type: Function
369 * Rank: Important(2)
370 * EnvConditions: N/A
371 * CaseDescription: 1. call OH_NativeBuffer_SetMetadataValue by abnormal input
372 *                  2. check ret
373 */
374 HWTEST_F(NativeBufferTest, OH_NativeBuffer_SetMetadataValue001, Function | MediumTest | Level2)
375 {
376     int32_t size = 1024;
377     uint8_t buff[size];
378     int32_t ret = OH_NativeBuffer_SetMetadataValue(nullptr, OH_HDR_STATIC_METADATA, size, buff);
379     ASSERT_NE(ret, GSERROR_OK);
380 }
381 
382 /*
383 * Function: OH_NativeBuffer_SetMetadataValue
384 * Type: Function
385 * Rank: Important(2)
386 * EnvConditions: N/A
387 * CaseDescription: 1. call OH_NativeBuffer_SetMetadataValue
388 *                  2. check ret
389 */
390 HWTEST_F(NativeBufferTest, OH_NativeBuffer_SetMetadataValue002, Function | MediumTest | Level2)
391 {
392     if (buffer == nullptr) {
393         buffer = OH_NativeBuffer_Alloc(&config);
394         ASSERT_NE(buffer, nullptr);
395     }
396     int32_t len = 60;
397     uint8_t outbuff[len];
398     for (int i = 0; i < 60; ++i) {
399         outbuff[i] = static_cast<uint8_t>(i);
400     }
401     int32_t ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_STATIC_METADATA, len, outbuff);
402     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
403         ASSERT_EQ(ret, GSERROR_OK);
404     }
405     ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_DYNAMIC_METADATA, len, outbuff);
406     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
407         ASSERT_EQ(ret, GSERROR_OK);
408     }
409     uint8_t type = OH_NativeBuffer_MetadataType::OH_VIDEO_HDR_HLG;
410     ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_METADATA_TYPE, sizeof(OH_NativeBuffer_MetadataType), &type);
411     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
412         ASSERT_EQ(ret, GSERROR_OK);
413     }
414     type = OH_NativeBuffer_MetadataType::OH_VIDEO_NONE;
415     ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_METADATA_TYPE, sizeof(OH_NativeBuffer_MetadataType), &type);
416     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
417         ASSERT_EQ(ret, GSERROR_INVALID_ARGUMENTS);
418     }
419     ret = OH_NativeBuffer_SetMetadataValue(buffer, static_cast<OH_NativeBuffer_MetadataKey>(-1), len, outbuff);
420     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
421         ASSERT_EQ(ret, GSERROR_INTERNAL);
422     }
423 }
424 
425 /*
426 * Function: OH_NativeBuffer_SetMetadataValue
427 * Type: Function
428 * Rank: Important(2)
429 * EnvConditions: N/A
430 * CaseDescription: 1. call OH_NativeBuffer_SetMetadataValue by abnormal input
431 *                  2. check ret
432 */
433 HWTEST_F(NativeBufferTest, OH_NativeBuffer_SetMetadataValue003, Function | MediumTest | Level2)
434 {
435     int32_t max_size = -1;
436     int32_t size = 60;
437     uint8_t buff[size];
438     int32_t ret = OH_NativeBuffer_SetMetadataValue(nullptr, OH_HDR_STATIC_METADATA, max_size, buff);
439     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
440         ASSERT_NE(ret, GSERROR_OK);
441     }
442 }
443 
444 /*
445 * Function: OH_NativeBuffer_GetMetadataValue
446 * Type: Function
447 * Rank: Important(2)
448 * EnvConditions: N/A
449 * CaseDescription: 1. call OH_NativeBuffer_GetMetadataValue by abnormal input
450 *                  2. check ret
451 */
452 HWTEST_F(NativeBufferTest, OH_NativeBuffer_GetMetadataValue001, Function | MediumTest | Level2)
453 {
454     int32_t size = 1024;
455     uint8_t *buff;
456     int32_t ret = OH_NativeBuffer_GetMetadataValue(nullptr, OH_HDR_STATIC_METADATA, &size, &buff);
457     if (buff != nullptr) {
458         delete[] buff;
459     }
460     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
461         ASSERT_NE(ret, GSERROR_OK);
462     }
463 }
464 
465 /*
466 * Function: OH_NativeBuffer_GetMetadataValue
467 * Type: Function
468 * Rank: Important(2)
469 * EnvConditions: N/A
470 * CaseDescription: 1. call OH_NativeBuffer_GetMetadataValue
471 *                  2. check ret
472 */
473 HWTEST_F(NativeBufferTest, OH_NativeBuffer_GetMetadataValue002, Function | MediumTest | Level2)
474 {
475     if (buffer == nullptr) {
476         buffer = OH_NativeBuffer_Alloc(&config);
477         ASSERT_NE(buffer, nullptr);
478     }
479     int32_t len = 60;
480     uint8_t outbuff[len];
481     for (int i = 0; i < 60; ++i) {
482         outbuff[i] = static_cast<uint8_t>(i);
483     }
484     int32_t ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_STATIC_METADATA, len, outbuff);
485     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
486         ASSERT_EQ(ret, GSERROR_OK);
487     }
488     int32_t buffSize = 0;
489     uint8_t *buff;
490     ret = OH_NativeBuffer_GetMetadataValue(buffer, OH_HDR_STATIC_METADATA, &buffSize, &buff);
491     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
492         if (buff != nullptr) {
493             ASSERT_EQ(memcmp(outbuff, buff, 60), 0);
494             delete[] buff;
495         }
496         ASSERT_EQ(ret, GSERROR_OK);
497     }
498     ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_DYNAMIC_METADATA, len, outbuff);
499     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
500         ASSERT_EQ(ret, GSERROR_OK);
501     }
502     ret = OH_NativeBuffer_GetMetadataValue(buffer, OH_HDR_DYNAMIC_METADATA, &buffSize, &buff);
503     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
504         if (buff != nullptr) {
505             ASSERT_EQ(memcmp(outbuff, buff, 60), 0);
506             delete[] buff;
507         }
508         ASSERT_EQ(ret, GSERROR_OK);
509     }
510     ret = OH_NativeBuffer_GetMetadataValue(buffer, static_cast<OH_NativeBuffer_MetadataKey>(-1), &buffSize, &buff);
511     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
512         ASSERT_EQ(ret, GSERROR_INTERNAL);
513     }
514 }
515 
516 /*
517 * Function: OH_NativeBuffer_GetMetadataValue
518 * Type: Function
519 * Rank: Important(2)
520 * EnvConditions: N/A
521 * CaseDescription: 1. call OH_NativeBuffer_GetMetadataValue
522 *                  2. check ret
523 */
524 HWTEST_F(NativeBufferTest, OH_NativeBuffer_GetMetadataValue003, Function | MediumTest | Level2)
525 {
526     if (buffer == nullptr) {
527         buffer = OH_NativeBuffer_Alloc(&config);
528         ASSERT_NE(buffer, nullptr);
529     }
530     uint8_t *buff;
531     int32_t ret = OH_NativeBuffer_GetMetadataValue(buffer, OH_HDR_STATIC_METADATA, nullptr, &buff);
532     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
533         ASSERT_NE(ret, GSERROR_OK);
534     }
535     uint8_t type = OH_NativeBuffer_MetadataType::OH_VIDEO_HDR_HLG;
536     uint8_t *resType = new uint8_t;
537     int32_t typeSize = sizeof(OH_NativeBuffer_MetadataType);
538     ret = OH_NativeBuffer_SetMetadataValue(buffer, OH_HDR_METADATA_TYPE, typeSize, &type);
539     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
540         ASSERT_EQ(ret, GSERROR_OK);
541     }
542     ret = OH_NativeBuffer_GetMetadataValue(buffer, OH_HDR_METADATA_TYPE, &typeSize, &resType);
543     if (ret != GSERROR_NOT_SUPPORT) { // some device not support set colorspace
544         ASSERT_EQ(type, *resType);
545         ASSERT_EQ(ret, GSERROR_OK);
546     }
547     delete resType;
548     resType = nullptr;
549 }
550 
551 /*
552 * Function: OH_NativeBuffer_Map
553 * Type: Function
554 * Rank: Important(2)
555 * EnvConditions: N/A
556 * CaseDescription: 1. call OH_NativeBuffer_Map by abnormal input
557 *                  2. check ret
558 */
559 HWTEST_F(NativeBufferTest, OHNativeBufferMap001, Function | MediumTest | Level2)
560 {
561     void *virAddr = nullptr;
562     int32_t ret = OH_NativeBuffer_Map(nullptr, &virAddr);
563     ASSERT_NE(ret, GSERROR_OK);
564 }
565 
566 /*
567 * Function: OH_NativeBuffer_Map
568 * Type: Function
569 * Rank: Important(2)
570 * EnvConditions: N/A
571 * CaseDescription: 1. call OH_NativeBuffer_Map
572 *                  2. check ret
573 */
574 HWTEST_F(NativeBufferTest, OHNativeBufferMap002, Function | MediumTest | Level2)
575 {
576     void *virAddr = nullptr;
577     int32_t ret = OH_NativeBuffer_Map(buffer, &virAddr);
578     ASSERT_EQ(ret, GSERROR_OK);
579     ASSERT_NE(virAddr, nullptr);
580 }
581 
582 /*
583 * Function: OH_NativeBuffer_Unmap
584 * Type: Function
585 * Rank: Important(2)
586 * EnvConditions: N/A
587 * CaseDescription: 1. call OH_NativeBuffer_Unmap by abnormal input
588 *                  2. check ret
589 */
590 HWTEST_F(NativeBufferTest, OHNativeBufferUnmap001, Function | MediumTest | Level2)
591 {
592     int32_t ret = OH_NativeBuffer_Unmap(nullptr);
593     ASSERT_NE(ret, GSERROR_OK);
594 }
595 
596 /*
597 * Function: OH_NativeBuffer_Unmap and OH_NativeBuffer_Unreference
598 * Type: Function
599 * Rank: Important(2)
600 * EnvConditions: N/A
601 * CaseDescription: 1. call OH_NativeBuffer_Unmap
602 *                  2. check ret
603 *                  3. call OH_NativeBuffer_Unreference
604 */
605 HWTEST_F(NativeBufferTest, OHNativeBufferUnmap002, Function | MediumTest | Level2)
606 {
607     int32_t ret = OH_NativeBuffer_Unmap(buffer);
608     ASSERT_EQ(ret, GSERROR_OK);
609     ret = OH_NativeBuffer_Unreference(buffer);
610     ASSERT_EQ(ret, GSERROR_OK);
611 }
612 
613 /*
614 * Function: OH_NativeBufferFromNativeWindowBuffer
615 * Type: Function
616 * Rank: Important(2)
617 * EnvConditions: N/A
618 * CaseDescription: 1. call OH_NativeBufferFromNativeWindowBuffer by abnormal input
619 *                  2. check ret
620 */
621 HWTEST_F(NativeBufferTest, NativeBufferFromNativeWindowBuffer001, Function | MediumTest | Level2)
622 {
623     OH_NativeBuffer* nativeBuffer = OH_NativeBufferFromNativeWindowBuffer(nullptr);
624     ASSERT_EQ(nativeBuffer, nullptr);
625 }
626 
627 /*
628 * Function: OH_NativeBufferFromNativeWindowBuffer
629 * Type: Function
630 * Rank: Important(2)
631 * EnvConditions: N/A
632 * CaseDescription: 1. call OH_NativeBufferFromNativeWindowBuffer
633 *                  2. check ret
634 */
635 HWTEST_F(NativeBufferTest, NativeBufferFromNativeWindowBuffer002, Function | MediumTest | Level2)
636 {
637     sptr<OHOS::IConsumerSurface> cSurface = IConsumerSurface::Create();
638     sptr<IBufferConsumerListener> listener = new BufferConsumerListener();
639     cSurface->RegisterConsumerListener(listener);
640     sptr<OHOS::IBufferProducer> producer = cSurface->GetProducer();
641     sptr<OHOS::Surface> pSurface = Surface::CreateSurfaceAsProducer(producer);
642     int32_t fence;
643     sptr<OHOS::SurfaceBuffer> sBuffer = nullptr;
644     BufferRequestConfig requestConfig = {
645         .width = 0x100,  // small
646         .height = 0x100, // small
647         .strideAlignment = 0x8,
648         .format = GRAPHIC_PIXEL_FMT_RGBA_8888,
649         .usage = BUFFER_USAGE_CPU_READ | BUFFER_USAGE_CPU_WRITE | BUFFER_USAGE_MEM_DMA,
650         .timeout = 0,
651     };
652     pSurface->RequestBuffer(sBuffer, fence, requestConfig);
653     NativeWindow* nativeWindow = OH_NativeWindow_CreateNativeWindow(&pSurface);
654     ASSERT_NE(nativeWindow, nullptr);
655     NativeWindowBuffer* nativeWindowBuffer = OH_NativeWindow_CreateNativeWindowBufferFromSurfaceBuffer(&sBuffer);
656     ASSERT_NE(nativeWindowBuffer, nullptr);
657     OH_NativeBuffer* nativeBuffer = OH_NativeBufferFromNativeWindowBuffer(nativeWindowBuffer);
658     ASSERT_NE(nativeBuffer, nullptr);
659 
660     int32_t ret = OH_NativeBuffer_FromNativeWindowBuffer(nativeWindowBuffer, nullptr);
661     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
662     OH_NativeBuffer* nativeBufferTmp = nullptr;
663     ret = OH_NativeBuffer_FromNativeWindowBuffer(nativeWindowBuffer, &nativeBufferTmp);
664     ASSERT_EQ(ret, OHOS::GSERROR_OK);
665     ASSERT_EQ(nativeBuffer, nativeBufferTmp);
666 
667     void *virAddr = nullptr;
668     OH_NativeBuffer_Planes outPlanes;
669     ret = OH_NativeBuffer_MapPlanes(nativeBuffer, &virAddr, &outPlanes);
670     ASSERT_NE(ret, OHOS::GSERROR_OK);
671 
672     sBuffer = nullptr;
673     cSurface = nullptr;
674     producer = nullptr;
675     pSurface = nullptr;
676     nativeWindow = nullptr;
677     nativeWindowBuffer = nullptr;
678 }
679 
680 /*
681 * Function: OH_NativeBuffer_FromNativeWindowBuffer
682 * Type: Function
683 * Rank: Important(2)
684 * EnvConditions: N/A
685 * CaseDescription: 1. call OH_NativeBuffer_FromNativeWindowBuffer by abnormal input
686 *                  2. check ret
687 */
688 HWTEST_F(NativeBufferTest, NativeBufferFromNativeWindowBuffer003, Function | MediumTest | Level2)
689 {
690     int32_t ret = OH_NativeBuffer_FromNativeWindowBuffer(nullptr, nullptr);
691     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
692 
693     NativeWindowBuffer nativeWindowBuffer;
694     ret = OH_NativeBuffer_FromNativeWindowBuffer(&nativeWindowBuffer, nullptr);
695     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
696 
697     OH_NativeBuffer_GetNativeBufferConfig(nullptr, nullptr);
698     OH_NativeBuffer_GetNativeBufferConfig(buffer, nullptr);
699     ASSERT_EQ(OH_NativeBuffer_GetBufferHandle(nullptr), nullptr);
700 }
701 
702 /*
703 * Function: OH_NativeBuffer_MapPlanes
704 * Type: Function
705 * Rank: Important(2)
706 * EnvConditions: N/A
707 * CaseDescription: 1. call OH_NativeBuffer_MapPlanes by abnormal input
708 *                  2. check ret
709 */
710 HWTEST_F(NativeBufferTest, OHNativeBufferMapPlanes001, Function | MediumTest | Level2)
711 {
712     int32_t ret = OH_NativeBuffer_MapPlanes(nullptr, nullptr, nullptr);
713     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
714 
715     OH_NativeBuffer *buffer = (OH_NativeBuffer *)0xFFFFFFFF;
716     ret = OH_NativeBuffer_MapPlanes(buffer, nullptr, nullptr);
717     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
718 
719     void *virAddr = nullptr;
720     ret = OH_NativeBuffer_MapPlanes(buffer, &virAddr, nullptr);
721     ASSERT_EQ(ret, OHOS::GSERROR_INVALID_ARGUMENTS);
722 }
723 
724 /*
725 * Function: OH_NativeBuffer_MapPlanes
726 * Type: Function
727 * Rank: Important(2)
728 * EnvConditions: N/A
729 * CaseDescription: 1. call OH_NativeBuffer_MapPlanes
730 *                  2. check ret
731 */
732 HWTEST_F(NativeBufferTest, OHNativeBufferMapPlanes002, Function | MediumTest | Level2)
733 {
734     sptr<OHOS::IConsumerSurface> cSurface = IConsumerSurface::Create();
735     sptr<IBufferConsumerListener> listener = new BufferConsumerListener();
736     cSurface->RegisterConsumerListener(listener);
737     sptr<OHOS::IBufferProducer> producer = cSurface->GetProducer();
738     sptr<OHOS::Surface> pSurface = Surface::CreateSurfaceAsProducer(producer);
739     int32_t fence;
740     sptr<OHOS::SurfaceBuffer> sBuffer = nullptr;
741     BufferRequestConfig requestConfig = {.width = 0x100, .height = 0x100, .strideAlignment = 0x8,
742         .format = GRAPHIC_PIXEL_FMT_YCBCR_420_SP,
743         .usage = BUFFER_USAGE_CPU_READ | BUFFER_USAGE_CPU_WRITE | BUFFER_USAGE_MEM_DMA, .timeout = 0};
744     pSurface->SetQueueSize(4);
745     int32_t formatType[] = {GRAPHIC_PIXEL_FMT_YCBCR_420_SP, GRAPHIC_PIXEL_FMT_YCRCB_420_SP,
746         GRAPHIC_PIXEL_FMT_YCBCR_420_P, GRAPHIC_PIXEL_FMT_YCRCB_420_P};
747     NativeWindow* nativeWindow;
748     NativeWindowBuffer* nativeWindowBuffer;
749     for (int32_t i = 0; i < sizeof(formatType) / sizeof(int32_t); i++) {
750         requestConfig.format = formatType[i];
751         pSurface->RequestBuffer(sBuffer, fence, requestConfig);
752         nativeWindow = OH_NativeWindow_CreateNativeWindow(&pSurface);
753         ASSERT_NE(nativeWindow, nullptr);
754         nativeWindowBuffer = OH_NativeWindow_CreateNativeWindowBufferFromSurfaceBuffer(&sBuffer);
755         ASSERT_NE(nativeWindowBuffer, nullptr);
756         OH_NativeBuffer* nativeBuffer = OH_NativeBufferFromNativeWindowBuffer(nativeWindowBuffer);
757         ASSERT_NE(nativeBuffer, nullptr);
758 
759         void *virAddr = nullptr;
760         OH_NativeBuffer_Planes outPlanes;
761         int32_t ret = OH_NativeBuffer_MapPlanes(nativeBuffer, &virAddr, &outPlanes);
762         if (ret != GSERROR_HDI_ERROR) {
763             ASSERT_EQ(ret, OHOS::GSERROR_OK);
764             ASSERT_NE(virAddr, nullptr);
765             ASSERT_EQ(outPlanes.planeCount, 3);
766             ASSERT_EQ(outPlanes.planes[0].offset, 0);
767             ASSERT_NE(outPlanes.planes[1].offset, 0);
768             ASSERT_NE(outPlanes.planes[2].offset, 0);
769             std::cout << "i:" << i << std::endl;
770             for (int32_t j = 0; j < 3; j++) {
771                 std::cout << "offset:" << outPlanes.planes[j].offset <<
772                     " rowStride:" << outPlanes.planes[j].rowStride << " columnStride:" <<
773                     outPlanes.planes[j].columnStride << std::endl;
774             }
775         }
776     }
777 
778     sBuffer = nullptr;
779     cSurface = nullptr;
780     producer = nullptr;
781     pSurface = nullptr;
782     nativeWindow = nullptr;
783     nativeWindowBuffer = nullptr;
784 }
785 
786 /*
787 * Function: OH_NativeBuffer_MapPlanes
788 * Type: Function
789 * Rank: Important(2)
790 * EnvConditions: N/A
791 * CaseDescription: 1. call OH_NativeBuffer_MapPlanes
792 *                  2. check ret
793 */
794 HWTEST_F(NativeBufferTest, OHNativeBufferMapPlanes003, Function | MediumTest | Level2)
795 {
796     sptr<OHOS::IConsumerSurface> cSurface = IConsumerSurface::Create();
797     sptr<IBufferConsumerListener> listener = new BufferConsumerListener();
798     cSurface->RegisterConsumerListener(listener);
799     sptr<OHOS::IBufferProducer> producer = cSurface->GetProducer();
800     sptr<OHOS::Surface> pSurface = Surface::CreateSurfaceAsProducer(producer);
801     int32_t fence;
802     sptr<OHOS::SurfaceBuffer> sBuffer = nullptr;
803     BufferRequestConfig requestConfig = {.width = 0x100, .height = 0x100, .strideAlignment = 0x8,
804         .format = GRAPHIC_PIXEL_FMT_YCBCR_420_SP,
805         .usage = BUFFER_USAGE_CPU_READ | BUFFER_USAGE_CPU_WRITE | BUFFER_USAGE_MEM_DMA, .timeout = 0};
806     pSurface->SetQueueSize(4);
807     int32_t formatType = GRAPHIC_PIXEL_FMT_YCBCR_420_SP;
808     NativeWindow* nativeWindow;
809     NativeWindowBuffer* nativeWindowBuffer;
810     requestConfig.format = formatType;
811     pSurface->RequestBuffer(sBuffer, fence, requestConfig);
812     nativeWindow = OH_NativeWindow_CreateNativeWindow(&pSurface);
813     ASSERT_NE(nativeWindow, nullptr);
814     nativeWindowBuffer = OH_NativeWindow_CreateNativeWindowBufferFromSurfaceBuffer(&sBuffer);
815     ASSERT_NE(nativeWindowBuffer, nullptr);
816     OH_NativeBuffer* nativeBuffer = OH_NativeBufferFromNativeWindowBuffer(nativeWindowBuffer);
817     ASSERT_NE(nativeBuffer, nullptr);
818     OH_NativeBuffer* nativeBufferTmp = nullptr;
819     for (int32_t i = 0; i < 1000; i++) {
820         int32_t ret = OH_NativeBuffer_FromNativeWindowBuffer(nativeWindowBuffer, &nativeBufferTmp);
821         ASSERT_EQ(ret, OHOS::GSERROR_OK);
822         ASSERT_EQ(nativeBuffer, nativeBufferTmp);
823     }
824     void *virAddr = nullptr;
825     OH_NativeBuffer_Planes outPlanes;
826     clock_t startTime, endTime;
827     startTime = clock();
828     for (int32_t i = 0; i < 1000; i++) {
829         int32_t ret = OH_NativeBuffer_MapPlanes(nativeBuffer, &virAddr, &outPlanes);
830         if (ret != GSERROR_HDI_ERROR) {
831             ASSERT_EQ(ret, OHOS::GSERROR_OK);
832             ASSERT_NE(virAddr, nullptr);
833             ASSERT_EQ(outPlanes.planeCount, 3);
834             ASSERT_EQ(outPlanes.planes[0].offset, 0);
835             ASSERT_NE(outPlanes.planes[1].offset, 0);
836             ASSERT_NE(outPlanes.planes[2].offset, 0);
837         }
838     }
839     endTime = clock();
840     cout << "OH_NativeBuffer_MapPlanes 1000 times cost time: " << (endTime - startTime) << "ms" << endl;
841     sBuffer = nullptr;
842     cSurface = nullptr;
843     producer = nullptr;
844     pSurface = nullptr;
845     nativeWindow = nullptr;
846     nativeWindowBuffer = nullptr;
847 }
848 }