From 24991d9e86566d43cf6d872c594aa179a137c9f9 Mon Sep 17 00:00:00 2001
From: dingpy <dingpanyun1@huawei.com>
Date: Tue, 29 Jul 2025 02:25:53 +0000
Subject: [PATCH] =?UTF-8?q?ION=E5=86=85=E5=AD=98=E7=BB=B4=E6=B5=8B?=
 =?UTF-8?q?=E9=9C=80=E6=B1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: dingpy <dingpanyun1@huawei.com>
---
 test/unittest/mock/include/mock_producer_surface.h |  4 ++--
 test/unittest/mock/src/mock_producer_surface.cpp   | 12 ++++++------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/test/unittest/mock/include/mock_producer_surface.h b/test/unittest/mock/include/mock_producer_surface.h
index 51a0c2e..11fc973 100644
--- a/test/unittest/mock/include/mock_producer_surface.h
+++ b/test/unittest/mock/include/mock_producer_surface.h
@@ -27,8 +27,8 @@ class MockProducerSurface : public ProducerSurface {
 public:
     MockProducerSurface(sptr<IBufferProducer>& producer);
     ~MockProducerSurface();
-    MOCK_METHOD3(RequestBuffer, GSError(sptr<SurfaceBuffer>& buffer, sptr<SyncFence>& fence,
-        BufferRequestConfig &config));
+    MOCK_METHOD4(RequestBuffer, GSError(sptr<SurfaceBuffer>& buffer, sptr<SyncFence>& fence,
+        BufferRequestConfig &config, bool isSdkFlag));
     MOCK_METHOD3(FlushBuffer, GSError(sptr<SurfaceBuffer>& buffer, int32_t fence, BufferFlushConfig &config));
 
     static void AllocDmaMemory(sptr<SurfaceBuffer> &buffer);
diff --git a/test/unittest/mock/src/mock_producer_surface.cpp b/test/unittest/mock/src/mock_producer_surface.cpp
index 2ab69ab..75ddb74 100644
--- a/test/unittest/mock/src/mock_producer_surface.cpp
+++ b/test/unittest/mock/src/mock_producer_surface.cpp
@@ -57,12 +57,12 @@ void MockProducerSurface::ReleaseDmaBuffer(sptr<SurfaceBuffer> &buffer)
 MockProducerSurface::MockProducerSurface(sptr<IBufferProducer>& producer) : ProducerSurface(producer)
 {
     AllocDmaMemory(buffer_);
-    ON_CALL(*this, RequestBuffer(_, _, _))
-        .WillByDefault([this](sptr<SurfaceBuffer> &buffer, sptr<SyncFence> &fence, BufferRequestConfig &config) {
-            buffer = buffer_;
-            return GSError::GSERROR_OK;
-        }
-    );
+    ON_CALL(*this, RequestBuffer(_, _, _, _))
+        .WillByDefault([this](sptr<SurfaceBuffer> &buffer, sptr<SyncFence> &fence, BufferRequestConfig &config,
+            bool isSdkFlag) {
+                buffer = buffer_;
+                return GSError::GSERROR_OK;
+            });
     ON_CALL(*this, FlushBuffer(_, _, _)).WillByDefault(Return(GSError::GSERROR_OK));
 }
 
-- 
Gitee