From 60034e85fe5f95fed2afbaedbb6065ee15ed72f5 Mon Sep 17 00:00:00 2001
From: chenbowen <you@example.com>
Date: Wed, 16 Jul 2025 18:31:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=B5=8B=E8=AF=95=E7=94=A8?=
 =?UTF-8?q?=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module/infra/service/codegen/CodegenServiceImplTest.java  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
index 8ca4a0e3..c0293306 100644
--- a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
+++ b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
@@ -460,7 +460,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
             assertEquals(column01, columns.get(0));
             assertEquals(column02, columns.get(1));
             return true;
-        }), isNull(), isNull())).thenReturn(codes);
+        }), isNull(), isNull(), eq(false))).thenReturn(codes);
         // 准备参数
         Long tableId = table.getId();
 
@@ -509,7 +509,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
             assertEquals(1, columns.size());
             assertPojoEquals(subColumn01, columns.size());
             return true;
-        }))).thenReturn(codes);
+        }), eq(false))).thenReturn(codes);
         // 准备参数
         Long tableId = table.getId();