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

Added error checking for missing info

Browse Source
This commit is contained in:
Logan Lipke
2020-06-16 12:09:56 -07:00
parent e81a74e254
commit 546392b8d6
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
stationStressTest.py
View File

@@ -28,9 +28,12 @@ class StressTester(LFCliBase):
return info return info
def cleanup(self): def cleanup(self):
sta_json = super().json_get("port/1/1/list?field=alias")['interfaces'] try:
sta_json = super().json_get("port/1/1/list?field=alias")['interfaces']
except TypeError:
sta_json = None
cx_json = super().json_get("cx") cx_json = super().json_get("cx")
endp_json = super().json_get("endp")['endpoint'] endp_json = super().json_get("endp")
# get and remove current stations # get and remove current stations
if sta_json is not None: if sta_json is not None:
@@ -49,10 +52,12 @@ class StressTester(LFCliBase):
#print(data) #print(data)
super().json_post(req_url, data) super().json_post(req_url, data)
time.sleep(.5) time.sleep(.5)
else:
print("No stations found to cleanup")
# get and remove current cxs # get and remove current cxs
if cx_json is not None: if cx_json is not None:
print("Removing old cross-connects") print("Removing old cross connects")
for name in list(cx_json): for name in list(cx_json):
if name != 'handler' and name != 'uri': if name != 'handler' and name != 'uri':
#print(name) #print(name)
@@ -63,11 +68,13 @@ class StressTester(LFCliBase):
} }
super().json_post(req_url, data) super().json_post(req_url, data)
time.sleep(.5) time.sleep(.5)
else:
print("No cross connects found to cleanup")
# get and remove current endps # get and remove current endps
if endp_json is not None: if endp_json is not None:
print("Removing old endpoints") print("Removing old endpoints")
for name in list(endp_json): for name in list(endp_json['endpoint']):
#print(list(name)[0]) #print(list(name)[0])
req_url = "cli-json/rm_endp" req_url = "cli-json/rm_endp"
data = { data = {
@@ -76,6 +83,8 @@ class StressTester(LFCliBase):
#print(data) #print(data)
super().json_post(req_url, data) super().json_post(req_url, data)
time.sleep(.5) time.sleep(.5)
else:
print("No endpoints found to cleanup")
def run(self): def run(self):
parser = argparse.ArgumentParser(description="Create max stations for each radio") parser = argparse.ArgumentParser(description="Create max stations for each radio")
@@ -205,7 +214,6 @@ class StressTester(LFCliBase):
padding_num = 1000 # uses all but the first number to create names for stations padding_num = 1000 # uses all but the first number to create names for stations
# clean up old stations # clean up old stations
print("Cleaning up old Stations")
self.cleanup() self.cleanup()
# create new stations # create new stations