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

Added 320MHz bandwidth support in client_connect_using_radio method

Browse Source 
Signed-off-by: jitendra-kushavah <jitendra.kushavah@candelatech.com>
This commit is contained in:
jitendra-kushavah
2025-05-23 15:50:03 +05:30
parent ebb516a9b8
commit 326eadfbc7
1 changed files with 53 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
lf_libs/lf_libs.py
View File

@@ -1731,10 +1731,10 @@ class lf_libs:
if name is not None:
allure.attach(name=name, body=str(data_table))
def get_radio_availabilities(self, num_stations_2g: int = 0, num_stations_5g: int = 0) -> tuple:
def get_radio_availabilities(self, num_stations_2g: int = 0, num_stations_5g: int = 0, num_stations_6g: int = 0) -> tuple:
"""
Get the port name of radios and how many stations to be created on each radio for the given num of
2g stations and 5g stations. This method takes in account the fact that same radio can't be used to
2g stations, 5g stations and 6g stations. This method takes in account the fact that same radio can't be used to
create a station on multiple band at the same time even though it supports both bands.
- Returns tuple[dict[str, int], dict[str, int]] or skips the test if not enough radios are available
@@ -1743,9 +1743,11 @@ class lf_libs:
message = None
requested_num_stations_2g = num_stations_2g
requested_num_stations_5g = num_stations_5g
requested_num_stations_6g = num_stations_6g
radio_dict_2g = {}
radio_dict_5g = {}
radio_dict_6g = {}
dict_all_radios_2g = {
"be200_radios": self.be200_radios,
"ax210_radios": self.ax210_radios,
@@ -1762,6 +1764,10 @@ class lf_libs:
"wave2_5g_radios": self.wave2_5g_radios,
"wave1_radios": self.wave1_radios
}
dict_all_radios_6g = {
"be200_radios": self.be200_radios,
"ax210_radios": self.ax210_radios
}
max_station_per_radio = {
"wave2_2g_radios": 64,
"wave2_5g_radios": 64,
@@ -1818,6 +1824,36 @@ class lf_libs:
message = f"Not enough radios available for connecting {requested_num_stations_5g} 5g clients!"
break
if num_stations_6g != 0:
for keys in dict_all_radios_6g:
if num_stations_6g == 0:
break
max_station = max_station_per_radio[keys]
if len(dict_all_radios_6g[keys]) > 0:
diff = max_station - num_stations_6g
for port_name in dict_all_radios_6g[keys]:
if port_name in radio_dict_2g:
continue
if port_name in radio_dict_5g:
continue
if diff >= 0:
radio_dict_6g[port_name] = num_stations_6g
num_stations_6g = 0
break
else:
radio_dict_6g[port_name] = max_station
num_stations_6g -= max_station
diff = max_station - num_stations_6g
if num_stations_6g != 0:
message = f"Not enough radios available for connecting {requested_num_stations_6g} 6g clients!"
return radio_dict_6g
logging.info(f"radio_dict_2g dict : {radio_dict_2g}")
logging.info(f"radio_dict_5g dict : {radio_dict_5g}")
logging.info(f"radio_dict_6g dict : {radio_dict_6g}")
if num_stations_2g != 0 or num_stations_5g != 0:
logging.info(f"Radio-2G-Stations dict : {num_stations_2g}")
logging.info(f"Radio-5G-Stations dict : {num_stations_5g}")
@@ -1832,7 +1868,7 @@ class lf_libs:
def client_connect_using_radio(self, ssid="[BLANK]", passkey="[BLANK]", security="wpa2", mode="BRIDGE", band=None,
vlan_id=[None], radio=None, client_type=0, station_name=[], dut_data=None,
sniff_radio=False, create_vlan=True, attach_port_info=True,
attach_station_data=True, timeout_sec=100):
attach_station_data=True, timeout_sec=100, enable_owe = False,is_bw320=False, is_ht160=False):
# pre cleanup
# if pre_cleanup:
# self.pre_cleanup()
@@ -1857,6 +1893,20 @@ class lf_libs:
client_connect = CreateStation(_host=self.manager_ip, _port=self.manager_http_port, _mode=client_type,
_sta_list=station_name, _password=passkey, _ssid=ssid, _security=security)
if enable_owe:
passwd = "[BLANK]"
client_connect.station_profile.set_command_param("add_sta", "key", passwd)
client_connect.station_profile.set_command_flag("add_sta", "use-owe", 1)
client_connect.station_profile.set_command_flag("add_sta", "wpa2_enable", 0)
logging.info(f"is_bw320:{is_bw320} and is_ht160: {is_ht160}")
if is_bw320:
client_connect.station_profile.set_command_flag("add_sta", "be320-enable", 1)
client_connect.station_profile.set_command_flag("add_sta", "ht160_enable", 1)
client_connect.station_profile.set_command_flag("add_sta", "disable_ht80", 0)
if is_ht160:
client_connect.station_profile.set_command_flag("add_sta", "ht160_enable", 1)
client_connect.station_profile.set_command_flag("add_sta", "disable_ht80", 0)
client_connect.upstream_port = upstream_port
client_connect.upstream_resource = 1
client_connect.radio = radio