# # import files in the current directory
# import datetime
# import sys
# import os
# import time
#
# sys.path.append(
#     os.path.dirname(
#         os.path.realpath(__file__)
#     )
# )
#
# if 'cloudsdk' not in sys.path:
#     sys.path.append(f'../libs/cloudsdk')
# if 'apnos' not in sys.path:
#     sys.path.append(f'../libs/apnos')
# if 'testrails' not in sys.path:
#     sys.path.append(f'../libs/testrails')
#
# from apnos import APNOS
# from cloudsdk import CloudSDK
# from cloudsdk import ProfileUtility
# from cloudsdk import FirmwareUtility
# import pytest
# import logging
# from configuration import APNOS_CREDENTIAL_DATA
# from configuration import RADIUS_SERVER_DATA
# from configuration import TEST_CASES
# from configuration import NOLA
# from testrail_api import APIClient
# from reporting import Reporting
#
#
# def pytest_addoption(parser):
#     parser.addini("force-upload", "firmware-upload option")
#
#     parser.addini("jfrog-base-url", "jfrog base url")
#     parser.addini("jfrog-user-id", "jfrog username")
#     parser.addini("jfrog-user-password", "jfrog password")
#
#     parser.addini("testbed-name", "cloud sdk base url")
#     parser.addini("equipment-model", "Equipment Model")
#
#     parser.addini("sdk-user-id", "cloud sdk username")
#     parser.addini("sdk-user-password", "cloud sdk user password")
#
#     parser.addini("sdk-customer-id", "cloud sdk customer id for the access points")
#     parser.addini("sdk-equipment-id", "cloud sdk customer id for the access points")
#
#     parser.addini("testrail-base-url", "testrail base url")
#     parser.addini("testrail-project", "testrail project name to use to generate test reports")
#     parser.addini("testrail-user-id", "testrail username")
#     parser.addini("testrail-user-password", "testrail user password")
#
#     parser.addini("lanforge-ip-address", "LANforge ip address to connect to")
#     parser.addini("lanforge-port-number", "LANforge port number to connect to")
#     parser.addini("lanforge-bridge-port", "LANforge port for bridge mode testing")
#     parser.addini("lanforge-2dot4g-prefix", "LANforge 2.4g prefix")
#     parser.addini("lanforge-5g-prefix", "LANforge 5g prefix")
#     parser.addini("lanforge-2dot4g-station", "LANforge station name for 2.4g")
#     parser.addini("lanforge-5g-station", "LANforge station name for 5g")
#     parser.addini("lanforge-2dot4g-radio", "LANforge radio for 2.4g")
#     parser.addini("lanforge-5g-radio", "LANforge radio for 5g")
#
#     parser.addini("jumphost_ip", "APNOS Jumphost IP Address")
#     parser.addini("jumphost_username", "APNOS Jumphost Username")
#     parser.addini("jumphost_password", "APNOS Jumphost password")
#     parser.addini("jumphost_port", "APNOS Jumphost ssh Port")
#
#     parser.addini("skip-open", "skip open ssid mode")
#     parser.addini("skip-wpa", "skip wpa ssid mode")
#     parser.addini("skip-wpa2", "skip wpa2 ssid mode")
#     parser.addini("skip-eap", "skip eap ssid mode")
#
#     parser.addini("radius_server_ip", "Radius server IP")
#     parser.addini("radius_port", "Radius Port")
#     parser.addini("radius_secret", "Radius shared Secret")
#
#     parser.addini("tr_url", "Test Rail URL")
#     parser.addini("tr_prefix", "Test Rail Prefix (Generally Testbed_name_)")
#     parser.addini("tr_user", "Testrail Username")
#     parser.addini("tr_pass", "Testrail Password")
#     parser.addini("tr_project_id", "Testrail Project ID")
#     parser.addini("milestone", "milestone Id")
#
#     parser.addini("num_stations", "Number of Stations/Clients for testing")
#
#     # change behaviour
#     parser.addoption(
#         "--skip-upgrade",
#         action="store_true",
#         default=False,
#         help="skip updating firmware on the AP (useful for local testing)"
#     )
#     # change behaviour
#     parser.addoption(
#         "--force-upgrade",
#         action="store_true",
#         default=False,
#         help="force Upgrading Firmware even if it is already latest version"
#     )
#     parser.addoption(
#         "--force-upload",
#         action="store_true",
#         default=False,
#         help="force Uploading Firmware even if it is already latest version"
#     )
#     # this has to be the last argument
#     # example: --access-points ECW5410 EA8300-EU
#     parser.addoption(
#         "--model",
#         # nargs="+",
#         default="ecw5410",
#         help="AP Model which is needed to test"
#     )
#     parser.addoption(
#         "--skip-testrail",
#         action="store_true",
#         default=False,
#         help="Stop using Testrails"
#     )
#
#
# """
# Test session base fixture
# """
#
#
# @pytest.fixture(scope="session")
# def testrun_session(request):
#     var = request.config.getoption("model")
#     yield var
#
#
# """
# Instantiate Objects for Test session
# """
#
#
# @pytest.fixture(scope="session")
# def instantiate_cloudsdk(testrun_session):
#     try:
#         sdk_client = CloudSDK(testbed=NOLA[testrun_session]["cloudsdk_url"],
#                               customer_id=NOLA[testrun_session]["customer_id"])
#     except:
#         sdk_client = False
#     yield sdk_client
#
#
# @pytest.fixture(scope="session")
# def instantiate_jFrog(request):
#     jfrog_cred = {
#         "user": request.config.getini("jfrog-user-id"),
#         "password": request.config.getini("jfrog-user-password")
#     }
#     yield jfrog_cred
#
#
# @pytest.fixture(scope="session")
# def instantiate_firmware(instantiate_cloudsdk, instantiate_jFrog):
#     try:
#         firmware_client = FirmwareUtility(jfrog_credentials=instantiate_jFrog, sdk_client=instantiate_cloudsdk)
#     except:
#         firmware_client = False
#     yield firmware_client
#
#
# @pytest.fixture(scope="session")
# def instantiate_profile(instantiate_cloudsdk):
#     try:
#         profile_object = ProfileUtility(sdk_client=instantiate_cloudsdk)
#     except:
#         profile_object = False
#     yield profile_object
#
#
# @pytest.fixture(scope="session")
# def instantiate_testrail(request):
#     if request.config.getoption("--skip-testrail"):
#         tr_client = Reporting()
#     else:
#         tr_client = APIClient(request.config.getini("tr_url"), request.config.getini("tr_user"),
#                               request.config.getini("tr_pass"), request.config.getini("tr_project_id"))
#     yield tr_client
#
#
# @pytest.fixture(scope="session")
# def instantiate_project(request, instantiate_testrail, testrun_session, get_latest_firmware):
#     if request.config.getoption("--skip-testrail"):
#         rid = "skip testrails"
#     else:
#         projId = instantiate_testrail.get_project_id(project_name=request.config.getini("tr_project_id"))
#         test_run_name = request.config.getini("tr_prefix") + testrun_session + "_" + str(
#             datetime.date.today()) + "_" + get_latest_firmware
#         instantiate_testrail.create_testrun(name=test_run_name, case_ids=list(TEST_CASES.values()), project_id=projId,
#                                             milestone_id=request.config.getini("milestone"),
#                                             description="Automated Nightly Sanity test run for new firmware build")
#         rid = instantiate_testrail.get_run_id(
#             test_run_name=request.config.getini("tr_prefix") + testrun_session + "_" + str(
#                 datetime.date.today()) + "_" + get_latest_firmware)
#     yield rid
#
#
# """
# Utility Fixtures
# """
#
#
# @pytest.fixture(scope="session")
# def get_equipment_id(testrun_session):
#     yield NOLA[testrun_session]["equipment_id"]
#
#
# @pytest.fixture(scope="session")
# def get_latest_firmware(testrun_session, instantiate_firmware):
#     try:
#         latest_firmware = instantiate_firmware.get_latest_fw_version(testrun_session)
#     except:
#         latest_firmware = False
#     yield latest_firmware
#
#
# @pytest.fixture(scope="session")
# def check_ap_firmware_ssh(request, testrun_session):
#     try:
#         ap_ssh = APNOS(APNOS_CREDENTIAL_DATA)
#         active_fw = ap_ssh.get_active_firmware()
#     except Exception as e:
#         active_fw = False
#     yield active_fw
#
#
# @pytest.fixture(scope="session")
# def check_ap_firmware_cloud(instantiate_cloudsdk, get_equipment_id):
#     yield instantiate_cloudsdk.get_ap_firmware_old_method(equipment_id=get_equipment_id)
#
#
# @pytest.fixture(scope="function")
# def get_ap_manager_status():
#     ap_ssh = APNOS(APNOS_CREDENTIAL_DATA)
#     status = ap_ssh.get_manager_state()
#     if "ACTIVE" not in status:
#         time.sleep(30)
#         ap_ssh = APNOS(APNOS_CREDENTIAL_DATA)
#         status = ap_ssh.get_manager_state()
#     yield status
#
#
# @pytest.fixture(scope="session")
# def should_upload_firmware(request):
#     yield request.config.getoption("--force-upload")
#
#
# @pytest.fixture(scope="session")
# def should_upgrade_firmware(request):
#     yield request.config.getoption("--force-upgrade")
#
#
# @pytest.fixture(scope="session")
# def upload_firmware(should_upload_firmware, instantiate_firmware, get_latest_firmware):
#     firmware_id = instantiate_firmware.upload_fw_on_cloud(fw_version=get_latest_firmware,
#                                                           force_upload=should_upload_firmware)
#     yield firmware_id
#
#
# @pytest.fixture(scope="session")
# def upgrade_firmware(request, instantiate_firmware, get_equipment_id, check_ap_firmware_cloud, get_latest_firmware,
#                      should_upgrade_firmware):
#     if get_latest_firmware != check_ap_firmware_cloud:
#         if request.config.getoption("--skip-upgrade"):
#             status = "skip-upgrade"
#         else:
#             status = instantiate_firmware.upgrade_fw(equipment_id=get_equipment_id, force_upload=False,
#                                                      force_upgrade=should_upgrade_firmware)
#     else:
#         if should_upgrade_firmware:
#             status = instantiate_firmware.upgrade_fw(equipment_id=get_equipment_id, force_upload=False,
#                                                      force_upgrade=should_upgrade_firmware)
#         else:
#             status = "skip-upgrade"
#     yield status
#
#
# @pytest.fixture(scope="session")
# def setup_profile_data(testrun_session):
#     profile_data = {}
#     for mode in "BRIDGE", "NAT", "VLAN":
#         profile_data[mode] = {}
#         for security in "OPEN", "WPA", "WPA2_P", "WPA2_E":
#             profile_data[mode][security] = {}
#             for radio in "2G", "5G":
#                 profile_data[mode][security][radio] = {}
#                 name_string = f"{'Sanity'}-{testrun_session}-{radio}_{security}_{mode}"
#                 passkey_string = f"{radio}-{security}_{mode}"
#                 profile_data[mode][security][radio]["profile_name"] = name_string
#                 profile_data[mode][security][radio]["ssid_name"] = name_string
#                 if mode == "VLAN":
#                     profile_data[mode][security][radio]["vlan"] = 100
#                 else:
#                     profile_data[mode][security][radio]["vlan"] = 1
#                 if mode != "NAT":
#                     profile_data[mode][security][radio]["mode"] = "BRIDGE"
#                 else:
#                     profile_data[mode][security][radio]["mode"] = "NAT"
#                 if security != "OPEN":
#                     profile_data[mode][security][radio]["security_key"] = passkey_string
#                 else:
#                     profile_data[mode][security][radio]["security_key"] = "[BLANK]"
#     yield profile_data
#
#
# """
# Profile Utility
# """
#
#
# @pytest.fixture(scope="class")
# def reset_profile(instantiate_profile):
#     instantiate_profile.profile_creation_ids["ssid"] = []
#     yield True
#
#
# @pytest.fixture(scope="function")
# def cleanup_cloud_profiles(instantiate_cloudsdk):
#     profile_object = ProfileUtility(sdk_client=instantiate_cloudsdk)
#     yield profile_object.cleanup_profiles()
#
#
# @pytest.fixture(scope="session")
# def create_radius_profile(instantiate_profile, testrun_session):
#     radius_info = {
#         "name": testrun_session + "-RADIUS-Sanity",
#         "ip": RADIUS_SERVER_DATA["ip"],
#         "port": RADIUS_SERVER_DATA["port"],
#         "secret": RADIUS_SERVER_DATA["secret"]
#     }
#     instantiate_profile.delete_profile_by_name(radius_info["name"])
#     instantiate_profile.get_default_profiles()
#     profile_info = instantiate_profile.create_radius_profile(radius_info=radius_info)
#     yield profile_info
#
#
# @pytest.fixture(scope="session")
# def set_rf_profile(instantiate_profile):
#     try:
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.set_rf_profile()
#     except:
#         profile = False
#     yield profile
#
#
# """
# BRIDGE MOde
# """
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_2g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_5g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_2g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA2_P']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_5g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA2_P']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_2g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA2_E']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, fiveg=False)
#     except Exception as e:
#         # (e)
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_5g_profile_bridge(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["BRIDGE"]['WPA2_E']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, two4g=False)
#     except Exception as e:
#         # (e)
#         profile = False
#     yield profile
#
#
# """
# NAT MOde
# """
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_2g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_5g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_2g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA2_P']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_5g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA2_P']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_2g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA2_E']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_5g_profile_nat(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["NAT"]['WPA2_E']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# """
# VLAN MOde
# """
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_2g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa_ssid_5g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_2g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA2_P']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_p_ssid_5g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA2_P']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_personal_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_2g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA2_E']['2G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, fiveg=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_wpa2_e_ssid_5g_profile_vlan(instantiate_profile, setup_profile_data):
#     try:
#         profile_data = setup_profile_data["VLAN"]['WPA2_E']['5G']
#         instantiate_profile.get_default_profiles()
#         profile = instantiate_profile.create_wpa2_enterprise_ssid_profile(profile_data=profile_data, two4g=False)
#     except:
#         profile = False
#     yield profile
#
#
# @pytest.fixture(scope="session")
# def create_ap_profile_bridge(instantiate_profile, testrun_session):
#     try:
#         profile_data = {
#             "profile_name": "%s-%s-%s" % ("Sanity", testrun_session, 'BRIDGE'),
#         }
#         profile_obj = instantiate_profile.set_ap_profile(profile_data=profile_data)
#     except Exception as e:
#         profile_obj = e
#     print(profile_obj)
#     yield profile_obj
#
#
# @pytest.fixture(scope="session")
# def create_ap_profile_nat(instantiate_profile, testrun_session):
#     profile_data = {
#         "profile_name": "%s-%s-%s" % ("Sanity", testrun_session, 'NAT'),
#     }
#     profile_obj = instantiate_profile.set_ap_profile(profile_data=profile_data)
#     yield profile_obj
#
#
# @pytest.fixture(scope="session")
# def create_ap_profile_vlan(instantiate_profile, testrun_session):
#     profile_data = {
#         "profile_name": "%s-%s-%s" % ("Sanity", testrun_session, 'VLAN'),
#     }
#     profile_obj = instantiate_profile.set_ap_profile(profile_data=profile_data)
#     yield profile_obj
#
#
# @pytest.fixture(scope="function")
# def get_current_profile_cloud(instantiate_profile):
#     ssid_names = []
#     print(instantiate_profile.profile_creation_ids["ssid"])
#     for i in instantiate_profile.profile_creation_ids["ssid"]:
#         ssid_names.append(instantiate_profile.get_ssid_name_by_profile_id(profile_id=i))
#
#     yield ssid_names
#
#
# """
# Profile Push Fixtures
# """
#
#
# @pytest.fixture(scope="function")
# def push_profile(instantiate_profile, get_equipment_id, setup_profile_data):
#     try:
#         instantiate_profile.push_profile_old_method(equipment_id=get_equipment_id)
#         status = True
#     except Exception as e:
#         status = False
#
#     yield status
#
#
# @pytest.fixture(scope="function")
# def get_lanforge_data(request):
#     lanforge_data = {
#         "lanforge_ip": request.config.getini("lanforge-ip-address"),
#         "lanforge-port-number": request.config.getini("lanforge-port-number"),
#         "lanforge_2dot4g": request.config.getini("lanforge-2dot4g-radio"),
#         "lanforge_5g": request.config.getini("lanforge-5g-radio"),
#         "lanforge_2dot4g_prefix": request.config.getini("lanforge-2dot4g-prefix"),
#         "lanforge_5g_prefix": request.config.getini("lanforge-5g-prefix"),
#         "lanforge_2dot4g_station": request.config.getini("lanforge-2dot4g-station"),
#         "lanforge_5g_station": request.config.getini("lanforge-5g-station"),
#         "lanforge_bridge_port": request.config.getini("lanforge-bridge-port"),
#         "lanforge_vlan_port": "eth1.100",
#         "vlan": 100
#     }
#     yield lanforge_data
#
#
# @pytest.fixture(scope="function")
# def update_ssid(request, instantiate_profile, setup_profile_data):
#     requested_profile = str(request.param).replace(" ","").split(",")
#     profile = setup_profile_data[requested_profile[0]][requested_profile[1]][requested_profile[2]]
#     status = instantiate_profile.update_ssid_name(profile_name=profile["profile_name"], new_profile_name=requested_profile[3])
#     setup_profile_data[requested_profile[0]][requested_profile[1]][requested_profile[2]]["profile_name"] = requested_profile[3]
#     setup_profile_data[requested_profile[0]][requested_profile[1]][requested_profile[2]]["ssid_name"] = requested_profile[3]
#     time.sleep(90)
#     yield status