mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-11-04 04:48:01 +00:00
235 lines
10 KiB
Python
Executable File
235 lines
10 KiB
Python
Executable File
#!/usr/bin/python3 -u
|
|
|
|
# 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
|
|
|
|
import sys
|
|
|
|
sys.path.append(f'../tests')
|
|
|
|
from UnitTestBase import *
|
|
from cloudsdk import CreateAPProfiles
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
|
|
|
|
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")
|
|
obj = CreateAPProfiles(command_line_args, cloud=cloud, client=client, fw_model=fw_model)
|
|
|
|
|
|
# Allow cmd-line to override
|
|
if command_line_args.psk_5g_wpa2:
|
|
obj.psk_data["5g"]["wpa2"]["name"] = command_line_args.psk_5g_wpa2
|
|
obj.psk_data["5g"]["wpa2"]["nat"] = command_line_args.psk_5g_wpa2
|
|
obj.psk_data["5g"]["wpa2"]["vlan"] = command_line_args.psk_5g_wpa2
|
|
if command_line_args.psk_5g_wpa:
|
|
obj.psk_data["5g"]["wpa"]["name"] = command_line_args.psk_5g_wpa
|
|
obj.psk_data["5g"]["wpa"]["nat"] = command_line_args.psk_5g_wpa
|
|
obj.psk_data["5g"]["wpa"]["vlan"] = command_line_args.psk_5g_wpa
|
|
if command_line_args.psk_2g_wpa2:
|
|
obj.psk_data["2g"]["wpa2"]["name"] = command_line_args.psk_2g_wpa2
|
|
obj.psk_data["2g"]["wpa2"]["nat"] = command_line_args.psk_2g_wpa2
|
|
obj.psk_data["2g"]["wpa2"]["vlan"] =command_line_args.psk_2g_wpa2
|
|
if command_line_args.psk_2g_wpa:
|
|
obj.psk_data["2g"]["wpa"]["name"] = command_line_args.psk_2g_wpa
|
|
obj.psk_data["2g"]["wpa"]["nat"] = command_line_args.psk_2g_wpa
|
|
obj.psk_data["2g"]["wpa"]["nat"] = command_line_args.psk_2g_wpa
|
|
if command_line_args.ssid_5g_wpa2:
|
|
obj.ssid_data["5g"]["wpa2"]["name"] = command_line_args.ssid_5g_wpa2
|
|
obj.ssid_data["5g"]["wpa2"]["nat"] = command_line_args.ssid_5g_wpa2
|
|
obj.ssid_data["5g"]["wpa2"]["vlan"] = command_line_args.ssid_5g_wpa2
|
|
if command_line_args.ssid_5g_wpa:
|
|
obj.ssid_data["5g"]["wpa"]["name"] = command_line_args.ssid_5g_wpa
|
|
obj.ssid_data["5g"]["wpa"]["nat"] = command_line_args.ssid_5g_wpa
|
|
obj.ssid_data["5g"]["wpa"]["vlan"] = command_line_args.ssid_5g_wpa
|
|
if command_line_args.ssid_2g_wpa2:
|
|
obj.ssid_data["2g"]["wpa2"]["name"] = command_line_args.ssid_2g_wpa2
|
|
obj.ssid_data["2g"]["wpa2"]["nat"] = command_line_args.ssid_2g_wpa2
|
|
obj.ssid_data["2g"]["wpa2"]["vlan"] = command_line_args.ssid_2g_wpa2
|
|
if command_line_args.ssid_2g_wpa:
|
|
obj.ssid_data["2g"]["wpa"]["name"] = command_line_args.ssid_2g_wpa
|
|
obj.ssid_data["2g"]["wpa"]["nat"] = command_line_args.ssid_2g_wpa
|
|
obj.ssid_data["2g"]["wpa"]["vlan"] = command_line_args.ssid_2g_wpa
|
|
|
|
print("creating Profiles")
|
|
ssid_template = "TipWlan-Cloud-Wifi"
|
|
|
|
|
|
|
|
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_profiles(ssid_template=ssid_template, 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")
|
|
|
|
main()
|
|
|