Browse Source

fix: 修复 EnergyControllerTest 编译错误

iwt 1 month ago
parent
commit
c38f1a8710

+ 4 - 4
cfc-backend/src/test/java/com/etotem/cfc/controller/EnergyControllerTest.java

@@ -59,7 +59,7 @@ public class EnergyControllerTest {
         when(userMapper.selectById(userId)).thenReturn(user);
         when(energyService.calculateFamilyEnergy(100L)).thenReturn(mockDto);
 
-        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(userId);
+        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(userId, null);
 
         assertEquals(200, result.getCode());
         assertNotNull(result.getData());
@@ -76,7 +76,7 @@ public class EnergyControllerTest {
     public void getFamilyEnergySandbox_UserNotFound() {
         when(userMapper.selectById(999L)).thenReturn(null);
 
-        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(999L);
+        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(999L, null);
 
         assertEquals(500, result.getCode());
         assertEquals("用户不存在", result.getMessage());
@@ -90,7 +90,7 @@ public class EnergyControllerTest {
 
         when(userMapper.selectById(1L)).thenReturn(user);
 
-        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(1L);
+        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(1L, null);
 
         assertEquals(500, result.getCode());
         assertEquals("用户未加入家庭", result.getMessage());
@@ -114,7 +114,7 @@ public class EnergyControllerTest {
         when(userMapper.selectById(userId)).thenReturn(user);
         when(energyService.calculateFamilyEnergy(200L)).thenReturn(mockDto);
 
-        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(userId);
+        Result<EnergySandboxDTO> result = controller.getFamilyEnergySandbox(userId, null);
 
         assertEquals(200, result.getCode());
         assertEquals(0, result.getData().getOverallScore());