From 47827e40f71696e04ad805296dedc44a03451db3 Mon Sep 17 00:00:00 2001
From: Hector Fernandez <hector@iatec.biz>
Date: Mon, 6 Jul 2020 17:51:09 +0200
Subject: In case iperf3 server is busy, clients wait and retry.

---
 test-cli/test/tests/qaudio.py    |  1 -
 test-cli/test/tests/qethernet.py | 19 +++++++++++++------
 test-cli/test/tests/qwifi.py     | 19 +++++++++++++------
 3 files changed, 26 insertions(+), 13 deletions(-)

(limited to 'test-cli/test/tests')

diff --git a/test-cli/test/tests/qaudio.py b/test-cli/test/tests/qaudio.py
index bfbb051..364d8b2 100644
--- a/test-cli/test/tests/qaudio.py
+++ b/test-cli/test/tests/qaudio.py
@@ -48,7 +48,6 @@ class Qaudio(unittest.TestCase):
                         if a != b:
                             self.fail("failed: sent and received DTMF sequence don't match.")
                 else:
-                    print(self.__dtmf_secuence_result)
                     self.fail("failed: received DTMF sequence is shorter than expected.")
             else:
                 self.fail("failed: unable to use multimon command.")
diff --git a/test-cli/test/tests/qethernet.py b/test-cli/test/tests/qethernet.py
index d38de34..81acef1 100644
--- a/test-cli/test/tests/qethernet.py
+++ b/test-cli/test/tests/qethernet.py
@@ -1,6 +1,7 @@
 import unittest
 import sh
 import json
+import time
 
 
 class Qethernet(unittest.TestCase):
@@ -11,6 +12,7 @@ class Qethernet(unittest.TestCase):
     params = None
     __bwreal = None
     __resultlist = None  # resultlist is a python list of python dictionaries
+    timebetweenattempts = 5
 
     # varlist content: serverip, bwexpected, port
     def __init__(self, testname, testfunc, varlist):
@@ -37,12 +39,17 @@ class Qethernet(unittest.TestCase):
         self.__resultlist = []
 
     def execute(self):
-        # execute iperf command against the server
-        try:
-            p = sh.iperf3("-c", self.__serverip, "-n", self.__numbytestx, "-f", "m", "-p", self.__port, "-J",
-                          _timeout=20)
-        except sh.TimeoutException:
-            self.fail("failed: iperf timeout reached")
+        # execute iperf command against the server, but it implements attempts in case the server is busy
+        iperfdone = False
+        while not iperfdone:
+            try:
+                p = sh.iperf3("-c", self.__serverip, "-n", self.__numbytestx, "-f", "m", "-p", self.__port, "-J",
+                              _timeout=20)
+                iperfdone = True
+            except sh.TimeoutException:
+                self.fail("failed: iperf timeout reached")
+            except sh.ErrorReturnCode:
+                time.sleep(self.timebetweenattempts)
 
         # check if it was executed succesfully
         if p.exit_code == 0:
diff --git a/test-cli/test/tests/qwifi.py b/test-cli/test/tests/qwifi.py
index 4522057..3dd9e38 100644
--- a/test-cli/test/tests/qwifi.py
+++ b/test-cli/test/tests/qwifi.py
@@ -2,6 +2,7 @@ import unittest
 import sh
 import re
 import json
+import time
 
 
 class Qwifi(unittest.TestCase):
@@ -12,6 +13,7 @@ class Qwifi(unittest.TestCase):
     params = None
     __bwreal = None
     __resultlist = None  # resultlist is a python list of python dictionaries
+    timebetweenattempts = 5
 
     # varlist content: serverip, bwexpected, port
     def __init__(self, testname, testfunc, varlist):
@@ -48,12 +50,17 @@ class Qwifi(unittest.TestCase):
                         'inet addr:(?!127\.0{1,3}\.0{1,3}\.0{0,2}1$)((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)',
                         p.stdout.decode('ascii'))
                     if result:
-                        # execute iperf command against the server
-                        try:
-                            p = sh.iperf3("-c", self.__serverip, "-n", self.__numbytestx, "-f", "m", "-p", self.__port,
-                                          "-J", _timeout=30)
-                        except sh.TimeoutException:
-                            self.fail("failed: iperf timeout reached")
+                        # execute iperf command against the server, but it implements attempts in case the server is busy
+                        iperfdone = False
+                        while not iperfdone:
+                            try:
+                                p = sh.iperf3("-c", self.__serverip, "-n", self.__numbytestx, "-f", "m", "-p", self.__port,
+                                              "-J", _timeout=20)
+                                iperfdone = True
+                            except sh.TimeoutException:
+                                self.fail("failed: iperf timeout reached")
+                            except sh.ErrorReturnCode:
+                                time.sleep(self.timebetweenattempts)
 
                         # check if it was executed succesfully
                         if p.exit_code == 0:
-- 
cgit v1.1