From 66a172863e1cff3869470399f5fe26a3157a49a5 Mon Sep 17 00:00:00 2001
From: zhaona45 <zhaona45@huawei.com>
Date: Thu, 14 Aug 2025 10:30:21 +0800
Subject: [PATCH] Add Auxiliary Picture Tdd.

Signed-off-by: zhaona45 <zhaona45@huawei.com>
---
 .../auxiliary_picture_test.cpp                | 57 +++++++++++++++++++
 1 file changed, 57 insertions(+)

diff --git a/frameworks/innerkitsimpl/test/unittest/auxiliary_picture_test/auxiliary_picture_test.cpp b/frameworks/innerkitsimpl/test/unittest/auxiliary_picture_test/auxiliary_picture_test.cpp
index 924747342..a1be0ad5d 100644
--- a/frameworks/innerkitsimpl/test/unittest/auxiliary_picture_test/auxiliary_picture_test.cpp
+++ b/frameworks/innerkitsimpl/test/unittest/auxiliary_picture_test/auxiliary_picture_test.cpp
@@ -497,6 +497,63 @@ HWTEST_F(AuxiliaryPictureTest, WritePixelsTest002, TestSize.Level2)
     EXPECT_EQ(ret, ERR_IMAGE_INVALID_PARAMETER);
 }
 
+/**
+ * @tc.name: WritePixelsTest003
+ * @tc.desc: Write buffer to pixels with invalid buffer size.
+ * @tc.type: FUNC
+ */
+HWTEST_F(AuxiliaryPictureTest, WritePixelsTest003, TestSize.Level2)
+{
+    std::unique_ptr<AuxiliaryPicture> auxiliaryPicture = CreateAuxiliaryPicture(AuxiliaryPictureType::GAINMAP);
+    ASSERT_NE(auxiliaryPicture, nullptr);
+    auxiliaryPicture->GetContentPixel()->SetEditable(true);
+    uint64_t bufferSize = auxiliaryPicture->GetContentPixel()->GetCapacity();
+    auto buffer = new uint8_t[bufferSize];
+    ASSERT_NE(buffer, nullptr);
+    bufferSize = 0;
+    uint32_t ret = auxiliaryPicture->WritePixels(buffer, bufferSize);
+    delete[] buffer;
+    EXPECT_EQ(ret, ERR_IMAGE_INVALID_PARAMETER);
+}
+
+/**
+ * @tc.name: WritePixelsTest004
+ * @tc.desc: Write buffer to pixels but the pixelmap in auxiliary picture is nullptr.
+ * @tc.type: FUNC
+ */
+HWTEST_F(AuxiliaryPictureTest, WritePixelsTest004, TestSize.Level2)
+{
+    std::unique_ptr<AuxiliaryPicture> auxiliaryPicture = CreateAuxiliaryPicture(AuxiliaryPictureType::GAINMAP);
+    ASSERT_NE(auxiliaryPicture, nullptr);
+    auxiliaryPicture->GetContentPixel()->SetEditable(true);
+    uint64_t bufferSize = auxiliaryPicture->GetContentPixel()->GetCapacity();
+    auto buffer = new uint8_t[bufferSize];
+    ASSERT_NE(buffer, nullptr);
+    std::shared_ptr<PixelMap> emptyPixelmap = nullptr;
+    auxiliaryPicture->SetContentPixel(emptyPixelmap);
+    uint32_t ret = auxiliaryPicture->WritePixels(buffer, bufferSize);
+    delete[] buffer;
+    EXPECT_EQ(ret, ERR_MEDIA_NULL_POINTER);
+}
+
+/**
+ * @tc.name: WritePixelsTest005
+ * @tc.desc: Write buffer to pixels but the pixelmap in auxiliary picture is not editable.
+ * @tc.type: FUNC
+ */
+HWTEST_F(AuxiliaryPictureTest, WritePixelsTest005, TestSize.Level2)
+{
+    std::unique_ptr<AuxiliaryPicture> auxiliaryPicture = CreateAuxiliaryPicture(AuxiliaryPictureType::GAINMAP);
+    ASSERT_NE(auxiliaryPicture, nullptr);
+    auxiliaryPicture->GetContentPixel()->SetEditable(false);
+    uint64_t bufferSize = auxiliaryPicture->GetContentPixel()->GetCapacity();
+    auto buffer = new uint8_t[bufferSize];
+    ASSERT_NE(buffer, nullptr);
+    uint32_t ret = auxiliaryPicture->WritePixels(buffer, bufferSize);
+    delete[] buffer;
+    EXPECT_EQ(ret, ERR_IMAGE_PIXELMAP_NOT_ALLOW_MODIFY);
+}
+
 /**
  * @tc.name: WritePixelsTest006
  * @tc.desc: Write buffer to pixels with invalid image info for pixelmap in auxiliary picture.
-- 
Gitee