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

sta-connect2: Support removing all stations on the resource.

Browse Source 
To better clean up system that is in unknown state.

Signed-off-by: Ben Greear <greearb@candelatech.com>
This commit is contained in:
Ben Greear
2020-12-10 12:05:32 -08:00
parent 42cc9f5896
commit d628d3ea47
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
py-scripts/sta_connect2.py
View File

@@ -37,7 +37,7 @@ class StaConnect2(LFCliBase):
_resource=1, _upstream_resource=1, _upstream_port="eth1", _resource=1, _upstream_resource=1, _upstream_port="eth1",
_sta_name=None, _sta_prefix='sta', _bringup_time_sec=300, _sta_name=None, _sta_prefix='sta', _bringup_time_sec=300,
debug_=False, _dut_security=OPEN, _exit_on_error=False, debug_=False, _dut_security=OPEN, _exit_on_error=False,
_cleanup_on_exit=True, _runtime_sec=60, _exit_on_fail=False): _cleanup_on_exit=True, _clean_all_sta=False, _runtime_sec=60, _exit_on_fail=False):
# do not use `super(LFCLiBase,self).__init__(self, host, port, _debugOn) # do not use `super(LFCLiBase,self).__init__(self, host, port, _debugOn)
# that is py2 era syntax and will force self into the host variable, making you # that is py2 era syntax and will force self into the host variable, making you
# very confused. # very confused.
@@ -56,6 +56,7 @@ class StaConnect2(LFCliBase):
self.upstream_port = _upstream_port self.upstream_port = _upstream_port
self.runtime_secs = _runtime_sec self.runtime_secs = _runtime_sec
self.cleanup_on_exit = _cleanup_on_exit self.cleanup_on_exit = _cleanup_on_exit
self.clean_all_sta = _clean_all_sta
self.sta_url_map = None # defer construction self.sta_url_map = None # defer construction
self.upstream_url = None # defer construction self.upstream_url = None # defer construction
self.station_names = [] self.station_names = []
@@ -139,14 +140,18 @@ class StaConnect2(LFCliBase):
return False return False
# remove old stations # remove old stations
print("Removing old stations") if self.clean_all_sta:
for sta_name in self.station_names: print("Removing all stations on resource.")
sta_url = self.get_station_url(sta_name) self.localrealm.remove_all_stations(self.resource)
response = self.json_get(sta_url) else:
if (response is not None) and (response["interface"] is not None): print("Removing old stations to be created by this test.")
for sta_name in self.station_names: for sta_name in self.station_names:
LFUtils.removePort(self.resource, sta_name, self.lfclient_url) sta_url = self.get_station_url(sta_name)
LFUtils.wait_until_ports_disappear(self.lfclient_url, self.station_names) response = self.json_get(sta_url)
if (response is not None) and (response["interface"] is not None):
for sta_name in self.station_names:
LFUtils.removePort(self.resource, sta_name, self.lfclient_url)
LFUtils.wait_until_ports_disappear(self.lfclient_url, self.station_names)
# Create stations and turn dhcp on # Create stations and turn dhcp on
self.station_profile = self.localrealm.new_station_profile() self.station_profile = self.localrealm.new_station_profile()