NFVBENCH-1 Number of seconds to wait for VMs to pass traffic can be

[nfvbench.git] / nfvbench / chain_clients.py

diff --git a/nfvbench/chain_clients.py b/nfvbench/chain_clients.py
index affccc8..bf51552 100644 (file)
--- a/nfvbench/chain_clients.py
+++ b/nfvbench/chain_clients.py
@@ -48,7 +48,9 @@ class BasicStageClient(object):
         self.host_ips = None
 
     def _ensure_vms_active(self):
-        for _ in range(self.config.generic_retry_count):
+        retry_count = (self.config.check_traffic_time_sec +
+                       self.config.generic_poll_sec - 1) / self.config.generic_poll_sec
+        for _ in range(retry_count):
             for i, instance in enumerate(self.vms):
                 if instance.status == 'ACTIVE':
                     continue