commit ba69d6e3786d4b1f73ea356cf60c1b6cce45e458
author Pearl Dsilva <pearl1594@gmail.com> Wed Dec 27 13:26:51 2023 -0500
committer Pearl Dsilva <pearl1594@gmail.com> Wed Dec 27 13:26:51 2023 -0500
tree 82adf521ed57054aae29dfa7b026bc187304cafa
parent 235d1f1d72c0e2d680d38578f473b87153af6b51

fix test assertion

server/src/test/java/com/cloud/network/router/CommandSetupHelperTest.java

diff --git a/server/src/test/java/com/cloud/network/router/CommandSetupHelperTest.java b/server/src/test/java/com/cloud/network/router/CommandSetupHelperTest.java
index 3dda7e1..1f9bd39 100644
--- a/server/src/test/java/com/cloud/network/router/CommandSetupHelperTest.java
+++ b/server/src/test/java/com/cloud/network/router/CommandSetupHelperTest.java
@@ -16,6 +16,7 @@
 // under the License.
 package com.cloud.network.router;
 
+import com.cloud.agent.api.Command;
 import com.cloud.agent.api.routing.VmDataCommand;
 import com.cloud.agent.manager.Commands;
 import com.cloud.configuration.ConfigurationManager;
@@ -25,6 +26,7 @@
 import com.cloud.dc.dao.DataCenterDao;
 import com.cloud.dc.dao.VlanDao;
 import com.cloud.network.NetworkModel;
+import com.cloud.network.Networks;
 import com.cloud.network.PublicIpAddress;
 import com.cloud.network.addr.PublicIp;
 import com.cloud.network.dao.IPAddressDao;
@@ -54,10 +56,15 @@
 import org.springframework.test.util.ReflectionTestUtils;
 
 import java.util.ArrayList;
+import java.util.EnumMap;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.*;
+
 @RunWith(MockitoJUnitRunner.class)
 public class CommandSetupHelperTest {
 
@@ -161,7 +168,7 @@
         PublicIpAddress publicIpAddress = new PublicIp(ipAddressVO, vlanVO, 0x0ac00000L);
         List<PublicIpAddress> pubIpList = new ArrayList<>(1);
         pubIpList.add(publicIpAddress);
-        Commands commands = Mockito.mock(Commands.class);
+        Commands commands = new Commands(Command.OnError.Stop);
         Map<String, String> vlanMacAddress = new HashMap<>();
         NicVO nicVO = new NicVO("nic", 1L, 2L, VirtualMachine.Type.User);
         NetworkVO networkVO = new NetworkVO();
@@ -190,5 +197,6 @@
         Mockito.when(dcDao.findById(ArgumentMatchers.anyLong())).thenReturn(dc);
 
         commandSetupHelper.createVpcAssociatePublicIPCommands(router, pubIpList, commands, vlanMacAddress);
+        Assert.assertEquals(2, commands.size());
     }
 }