#!/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]") 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 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) # 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) 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 # obj.create_radius_profile(radius_name, rid, key) pass ap_object.create_ssid_profiles(ssid_template=ssid_template, skip_eap=True, 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()