add small sleep

Signed-off-by: Emil Gustafsson <[email protected]>

Author: Emil Gustafsson

src/main/java/com/cisco/trex/stateless/TRexClient.java

@@ -395,10 +395,10 @@ public void removeRxFilters(int portIndex, int profileId) {
    */
   public void waitOnTrafficToFinish(int timeoutInSecounds, int rxDelayMs, Port... ports) {
     long endTime = System.currentTimeMillis() + timeoutInSecounds * 1000;
-    List<Port> portsNotIdleYet = new ArrayList<>(Arrays.asList(ports));
+    List<Port> portsStillSendingTraffic = new ArrayList<>(Arrays.asList(ports));
 
-    while (!portsNotIdleYet.isEmpty()) {
-      Iterator<Port> iter = portsNotIdleYet.iterator();
+    while (!portsStillSendingTraffic.isEmpty()) {
+      Iterator<Port> iter = portsStillSendingTraffic.iterator();
       while (iter.hasNext()) {
         if (getPortStatus(iter.next().getIndex()).get().getState() != "TX") {
           iter.remove();
@@ -407,6 +407,9 @@ public void waitOnTrafficToFinish(int timeoutInSecounds, int rxDelayMs, Port...
       if (System.currentTimeMillis() > endTime) {
         break;
       }
+      if (!portsStillSendingTraffic.isEmpty()) {
+        sleepMilliSecounds(10);
+      }
     }
 
     removeRxFiltersWithDelay(rxDelayMs, ports);
@@ -430,17 +433,21 @@ protected void removeRxFiltersWithDelay(int rxDelayMs, Port... ports) {
       rxDelayToUse = rxDelayMs;
     }
 
-    try {
-      Thread.sleep(rxDelayToUse);
-    } catch (InterruptedException e) {
-      // Do nothing
-    }
+    sleepMilliSecounds(rxDelayToUse);
 
     for (Port port : ports) {
       removeRxFilters(port.getIndex(), 0);
     }
   }
 
+  protected void sleepMilliSecounds(int secounds) {
+    try {
+      Thread.sleep(secounds);
+    } catch (InterruptedException e) {
+      // Do nothing
+    }
+  }
+
   /** Set promiscuous mode, Enable interface to receive packets from all mac addresses */
   public void setPromiscuousMode(int portIndex, boolean enabled) {
     Map<String, Object> payload = createPayload(portIndex);