From 330c99ca57bc65417b486fbb72d7569a4d44ae2a Mon Sep 17 00:00:00 2001 From: Pearl Dsilva Date: Fri, 19 Jan 2024 12:53:23 -0500 Subject: [PATCH] fix test failure --- .../java/org/apache/cloudstack/service/NsxElementTest.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxElementTest.java b/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxElementTest.java index ae3143e32b8..39fb2e06120 100644 --- a/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxElementTest.java +++ b/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxElementTest.java @@ -273,8 +273,13 @@ public class NsxElementTest { 5L, 2L, 15L); rule.setState(FirewallRule.State.Revoke); Network.Service service = new Network.Service("service1", new Network.Capability("capability")); - + VpcVO vpcVO = Mockito.mock(VpcVO.class); + when(vpcDao.findById(1L)).thenReturn(vpcVO); + when(vpcVO.getDomainId()).thenReturn(2L); + IPAddressVO ipAddress = new IPAddressVO(new Ip("10.1.13.10"), 1L, 1L, 1L,false); + when(ApiDBUtils.findIpAddressById(anyLong())).thenReturn(ipAddress); when(nsxElement.canHandle(networkVO, service)).thenReturn(true); + when(nsxService.deletePortForwardRule(any(NsxNetworkRule.class))).thenReturn(true); assertTrue(nsxElement.applyPFRules(networkVO, List.of(rule))); }