From 6fcdf2092fd40bb335a3cbfa155bb51032498731 Mon Sep 17 00:00:00 2001 From: Pearl Dsilva Date: Tue, 14 Nov 2023 22:43:25 -0500 Subject: [PATCH] build failure - fix unit test --- .../java/org/apache/cloudstack/service/NsxApiClientTest.java | 3 +++ 1 file changed, 3 insertions(+) diff --git a/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxApiClientTest.java b/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxApiClientTest.java index 1865a008151..603f5fd628d 100644 --- a/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxApiClientTest.java +++ b/plugins/network-elements/nsx/src/test/java/org/apache/cloudstack/service/NsxApiClientTest.java @@ -16,6 +16,7 @@ // under the License. package org.apache.cloudstack.service; +import com.cloud.network.Network; import com.vmware.nsx_policy.infra.domains.Groups; import com.vmware.nsx_policy.model.Group; import com.vmware.nsx_policy.model.PathExpression; @@ -72,6 +73,7 @@ public class NsxApiClientTest { NsxNetworkRule rule = Mockito.mock(NsxNetworkRule.class); Mockito.when(rule.getSourceCidrList()).thenReturn(List.of("ANY")); Mockito.when(rule.getTrafficType()).thenReturn("Ingress"); + Mockito.when(rule.getService()).thenReturn(Network.Service.NetworkACL); String segmentName = "segment"; List sourceGroups = client.getGroupsForTraffic(rule, segmentName, true); List destinationGroups = client.getGroupsForTraffic(rule, segmentName, false); @@ -84,6 +86,7 @@ public class NsxApiClientTest { NsxNetworkRule rule = Mockito.mock(NsxNetworkRule.class); Mockito.when(rule.getSourceCidrList()).thenReturn(List.of("ANY")); Mockito.when(rule.getTrafficType()).thenReturn("Egress"); + Mockito.when(rule.getService()).thenReturn(Network.Service.NetworkACL); String segmentName = "segment"; List sourceGroups = client.getGroupsForTraffic(rule, segmentName, true); List destinationGroups = client.getGroupsForTraffic(rule, segmentName, false);