mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-11-03 04:18:15 +00:00
4bdf061d24
This is pure 'git mv'. No changes were made to files. They will not work until we fix up the include paths again, that will be handled in future commits. Signed-off-by: Ben Greear <greearb@candelatech.com>
288 lines
13 KiB
Python
Executable File
288 lines
13 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
# Example to set profile on NOLA-12 testbed:
|
|
#./sdk_set_profile.py --testrail-user-id NONE --model ecw5410 --ap-jumphost-address localhost --ap-jumphost-port 8823 --ap-jumphost-password pumpkin77 \
|
|
# --ap-jumphost-tty /dev/ttyAP1 --testbed "NOLA-12" --lanforge-ip-address localhost --lanforge-port-number 8822 \
|
|
# --default-ap-profile TipWlan-2-Radios --sdk-base-url https://wlan-portal-svc-ben-testbed.cicd.lab.wlan.tip.build --skip-radius
|
|
|
|
# Example to set profile on NOLA-01 testbed
|
|
#./sdk_set_profile.py --testrail-user-id NONE --model ecw5410 --ap-jumphost-address localhost --ap-jumphost-port 8803 \
|
|
# --ap-jumphost-password pumpkin77 --ap-jumphost-tty /dev/ttyAP1 --testbed "NOLA-01" --lanforge-ip-address localhost \
|
|
# --lanforge-port-number 8802 --default-ap-profile TipWlan-2-Radios --sdk-base-url https://wlan-portal-svc.cicd.lab.wlan.tip.build \
|
|
# --skip-radius
|
|
|
|
from UnitTestBase import *
|
|
from cloudsdk import CreateAPProfiles
|
|
|
|
parser = argparse.ArgumentParser(description="SDK Set Profile", add_help=False)
|
|
parser.add_argument("--default-ap-profile", type=str,
|
|
help="Default AP profile to use as basis for creating new ones, typically: TipWlan-2-Radios or TipWlan-3-Radios",
|
|
required=True)
|
|
parser.add_argument("--skip-radius", dest="skip_radius", action='store_true',
|
|
help="Should we skip the RADIUS configs or not")
|
|
parser.add_argument("--skip-wpa", dest="skip_wpa", action='store_true',
|
|
help="Should we skip the WPA ssid or not")
|
|
parser.add_argument("--skip-wpa2", dest="skip_wpa2", action='store_true',
|
|
help="Should we skip the WPA2 ssid or not")
|
|
parser.set_defaults(skip_radius=False)
|
|
parser.set_defaults(skip_wpa=False)
|
|
parser.set_defaults(skip_wpa2=False)
|
|
parser.add_argument("--skip-profiles", dest="skip_profiles", action='store_true',
|
|
help="Should we skip creating new ssid profiles?")
|
|
parser.set_defaults(skip_profiles=False)
|
|
|
|
parser.add_argument("--psk-5g-wpa2", type=str,
|
|
help="Allow over-riding the 5g-wpa2 PSK value.")
|
|
parser.add_argument("--psk-5g-wpa", type=str,
|
|
help="Allow over-riding the 5g-wpa PSK value.")
|
|
parser.add_argument("--psk-2g-wpa2", type=str,
|
|
help="Allow over-riding the 2g-wpa2 PSK value.")
|
|
parser.add_argument("--psk-2g-wpa", type=str,
|
|
help="Allow over-riding the 2g-wpa PSK value.")
|
|
|
|
parser.add_argument("--ssid-5g-wpa2", type=str,
|
|
help="Allow over-riding the 5g-wpa2 SSID value.")
|
|
parser.add_argument("--ssid-5g-wpa", type=str,
|
|
help="Allow over-riding the 5g-wpa SSID value.")
|
|
parser.add_argument("--ssid-2g-wpa2", type=str,
|
|
help="Allow over-riding the 2g-wpa2 SSID value.")
|
|
parser.add_argument("--ssid-2g-wpa", type=str,
|
|
help="Allow over-riding the 2g-wpa SSID value.")
|
|
|
|
base = UnitTestBase("skd-set-profile", parser)
|
|
|
|
command_line_args = base.command_line_args
|
|
|
|
# cmd line takes precedence over env-vars.
|
|
cloudSDK_url = command_line_args.sdk_base_url # was os.getenv('CLOUD_SDK_URL')
|
|
local_dir = command_line_args.local_dir # was os.getenv('SANITY_LOG_DIR')
|
|
report_path = command_line_args.report_path # was os.getenv('SANITY_REPORT_DIR')
|
|
report_template = command_line_args.report_template # was os.getenv('REPORT_TEMPLATE')
|
|
|
|
## TestRail Information
|
|
tr_user = command_line_args.testrail_user_id # was os.getenv('TR_USER')
|
|
tr_pw = command_line_args.testrail_user_password # was os.getenv('TR_PWD')
|
|
milestoneId = command_line_args.milestone # was os.getenv('MILESTONE')
|
|
projectId = command_line_args.testrail_project # was os.getenv('PROJECT_ID')
|
|
testRunPrefix = command_line_args.testrail_run_prefix # os.getenv('TEST_RUN_PREFIX')
|
|
|
|
##Jfrog credentials
|
|
jfrog_user = command_line_args.jfrog_user_id # was os.getenv('JFROG_USER')
|
|
jfrog_pwd = command_line_args.jfrog_user_password # was os.getenv('JFROG_PWD')
|
|
|
|
##EAP Credentials
|
|
identity = command_line_args.eap_id # was os.getenv('EAP_IDENTITY')
|
|
ttls_password = command_line_args.ttls_password # was os.getenv('EAP_PWD')
|
|
|
|
## AP Credentials
|
|
ap_username = command_line_args.ap_username # was os.getenv('AP_USER')
|
|
|
|
##LANForge Information
|
|
lanforge_ip = command_line_args.lanforge_ip_address
|
|
lanforge_port = command_line_args.lanforge_port_number
|
|
lanforge_prefix = command_line_args.lanforge_prefix
|
|
lanforge_2g_radio = command_line_args.lanforge_2g_radio
|
|
lanforge_5g_radio = command_line_args.lanforge_5g_radio
|
|
|
|
build = command_line_args.build_id
|
|
|
|
logger = base.logger
|
|
hdlr = base.hdlr
|
|
|
|
if command_line_args.testbed == None:
|
|
print("ERROR: Must specify --testbed argument for this test.")
|
|
sys.exit(1)
|
|
|
|
client: TestRail_Client = TestRail_Client(command_line_args)
|
|
|
|
###Get Cloud Bearer Token
|
|
cloud: CloudSDK = CloudSDK(command_line_args)
|
|
bearer = cloud.get_bearer(cloudSDK_url, cloud_type)
|
|
|
|
cloud.assert_bad_response = True
|
|
|
|
model_id = command_line_args.model
|
|
equipment_id = command_line_args.equipment_id
|
|
|
|
print("equipment-id: %s"%(equipment_id))
|
|
|
|
if equipment_id == "-1":
|
|
eq_id = ap_ssh_ovsh_nodec(command_line_args, 'id')
|
|
print("EQ Id: %s"%(eq_id))
|
|
|
|
# Now, query equipment to find something that matches.
|
|
eq = cloud.get_customer_equipment(cloudSDK_url, bearer, customer_id)
|
|
for item in eq:
|
|
for e in item['items']:
|
|
print(e['id'], " ", e['inventoryId'])
|
|
if e['inventoryId'].endswith("_%s"%(eq_id)):
|
|
print("Found equipment ID: %s inventoryId: %s"%(e['id'], e['inventoryId']))
|
|
equipment_id = str(e['id'])
|
|
|
|
if equipment_id == -1:
|
|
print("ERROR: Could not find equipment-id.")
|
|
sys.exit(1)
|
|
|
|
###Get Current AP Firmware and upgrade
|
|
try:
|
|
ap_cli_info = ssh_cli_active_fw(command_line_args)
|
|
ap_cli_fw = ap_cli_info['active_fw']
|
|
except Exception as ex:
|
|
print(ex)
|
|
logging.error(logging.traceback.format_exc())
|
|
ap_cli_info = "ERROR"
|
|
print("FAILED: Cannot Reach AP CLI.");
|
|
sys.exit(1)
|
|
|
|
fw_model = ap_cli_fw.partition("-")[0]
|
|
print('Current Active AP FW from CLI:', ap_cli_fw)
|
|
###Find Latest FW for Current AP Model and Get FW ID
|
|
|
|
############################################################################
|
|
#################### Create Report #########################################
|
|
############################################################################
|
|
|
|
# Create Report Folder for Today
|
|
today = str(date.today())
|
|
try:
|
|
os.mkdir(report_path + today)
|
|
except OSError:
|
|
print("Creation of the directory %s failed" % report_path)
|
|
else:
|
|
print("Successfully created the directory %s " % report_path)
|
|
|
|
logger.info('Report data can be found here: ' + report_path + today)
|
|
|
|
|
|
##Get Bearer Token to make sure its valid (long tests can require re-auth)
|
|
bearer = cloud.get_bearer(cloudSDK_url, cloud_type)
|
|
|
|
radius_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "Radius")
|
|
|
|
prof_5g_eap_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP")
|
|
prof_5g_wpa2_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2")
|
|
prof_5g_wpa_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA")
|
|
prof_2g_eap_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP")
|
|
prof_2g_wpa2_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2")
|
|
prof_2g_wpa_name = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA")
|
|
|
|
prof_5g_eap_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP_NAT")
|
|
prof_5g_wpa2_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2_NAT")
|
|
prof_5g_wpa_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA_NAT")
|
|
prof_2g_eap_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP_NAT")
|
|
prof_2g_wpa2_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2_NAT")
|
|
prof_2g_wpa_name_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA_NAT")
|
|
|
|
prof_5g_eap_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP_VLAN")
|
|
prof_5g_wpa2_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2_VLAN")
|
|
prof_5g_wpa_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA_VLAN")
|
|
prof_2g_eap_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP_VLAN")
|
|
prof_2g_wpa2_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2_VLAN")
|
|
prof_2g_wpa_name_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA_VLAN")
|
|
|
|
|
|
|
|
prof_names = [prof_5g_wpa2_name, prof_5g_wpa_name, prof_2g_wpa2_name, prof_2g_wpa_name,
|
|
prof_5g_wpa2_name_nat, prof_5g_wpa_name_nat, prof_2g_wpa2_name_nat, prof_2g_wpa_name_nat,
|
|
prof_5g_wpa2_name_vlan, prof_5g_wpa_name_vlan, prof_2g_wpa2_name_vlan, prof_2g_wpa_name_vlan]
|
|
|
|
prof_names_eap = [prof_5g_eap_name, prof_2g_eap_name,
|
|
prof_5g_eap_name_nat, prof_2g_eap_name_nat,
|
|
prof_5g_eap_name_vlan, prof_2g_eap_name_vlan]
|
|
|
|
# TOOD: Allow configuring this on cmd line
|
|
ssid_5g_eap = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP")
|
|
ssid_5g_wpa2 = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2")
|
|
ssid_5g_wpa = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA")
|
|
ssid_2g_eap = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP")
|
|
ssid_2g_wpa2 = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2")
|
|
ssid_2g_wpa = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA")
|
|
|
|
ssid_5g_eap_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP_NAT")
|
|
ssid_5g_wpa2_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2_NAT")
|
|
ssid_5g_wpa_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA_NAT")
|
|
ssid_2g_eap_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP_NAT")
|
|
ssid_2g_wpa2_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2_NAT")
|
|
ssid_2g_wpa_nat = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA_NAT")
|
|
|
|
ssid_5g_eap_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_EAP_VLAN")
|
|
ssid_5g_wpa2_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA2_VLAN")
|
|
ssid_5g_wpa_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "5G_WPA_VLAN")
|
|
ssid_2g_eap_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_EAP_VLAN")
|
|
ssid_2g_wpa2_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA2_VLAN")
|
|
ssid_2g_wpa_vlan = "%s-%s-%s"%(command_line_args.testbed, fw_model, "2G_WPA_VLAN")
|
|
|
|
psk_5g_wpa2 = "%s-%s"%(fw_model, "5G_WPA2")
|
|
psk_5g_wpa = "%s-%s"%(fw_model, "5G_WPA")
|
|
psk_2g_wpa2 = "%s-%s"%(fw_model, "2G_WPA2")
|
|
psk_2g_wpa = "%s-%s"%(fw_model, "2G_WPA")
|
|
|
|
psk_5g_wpa2_nat = "%s-%s"%(fw_model, "5G_WPA2_NAT")
|
|
psk_5g_wpa_nat = "%s-%s"%(fw_model, "5G_WPA_NAT")
|
|
psk_2g_wpa2_nat = "%s-%s"%(fw_model, "2G_WPA2_NAT")
|
|
psk_2g_wpa_nat = "%s-%s"%(fw_model, "2G_WPA_NAT")
|
|
|
|
psk_5g_wpa2_vlan = "%s-%s"%(fw_model, "5G_WPA2_VLAN")
|
|
psk_5g_wpa_vlan = "%s-%s"%(fw_model, "5G_WPA_VLAN")
|
|
psk_2g_wpa2_vlan = "%s-%s"%(fw_model, "2G_WPA2_VLAN")
|
|
psk_2g_wpa_vlan = "%s-%s"%(fw_model, "2G_WPA_VLAN")
|
|
|
|
# Allow cmd-line to override
|
|
if command_line_args.psk_5g_wpa2:
|
|
psk_5g_wpa2 = command_line_args.psk_5g_wpa2
|
|
if command_line_args.psk_5g_wpa:
|
|
psk_5g_wpa = command_line_args.psk_5g_wpa
|
|
if command_line_args.psk_2g_wpa2:
|
|
psk_2g_wpa2 = command_line_args.psk_2g_wpa2
|
|
if command_line_args.psk_2g_wpa:
|
|
psk_2g_wpa = command_line_args.psk_2g_wpa
|
|
|
|
if command_line_args.ssid_5g_wpa2:
|
|
ssid_5g_wpa2 = command_line_args.ssid_5g_wpa2
|
|
if command_line_args.ssid_5g_wpa:
|
|
ssid_5g_wpa = command_line_args.ssid_5g_wpa
|
|
if command_line_args.ssid_2g_wpa2:
|
|
ssid_2g_wpa2 = command_line_args.ssid_2g_wpa2
|
|
if command_line_args.ssid_2g_wpa:
|
|
ssid_2g_wpa = command_line_args.ssid_2g_wpa
|
|
|
|
|
|
|
|
print("creating Profiles")
|
|
ssid_template = "TipWlan-Cloud-Wifi"
|
|
|
|
# Data Structure to Give to CreateAPProfiles.create_ap_profiles()
|
|
ssid_profile_data = {
|
|
"ssid_template" : ssid_template,
|
|
|
|
"2G" :
|
|
{
|
|
"eap": {"info" : [ssid_2g_eap ,prof_2g_eap_name, prof_2g_eap_name_nat, prof_2g_eap_name_vlan], "psk" :[]},
|
|
"wpa": {"info" : [ssid_2g_wpa, prof_2g_wpa_name, prof_2g_wpa_name_nat, prof_2g_wpa_name_vlan], "psk" : [psk_2g_wpa, psk_2g_wpa_nat, psk_2g_wpa_vlan]},
|
|
"wpa2": {"info" : [ssid_2g_wpa2, prof_2g_wpa2_name, prof_2g_wpa2_name_nat, prof_2g_wpa2_name_vlan], "psk" : [psk_2g_wpa2, psk_2g_wpa2_nat, psk_2g_wpa2_vlan]},
|
|
},
|
|
|
|
"5G" :
|
|
{
|
|
"eap": {"info" : [ssid_5g_eap, prof_5g_eap_name, prof_5g_eap_name_nat, prof_5g_eap_name_vlan], "psk" : []},
|
|
"wpa": {"info" : [ssid_5g_wpa, prof_5g_wpa_name, prof_5g_wpa_name_nat, prof_5g_wpa_name_vlan], "psk" : [psk_5g_wpa, psk_5g_wpa_nat, psk_5g_wpa_vlan]},
|
|
"wpa2":{"info" : [ssid_5g_wpa2, prof_5g_wpa2_name, prof_5g_wpa2_name_nat, prof_5g_wpa_name_vlan], "psk" : [psk_2g_wpa2, psk_5g_wpa2_nat, psk_5g_wpa2_vlan]}
|
|
}
|
|
}
|
|
|
|
obj = CreateAPProfiles(command_line_args, cloud=cloud, client= client)
|
|
|
|
if not command_line_args.skip_profiles:
|
|
if not command_line_args.skip_radius:
|
|
obj.create_radius_profile(radius_name, rid, key)
|
|
obj.create_ssid_profile(ssid_profile_data= ssid_profile_data, skip_wpa2=command_line_args.skip_wpa2,
|
|
skip_wpa=command_line_args.skip_wpa, skip_eap=command_line_args.skip_radius)
|
|
|
|
print("Create AP with equipment-id: ", equipment_id)
|
|
obj.create_ap_bridge_profile(eq_id=equipment_id, fw_model=fw_model)
|
|
obj.validate_changes()
|
|
|
|
print("Profiles Created")
|
|
|
|
|