scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-11-04 04:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

sta_connect.py: Updated file with autopep8

Browse Source
This commit is contained in:
Logan Lipke
2022-05-31 11:24:08 -07:00
committed by shivam
parent ba8d2dd319
commit e6beb7c4ed
1 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
py-scripts/sta_connect.py
View File

@@ -133,7 +133,6 @@ class StaConnect(Realm):
else: else:
self._fail("Not all ports disappeared") self._fail("Not all ports disappeared")
def num_associated(self, bssid): def num_associated(self, bssid):
counter = 0 counter = 0
# print("there are %d results" % len(self.station_results)) # print("there are %d results" % len(self.station_results))
@@ -168,7 +167,7 @@ class StaConnect(Realm):
if not self.finish_test(): if not self.finish_test():
self._fail("finish_test stage failed") self._fail("finish_test stage failed")
return False, "finish_test" return False, "finish_test"
# remove all endpoints and cxs # remove all endpoints and cxs
if self.cleanup_on_exit: if self.cleanup_on_exit:
if not self.cleanup(): if not self.cleanup():
@@ -208,7 +207,7 @@ class StaConnect(Realm):
self._fail("Warning: %s lacks ip address" % self.get_upstream_url()) self._fail("Warning: %s lacks ip address" % self.get_upstream_url())
return False return False
# Pre-cleanup of stations being used # Pre-cleanup of stations being used
self.remove_stations() self.remove_stations()
# Create stations and turn dhcp on # Create stations and turn dhcp on
@@ -379,7 +378,7 @@ class StaConnect(Realm):
self._pass("All cxs appeared") self._pass("All cxs appeared")
else: else:
self._fail("Not all cxs appeared") self._fail("Not all cxs appeared")
return True return True
def start(self): def start(self):
@@ -450,12 +449,12 @@ class StaConnect(Realm):
ptest_b_tx = ptest['endpoint']['tx bytes'] ptest_b_tx = ptest['endpoint']['tx bytes']
ptest_b_rx = ptest['endpoint']['rx bytes'] ptest_b_rx = ptest['endpoint']['rx bytes']
if (self.compare_vals("testTCP-A TX", ptest_a_tx) & self.compare_vals("testTCP-A RX", ptest_a_rx) & if (self.compare_vals("testTCP-A TX", ptest_a_tx) & self.compare_vals("testTCP-A RX", ptest_a_rx) &
self.compare_vals("testTCP-B TX", ptest_b_tx) & self.compare_vals("testTCP-B RX", ptest_b_rx)): self.compare_vals("testTCP-B TX", ptest_b_tx) & self.compare_vals("testTCP-B RX", ptest_b_rx)):
return True return True
else: else:
return False return False
except Exception as e: except Exception as e:
self.error(e) self.error(e)
@@ -497,7 +496,7 @@ class StaConnect(Realm):
kpi_csv.kpi_dict['Units'] = "bps" kpi_csv.kpi_dict['Units'] = "bps"
kpi_csv.kpi_dict['Graph-Group'] = "Endpoint RX bytes" kpi_csv.kpi_dict['Graph-Group'] = "Endpoint RX bytes"
kpi_csv.kpi_csv_write_dict(kpi_csv.kpi_dict) kpi_csv.kpi_csv_write_dict(kpi_csv.kpi_dict)
if csv_outfile is not None: if csv_outfile is not None:
current_time = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()) current_time = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
csv_outfile = "{}_{}-sta_connect.csv".format( csv_outfile = "{}_{}-sta_connect.csv".format(
@@ -559,18 +558,16 @@ Example:
_dut_bssid=args.dut_bssid, _dut_ssid=args.ssid, _dut_security=args.security, _dut_bssid=args.dut_bssid, _dut_ssid=args.ssid, _dut_security=args.security,
_sta_name=["sta0000"]) _sta_name=["sta0000"])
success, method = staConnect.run() success, method = staConnect.run()
if not success: if not success:
print("FAIL: staConnect.run method failed at stage \"{method}\"".format(method=method)) print("FAIL: staConnect.run method failed at stage \"{method}\"".format(method=method))
else: else:
print("PASS: staConnect tests finished successfully") print("PASS: staConnect tests finished successfully")
staConnect.get_result_list() staConnect.get_result_list()
staConnect.generate_report(test_rig=args.test_rig, test_tag=args.test_tag, dut_hw_version=args.dut_hw_version, staConnect.generate_report(test_rig=args.test_rig, test_tag=args.test_tag, dut_hw_version=args.dut_hw_version,
dut_sw_version=args.dut_sw_version, dut_model_num=args.dut_model_num, dut_sw_version=args.dut_sw_version, dut_model_num=args.dut_model_num,
dut_serial_num=args.dut_serial_num, test_id=args.test_id, csv_outfile=args.csv_outfile) dut_serial_num=args.dut_serial_num, test_id=args.test_id, csv_outfile=args.csv_outfile)
if not staConnect.passes(): if not staConnect.passes():
print("FAIL: Some tests failed") print("FAIL: Some tests failed")
else: else: