diff --git a/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.cpp b/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.cpp
index 8de6b6a506e64d7ef5f9b5120ab0c50be5b4ea27..af9d50889729ed5aab3867666bb4f77595dd5041 100644
--- a/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.cpp
+++ b/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.cpp
@@ -310,7 +310,7 @@ HWTEST_F(AVThumbnailGeneratorUnitTest, GenerateAlignmentAvBuffer, TestSize.Level
     ASSERT_NE(avThumbnailGenerator_, nullptr);
     auto mockSurfaceBuffer = new MockSurfaceBuffer();
     EXPECT_CALL(*(mockSurfaceBuffer), GetWidth()).WillRepeatedly(Return(NUM_TEST));
-    EXPECT_CALL(*(mockSurfaceBuffer), Alloc(_)).WillRepeatedly(Return(GSError::GSERROR_OK));
+    EXPECT_CALL(*(mockSurfaceBuffer), Alloc(_, _)).WillRepeatedly(Return(GSError::GSERROR_OK));
     EXPECT_CALL(*(mockSurfaceBuffer), GetHeight()).WillRepeatedly(Return(NUM_TEST));
     EXPECT_CALL(*(mockSurfaceBuffer), GetFormat()).
         WillRepeatedly(Return(static_cast<int32_t>(GraphicPixelFormat::GRAPHIC_PIXEL_FMT_YCBCR_P010)));
diff --git a/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.h b/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.h
index 8eb96002eed91c436db96ee94b69d1919fbb6898..1a85ec480c4b4878ff8c2233159e8037510aabee 100644
--- a/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.h
+++ b/test/unittest/avmetadata_helper_unittest/av_thumbnail_generator_unittest.h
@@ -84,7 +84,8 @@ public:
     MOCK_METHOD(void, SetBufferHandle, (BufferHandle *handle), (override));
     MOCK_METHOD(BufferHandle*, GetBufferHandle, (), (const, override));
     MOCK_METHOD(void*, GetVirAddr, (), (override));
-    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config), (override));
+    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config, const sptr<SurfaceBuffer>& previousBuffer),
+        (override));
     MOCK_METHOD(GSError, Map, (), (override));
     MOCK_METHOD(GSError, Unmap, (), (override));
     MOCK_METHOD(GSError, FlushCache, (), (override));
diff --git a/test/unittest/avmetadatahelper_impl_unittest/mock/mock_avbuffer.h b/test/unittest/avmetadatahelper_impl_unittest/mock/mock_avbuffer.h
index a425eb99932bc08d47b7a51f5f19814790c9e0fa..e4c43c0b3e8706758d0347f13d58d23748b9db94 100644
--- a/test/unittest/avmetadatahelper_impl_unittest/mock/mock_avbuffer.h
+++ b/test/unittest/avmetadatahelper_impl_unittest/mock/mock_avbuffer.h
@@ -128,7 +128,8 @@ public:
     MOCK_METHOD(void, SetBufferHandle, (BufferHandle *handle), (override));
     MOCK_METHOD(BufferHandle*, GetBufferHandle, (), (const, override));
     MOCK_METHOD(void*, GetVirAddr, (), (override));
-    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config), (override));
+    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config, const sptr<SurfaceBuffer>& previousBuffer),
+        (override));
     MOCK_METHOD(GSError, Map, (), (override));
     MOCK_METHOD(GSError, Unmap, (), (override));
     MOCK_METHOD(GSError, FlushCache, (), (override));
diff --git a/test/unittest/native_avmetadatahelper_unittest/include/avmetadata_unittest.h b/test/unittest/native_avmetadatahelper_unittest/include/avmetadata_unittest.h
index d346eb596f8264a80041bf7354f0d024a2e7cf44..69adf2319749bb693969df96ae1fee045c056f45 100644
--- a/test/unittest/native_avmetadatahelper_unittest/include/avmetadata_unittest.h
+++ b/test/unittest/native_avmetadatahelper_unittest/include/avmetadata_unittest.h
@@ -82,7 +82,8 @@ public:
     MOCK_METHOD(void, SetBufferHandle, (BufferHandle *handle), (override));
     MOCK_METHOD(BufferHandle*, GetBufferHandle, (), (const, override));
     MOCK_METHOD(void*, GetVirAddr, (), (override));
-    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config), (override));
+    MOCK_METHOD(GSError, Alloc, (const BufferRequestConfig &config, const sptr<SurfaceBuffer>& previousBuffer),
+        (override));
     MOCK_METHOD(GSError, Map, (), (override));
     MOCK_METHOD(GSError, Unmap, (), (override));
     MOCK_METHOD(GSError, FlushCache, (), (override));