mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-11-02 20:07:57 +00:00
210 lines
9.2 KiB
Python
Executable File
210 lines
9.2 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", default=False)
|
|
parser.add_argument("--skip-wpa", dest="skip_wpa", action='store_true',
|
|
help="Should we skip the WPA ssid or not", default=False)
|
|
parser.add_argument("--skip-wpa2", dest="skip_wpa2", action='store_true',
|
|
help="Should we skip the WPA2 ssid or not", default=False)
|
|
parser.add_argument("--skip-profiles", dest="skip_profiles", action='store_true',
|
|
help="Should we skip creating new ssid profiles?", default=False)
|
|
|
|
parser.add_argument("--psk-5g-wpa2", dest="psk_5g_wpa2", type=str,
|
|
help="Allow over-riding the 5g-wpa2 PSK value.")
|
|
parser.add_argument("--psk-5g-wpa", dest="psk_5g_wpa", type=str,
|
|
help="Allow over-riding the 5g-wpa PSK value.")
|
|
parser.add_argument("--psk-2g-wpa2", dest="psk_2g_wpa2", type=str,
|
|
help="Allow over-riding the 2g-wpa2 PSK value.")
|
|
parser.add_argument("--psk-2g-wpa", dest="psk_2g_wpa", type=str,
|
|
help="Allow over-riding the 2g-wpa PSK value.")
|
|
|
|
parser.add_argument("--ssid-5g-wpa2", dest="ssid_5g_wpa2", type=str,
|
|
help="Allow over-riding the 5g-wpa2 SSID value.")
|
|
parser.add_argument("--ssid-5g-wpa", dest="ssid_5g_wpa", type=str,
|
|
help="Allow over-riding the 5g-wpa SSID value.")
|
|
parser.add_argument("--ssid-2g-wpa2", dest="ssid_2g_wpa2", type=str,
|
|
help="Allow over-riding the 2g-wpa2 SSID value.")
|
|
parser.add_argument("--ssid-2g-wpa", dest="ssid_2g_wpa", type=str,
|
|
help="Allow over-riding the 2g-wpa SSID value.")
|
|
|
|
parser.add_argument("--mode", dest="mode", choices=['bridge', 'nat', 'vlan'], type=str,
|
|
help="Mode of AP Profile [bridge/nat/vlan]", required=True)
|
|
|
|
# Not implemented yet.
|
|
#parser.add_argument("--rf-mode", type=str,
|
|
# choices=["modeN", "modeAC", "modeGN", "modeX", "modeA", "modeB", "modeG", "modeAB"],
|
|
# help="Allow over-riding the 2g-wpa SSID value.")
|
|
|
|
|
|
reporting = Reporting(reports_root=os.getcwd() + "/reports/")
|
|
|
|
base = UnitTestBase("skd-set-profile", parser)
|
|
|
|
command_line_args = base.command_line_args
|
|
print(command_line_args.mode)
|
|
|
|
# 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
|
|
|
|
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(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)
|
|
|
|
# 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")
|
|
|
|
|
|
|
|
args = command_line_args
|
|
|
|
print("Profiles Created")
|
|
|
|
ap_object = CreateAPProfiles(args, cloud=cloud, client=client, fw_model=fw_model)
|
|
|
|
# Logic to create AP Profiles (Bridge Mode)
|
|
|
|
ap_object.set_ssid_psk_data(ssid_2g_wpa=args.ssid_2g_wpa,
|
|
ssid_5g_wpa=args.ssid_5g_wpa,
|
|
psk_2g_wpa=args.psk_2g_wpa,
|
|
psk_5g_wpa=args.psk_5g_wpa,
|
|
ssid_2g_wpa2=args.ssid_2g_wpa2,
|
|
ssid_5g_wpa2=args.ssid_5g_wpa2,
|
|
psk_2g_wpa2=args.psk_2g_wpa2,
|
|
psk_5g_wpa2=args.psk_5g_wpa2)
|
|
|
|
print(ap_object)
|
|
rid = client.get_run_id(test_run_name=args.testrail_run_prefix + fw_model + "_" + today + "_" + "ecw5410-2021-02-12-pending-e8bb466")
|
|
print("creating Profiles")
|
|
ssid_template = "TipWlan-Cloud-Wifi"
|
|
|
|
if not args.skip_profiles:
|
|
if not args.skip_radius:
|
|
# Radius Profile needs to be set here
|
|
radius_name = "Test_Radius" + str(time.time()).split(".")[0]
|
|
radius_template = "templates/radius_profile_template.json"
|
|
ap_object.create_radius_profile(radius_name=radius_name, radius_template=radius_template, rid=rid, key=fw_model)
|
|
|
|
ap_object.create_ssid_profiles(ssid_template=ssid_template, skip_eap=args.skip_radius, skip_wpa=args.skip_wpa,
|
|
skip_wpa2=args.skip_wpa2, mode=args.mode)
|
|
|
|
print("Create AP with equipment-id: ", equipment_id)
|
|
ap_object.create_ap_profile(eq_id=equipment_id, fw_model=fw_model, mode=args.mode)
|
|
ap_object.validate_changes(mode=args.mode)
|
|
|
|
print("Profiles Created")
|
|
|
|
|
|
main()
|
|
|