|
|
@@ -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());
|