lf_interop_throughput.py: check cx states are running before testing

Added get_cx_states to retrieve current state of cross connects and wait
to begin test duration until 'Run' status is returned.

Verified with:
./lf_interop_throughput.py --mgr 192.168.91.50 --mgr_port 8080 --ssid Quincefruit --passwd lanforge --security wpa2 --upstream_port 1.050.01 --traffic_type lf_udp --download 100000000 --test_duration 1m --packet_size 1500 --incremental_capacity 1 --precleanup --postcleanup

devices: 1.50,1.70

Signed-off-by: Cameron LaPlante <[email protected]>

Author: claplante-candela

py-scripts/lf_interop_throughput.py

@@ -604,6 +604,19 @@ def cleanup(self):
         logger.info("cleanup done")
         self.cx_profile.cleanup()
 
+    def get_cx_states(self, device_names):
+        '''
+        Get the cx states of the devices (ie Run, Stopped, WAITING)
+        '''
+        cx_state_list = []
+        for device in device_names:
+            try:
+                device_data = self.json_get('/cx/all')[device]
+                cx_state_list.append(device_data['state'])
+            except KeyError:
+                logger.error("Error: %s key not found in cx data", device)
+        return cx_state_list
+
     def monitor(self, iteration, individual_df, device_names, incremental_capacity_list, overall_start_time, overall_end_time):
 
         throughput, upload, download, upload_throughput, download_throughput, connections_upload, connections_download = {}, [], [], [], [], {}, {}
@@ -623,9 +636,19 @@ def monitor(self, iteration, individual_df, device_names, incremental_capacity_l
         connections_upload_realtime = dict.fromkeys(list(self.cx_profile.created_cx.keys()), float(0))
         connections_download_realtime = dict.fromkeys(list(self.cx_profile.created_cx.keys()), float(0))
 
-        # provide time for connections to start and then initiaize the start and end time.
-        print("Waiting for connections to start...")
-        time.sleep(10)
+        # loop to get_cx_states until one return 'Running'
+        cx_states_down = True
+        logger.info("Waiting for cx to start")
+
+        while cx_states_down:
+            states = self.get_cx_states(list(self.cx_profile.created_cx.keys()))
+            logger.info("states: {}".format(states))
+
+            for cx_state in states:
+                if cx_state == 'Run':
+                    cx_states_down = False
+            time.sleep(2)
+
         start_time = datetime.now()
         end_time = start_time + timedelta(seconds=int(self.test_duration))
         logger.info("Monitoring cx and endpoints")
@@ -1718,11 +1741,11 @@ def parse_args():
     parser = argparse.ArgumentParser(
         prog="lf_interop_throughputput.py",
         formatter_class=argparse.RawTextHelpFormatter,
-        epilog='''\
+        epilog=r'''\
             Provides the available devices and allows user to run the wifi capacity test
             on particular devices by specifying direction as upload, download and bidirectional including different types of loads and incremental capacity.
             ''',
-        description='''\
+        description=r'''\
 
             NAME: lf_interop_throughputput.py