test: Fix populateIpAddressUsageResponseTest after Copilot refactor

This commit is contained in:
Prashant Bhanage 2026-07-05 19:47:36 +05:30
parent c407a0c995
commit 6c3b5972aa
1 changed files with 2 additions and 5 deletions

View File

@ -962,20 +962,17 @@ public class ApiResponseHelperTest {
when(entityManagerMock.findByIdIncludingRemoved(IPAddressVO.class, usageId.toString())).thenReturn(ipAddress);
when(ipAddress.getUuid()).thenReturn("ip-address-uuid");
when(ipAddress.getId()).thenReturn(usageId);
when(ipAddress.getAssociatedWithNetworkId()).thenReturn(null);
when(ipAddress.getSourceNetworkId()).thenReturn(23L);
// Act
Object resourceDetails = invokeUsageDetailsHelper("populateIpAddressUsageResponse",
new Class<?>[] {Usage.class, UsageRecordResponse.class}, usageRecord, response);
// Assert
// Assert
assertResponseField(response, "usageId", "ip-address-uuid");
assertResponseField(response, "isSourceNat", Boolean.TRUE);
assertResponseField(response, "isSystem", Boolean.TRUE);
assertUsageResourceDetails(resourceDetails, ResourceTag.ResourceObjectType.PublicIpAddress, usageId);
verify(entityManagerMock).findByIdIncludingRemoved(IPAddressVO.class, usageId.toString());
verify(ipAddress).getSourceNetworkId();
}
@Test
@ -1469,4 +1466,4 @@ public class ApiResponseHelperTest {
assertDescriptionContains(response, "volume " + volumeName + " (" + volumeUuid + ")");
assertUsageResourceDetails(resourceDetails, ResourceTag.ResourceObjectType.Volume, volumeId);
}
}
}