wlan-testing/tests/fixtures_3x.py

import allure
import pytest
import os
import sys
""" Environment Paths """
if "libs" not in sys.path:
    sys.path.append(f'../libs')
for folder in 'py-json', 'py-scripts':
    if folder not in sys.path:
        sys.path.append(f'../lanforge/lanforge-scripts/{folder}')

sys.path.append(
    os.path.dirname(
        os.path.realpath(__file__)
    )
)
sys.path.append(f"../lanforge/lanforge-scripts/py-scripts/tip-cicd-sanity")

sys.path.append(f'../libs')
sys.path.append(f'../libs/lanforge/')
sys.path.append(f'../lanforge/lanforge-scripts')

from LANforge.LFUtils import *

if 'py-json' not in sys.path:
    sys.path.append('../py-scripts')
from controller.controller_3x.controller import CController


class Fixtures_3x:

    def __init__(self, configuration={}, run_lf=False, cc_1=False):
        self.lab_info = configuration
        self.run_lf = run_lf
        self.cc_1 = cc_1
        # print(self.lab_info)
        print("cc.1")
        self.controller_obj = ""


    def setup_profiles(self, request, param, run_lf, instantiate_profile, get_configuration, get_markers, lf_tools, lf_reports):
        table1= []
        instantiate_profile_obj = instantiate_profile(controller_data=get_configuration['controller'],
                                                  timeout="10", ssid_data=None,
                                                  ap_data=self.lab_info['access_point'], type=0)

        instantiate_profile_obj.enable_all_bands()


        if run_lf:
            return 0
        print("check params")
        # gives parameter value of setup_params_general
        parameter = dict(param)
        print("parameter", parameter)

        test_cases = {}
        profile_data= {}
        var = ""
        list_key = list(parameter.keys())

        if parameter['mode'] not in ["BRIDGE", "NAT", "VLAN"]:
            print("Invalid Mode: ", parameter['mode'])
            return test_cases
        profile_data["ssid"] = {}
        lf_dut_data = []
        for i in parameter["ssid_modes"]:
            profile_data["ssid"][i] = []
            for j in range(len(parameter["ssid_modes"][i])):
                data = parameter["ssid_modes"][i][j]
                profile_data["ssid"][i].append(data)

        # profile data will give ssid data like {'ssid': {'wpa2_personal': [{'ssid_name': 'ssid_wpa2_2g', 'appliedRadios': ['2G'], 'security_key': 'something'}, {'ssid_name': 'ssid_wpa2_5g', 'appliedRadios': ['5G'], 'security_key': 'something'}], 'wpa3_personal': [{'ssid_name': 'ssid_wpa2_5g', 'appliedRadios': ['6G'], 'security_key': 'something'}]}}
        print("profile data", profile_data)


        # create wlan
        for mode in profile_data['ssid']:
            if mode == "wpa2_personal":
                for j in profile_data["ssid"][mode]:
                    if mode in get_markers.keys() and get_markers[mode]:
                        try:
                            if j["appliedRadios"].__contains__("2G"):
                                lf_dut_data.append(j)
                            if j["appliedRadios"].__contains__("5G"):
                                lf_dut_data.append(j)
                            if j["appliedRadios"].__contains__("6G"):
                                lf_dut_data.append(j)
                            j["appliedRadios"] = list(set(j["appliedRadios"]))
                            j['security'] = 'wpa2'
                            # print("dut data", lf_dut_data)
                            test_cases["wpa_2g"] = True
                        except Exception as e:
                            print(e)
                            test_cases["wpa2_personal"] = False
            if mode == "wpa3_personal":
                for j in profile_data["ssid"][mode]:
                    if mode in get_markers.keys() and get_markers[mode]:
                        print( profile_data["ssid"][mode])
                        print(j)
                        try:
                            if j["appliedRadios"].__contains__("2G"):
                                lf_dut_data.append(j)
                            if j["appliedRadios"].__contains__("5G"):
                                lf_dut_data.append(j)
                            if j["appliedRadios"].__contains__("6G"):
                                lf_dut_data.append(j)
                                print(lf_dut_data)
                            j["appliedRadios"] = list(set(j["appliedRadios"]))
                            j['security'] = 'wpa3'
                            test_cases["wpa_2g"] = True
                        except Exception as e:
                            print(e)
                            test_cases["wpa3_personal"] = False

        # lf dut data [{'ssid_name': 'ssid_wpa2_2g', 'appliedRadios': ['2G'], 'security_key': 'something', 'security': 'wpa2'}, {'ssid_name': 'ssid_wpa2_5g', 'appliedRadios': ['5G'], 'security_key': 'something', 'security': 'wpa2'}, {'ssid_name': 'ssid_wpa2_5g', 'appliedRadios': ['6G'], 'security_key': 'something', 'security': 'wpa3'}]
        print("lf dut data", lf_dut_data)
        allure.attach(name="wlan data passing", body=str(parameter))
        ap = instantiate_profile_obj.show_ap_summary()
        print("show ap summ", ap)
        allure.attach(name="show ap summary", body=str(ap))

        print("create 3 wlans on slot1,2 and 3")
        for ap_name in range(len(self.lab_info['access_point'])):
            print("ap ", ap_name)
            # instantiate controller object

            print("set channel and bandwidth")
            instantiate_profile_obj = instantiate_profile(controller_data=get_configuration['controller'],
                                                          timeout="10", ssid_data=lf_dut_data,
                                                          ap_data=self.lab_info['access_point'], type=ap_name)
            instantiate_profile_obj.set_channel(band="6g", channel=self.lab_info['access_point'][ap_name]["channel_6g"],
                                                slot=self.lab_info['access_point'][ap_name]["6g_slot"])
            allure.attach(name="set 6g channel on " + str(ap_name + 1) + " ap ", body=str(self.lab_info['access_point'][ap_name]["channel_6g"]))
            instantiate_profile_obj.set_channel(band="5g", channel=self.lab_info['access_point'][ap_name]["channel_5g"],
                                                slot=self.lab_info['access_point'][ap_name]["5g_slot"])
            allure.attach(name="set 5g channel on " + str(ap_name + 1) + " ap ", body=str(self.lab_info['access_point'][ap_name]["channel_5g"]))
            instantiate_profile_obj.set_channel(band="2g", channel=self.lab_info['access_point'][ap_name]["channel_2g"],
                                                slot=self.lab_info['access_point'][ap_name]["2g_slot"])
            allure.attach(name="set 2g channel on " + str(ap_name + 1) + " ap ", body=str(self.lab_info['access_point'][ap_name]["channel_2g"]))
            print(self.lab_info['access_point'][ap_name]["ap_name"])
            check_admin = instantiate_profile_obj.check_admin_state_2ghz(ap_name=self.lab_info['access_point'][ap_name]["ap_name"])
            allure.attach(name="2ghz ap admin state for " + str(ap_name + 1) + " ap", body=str(check_admin))

            check_admin_5 = instantiate_profile_obj.check_admin_state_5ghz(ap_name=self.lab_info['access_point'][ap_name]["ap_name"])
            allure.attach(name="5ghz ap admin state for " + str(ap_name + 1) + " ap", body=str(check_admin_5))

            check_admin_6 = instantiate_profile_obj.check_admin_state_6ghz(ap_name=self.lab_info['access_point'][ap_name]["ap_name"])
            allure.attach(name="6ghz ap admin state for " + str(ap_name + 1) + " ap", body=str(check_admin_6))

            # table1.append(tab)


            if ap_name == 0:
                for band in range(len(lf_dut_data)):
                    if lf_dut_data[band]["appliedRadios"] == ["2G"]:
                        instantiate_profile_obj.no_logging_console()
                        instantiate_profile_obj.line_console()
                        id_slot = instantiate_profile_obj.get_slot_id_wlan()
                        ssid_name = instantiate_profile_obj.get_ssid_name_on_id()
                        if id_slot[0] == "1":
                            instantiate_profile_obj.show_shutdown_2ghz_ap()
                            instantiate_profile_obj.disable_wlan(wlan=ssid_name[0])
                            instantiate_profile_obj.ap_2ghz_shutdown()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.delete_wlan(ssid=ssid_name[0])
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa2(id="1", wlan=lf_dut_data[0]['ssid_name'],
                                                                     wlanssid=lf_dut_data[0]['ssid_name'],
                                                                     key=lf_dut_data[0]['security_key'])
                        else:
                            print(lf_dut_data[0]['ssid_name'])
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.show_shutdown_2ghz_ap()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa2(id="1", wlan=lf_dut_data[0]['ssid_name'],
                                                                     wlanssid=lf_dut_data[0]['ssid_name'],
                                                                     key=lf_dut_data[0]['security_key'])
                            instantiate_profile_obj.get_ssids()

                        if "roam_type" not in list_key:
                            # add wlan to single ap
                            instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(wlan=lf_dut_data[0]['ssid_name'])
                        instantiate_profile_obj.enable_wlan( wlan=lf_dut_data[0]['ssid_name'])
                        instantiate_profile_obj.enable_2ghz_netwrk(id="1", wlan=lf_dut_data[0]['ssid_name'],
                                                                   wlanssid=lf_dut_data[0]['ssid_name'],
                                                                   key=lf_dut_data[0]['security_key'])
                        instantiate_profile_obj.enable_ap_2ghz()
                        if parameter["ft+psk"] == True:
                            instantiate_profile_obj.enable_ft_psk(ssid=lf_dut_data[0]['ssid_name'],
                                                                  key=lf_dut_data[0]['security_key'])
                        if parameter["ft-otd"] == True:
                            instantiate_profile_obj.enable_ftotd_psk(ssid=lf_dut_data[0]['ssid_name'],
                                                                  key=lf_dut_data[0]['security_key'])

                        if 'dtim' in list_key:
                            instantiate_profile_obj.set_dtim_2ghz(wlan=lf_dut_data[0]['ssid_name'], value=parameter["dtim"])
                        instantiate_profile_obj.get_ssids()

                    if lf_dut_data[band]["appliedRadios"] == ["5G"]:
                        instantiate_profile_obj.no_logging_console()
                        instantiate_profile_obj.line_console()
                        id_slot = instantiate_profile_obj.get_slot_id_wlan()
                        print(id_slot)
                        ssid_name = instantiate_profile_obj.get_ssid_name_on_id()
                        print(ssid_name)
                        if id_slot[1] == "2":
                            # if ssid present on slot 2 delete it and create new
                            instantiate_profile_obj.show_shutdown_5ghz_ap()
                            instantiate_profile_obj.disable_wlan(wlan=ssid_name[1])
                            instantiate_profile_obj.ap_5ghz_shutdown()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.delete_wlan(ssid=ssid_name[1])
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa2(id="2", wlan=lf_dut_data[1]['ssid_name'],
                                                                     wlanssid=lf_dut_data[1]['ssid_name'],
                                                                     key=lf_dut_data[1]['security_key'])
                        else:
                            print(lf_dut_data[1]['ssid_name'])
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.show_shutdown_5ghz_ap()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa2(id="2", wlan=lf_dut_data[1]['ssid_name'],
                                                                     wlanssid=lf_dut_data[1]['ssid_name'],
                                                                     key=lf_dut_data[1]['security_key'])
                            instantiate_profile_obj.get_ssids()

                        if "roam_type" not in list_key:
                            instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(wlan=lf_dut_data[1]['ssid_name'])
                        instantiate_profile_obj.enable_wlan(wlan=lf_dut_data[1]['ssid_name'])
                        instantiate_profile_obj.enable_5ghz_netwrk(id="2", wlan=lf_dut_data[1]['ssid_name'],
                                                                   wlanssid=lf_dut_data[1]['ssid_name'],
                                                                   key=lf_dut_data[1]['security_key'])
                        instantiate_profile_obj.enable_ap_5ghz()
                        if parameter["ft+psk"] == True:
                            instantiate_profile_obj.enable_ft_psk(ssid=lf_dut_data[1]['ssid_name'],
                                                                  key=lf_dut_data[1]['security_key'])
                        if parameter["ft-otd"] == True:
                            instantiate_profile_obj.enable_ftotd_psk(ssid=lf_dut_data[1]['ssid_name'],
                                                                  key=lf_dut_data[1]['security_key'])
                        if 'dtim' in list_key:
                            instantiate_profile_obj.set_dtim_5ghz(wlan=lf_dut_data[1]['ssid_name'], value=parameter["dtim"])

                        instantiate_profile_obj.get_ssids()
                    if lf_dut_data[band]["appliedRadios"] == ["6G"]:
                        instantiate_profile_obj.no_logging_console()
                        instantiate_profile_obj.line_console()
                        id_slot = instantiate_profile_obj.get_slot_id_wlan()
                        print(id_slot)
                        ssid_name = instantiate_profile_obj.get_ssid_name_on_id()
                        print(ssid_name)
                        if id_slot[2] == "3":
                            instantiate_profile_obj.show_shutdown_6ghz_ap()
                            # instantiate_profile_obj.show_shutdown_5ghz_ap()
                            # instantiate_profile_obj.show_shutdown_2ghz_ap()
                            instantiate_profile_obj.disable_wlan(wlan=ssid_name[2])
                            instantiate_profile_obj.ap_6ghz_shutdown()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.delete_wlan(ssid=ssid_name[2])
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa3(id="3", wlan=lf_dut_data[2]['ssid_name'],
                                                                              wlanssid=lf_dut_data[2]['ssid_name'],
                                                                              key=lf_dut_data[2]['security_key'])
                        else:
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.show_shutdown_6ghz_ap()
                            instantiate_profile_obj.get_ssids()
                            instantiate_profile_obj.create_wlan_wpa3(id="3", wlan=lf_dut_data[2]['ssid_name'],
                                                                     wlanssid=lf_dut_data[2]['ssid_name'],
                                                                     key=lf_dut_data[2]['security_key'])
                            instantiate_profile_obj.get_ssids()

                        if "roam_type" not in list_key:
                            instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(
                                wlan=lf_dut_data[2]['ssid_name'])
                        instantiate_profile_obj.enable_wlan( wlan=lf_dut_data[2]['ssid_name'])
                        instantiate_profile_obj.enable_6ghz_netwrk(id="3", wlan=lf_dut_data[2]['ssid_name'],
                                                                   wlanssid=lf_dut_data[2]['ssid_name'],
                                                                   key=lf_dut_data[2]['security_key'])
                        instantiate_profile_obj.enable_ap_6ghz()
                        # if parameter["ft+psk"] == True:
                        #      instantiate_profile_obj.enable_ft_sae(ssid=lf_dut_data[2]['ssid_name'], key=lf_dut_data[2]['security_key'])
                        if parameter["ft-dot1x"] == True:
                            instantiate_profile_obj.enable_ft_dot1x_wpa3(ssid=lf_dut_data[2]['ssid_name'])
                        if parameter["ft-dot1x_sha256"] == True:
                            instantiate_profile_obj.enable_ft_dot1x_sha256_wpa3(ssid=lf_dut_data[2]['ssid_name'],
                                                                                radius=get_configuration['controller']["radius"])
                        instantiate_profile_obj.get_ssids()

        twog_sum = instantiate_profile_obj.show_2ghz_summary()
        fiveg_sum = instantiate_profile_obj.show_5ghz_summary()
        sixg_sum = instantiate_profile_obj.show_6ghz_summary()
        allure.attach(name="ap admin state 2ghz ", body=str(twog_sum))
        allure.attach(name="ap admin state 5ghz ", body=str(fiveg_sum))
        allure.attach(name="ap admin state 6ghz", body=str(sixg_sum))
        sumy = instantiate_profile_obj.get_ssids()
        allure.attach(name="wlan summary after creating test wlan", body=str(sumy))
        if "roam_type" in list_key:
            print("after creating wlan's assign wlan to respective tag policy")
            for ap_name in range(len(self.lab_info['access_point'])):
                print("ap ", ap_name)
                instantiate_profile_obj = instantiate_profile(controller_data=get_configuration['controller'], timeout="10",
                                                              ssid_data=lf_dut_data, ap_data=self.lab_info['access_point'],
                                                              type=ap_name)
                if parameter["roam_type"] == "fiveg_to_fiveg":
                    # add 5g wlan to both ap's
                    instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(wlan=lf_dut_data[1]['ssid_name'])
                if parameter["roam_type"] == "fiveg_to_sixg":
                    if ap_name == 0:
                        # add 5g ssid to 5g ap
                        instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(wlan=lf_dut_data[1]['ssid_name'])
                    if ap_name == 1:
                        # add 6g ssid to 6g ap
                        instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(wlan=lf_dut_data[2]['ssid_name'])

                else:
                    instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(
                        wlan=lf_dut_data[0]['ssid_name'])
                    instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(
                        wlan=lf_dut_data[1]['ssid_name'])
                    instantiate_profile_obj.config_wireless_tag_policy_and_policy_profile(
                        wlan=lf_dut_data[2]['ssid_name'])




        bssid_list_2g = []
        bssid_list_5g = []
        ssid_data_list = []

        for ap_name in range(len(self.lab_info['access_point'])):
            print("ap ", ap_name)
            # instantiate controller object
            instantiate_profile_obj = instantiate_profile(controller_data=get_configuration['controller'], timeout="10",
                                                      ssid_data=lf_dut_data, ap_data=self.lab_info['access_point'], type=ap_name)
            bss2_info = instantiate_profile_obj.get_ap_bssid_2g()
            allure.attach(name="wlan 2g bssid info", body=str(bss2_info))
            bss5_info = instantiate_profile_obj.get_ap_bssid_5g()
            allure.attach(name="wlan 5g bssid info", body=str(bss5_info))
            bss6_info = instantiate_profile_obj.get_ap_bssid_6g()
            allure.attach(name="wlan 6g bssid info", body=str(bss6_info))

            bssid_2g = instantiate_profile_obj.cal_bssid_2g()
            print("bssid 2g", bssid_2g)
            lst_2g = bssid_list_2g.append(bssid_2g)

            bssid_5g = instantiate_profile_obj.cal_bssid_5g()
            print("bssid 5g ", bssid_5g)
            lst_5g = bssid_list_5g.append(bssid_5g)
            # print(bssid_5g)
            # print(bssid_list_2g)
            # print(bssid_list_5g)
            ssid_data = []
            try:
                idx_mapping = {}
                bssid = ""
                for interface in range(len(lf_dut_data)):
                    if interface == 0:
                        bssid = bssid_2g
                    if interface == 1:
                        bssid = bssid_5g
                    if lf_dut_data[interface]['security'] == "psk2":
                        lf_dut_data[interface]['security'] = "WPA2"
                    ssid = ["ssid_idx=" + str(interface) +
                            " ssid=" + lf_dut_data[interface]['ssid_name'] +
                            " security=" + lf_dut_data[interface]['security'] +
                            " password=" + lf_dut_data[interface]['security_key'] +
                            " bssid=" + bssid
                            ]
                    ssid_data.append(ssid)

            except Exception as e:
                print(e)
                pass
            # print("nikita",ssid_data)
            ssid_data_list.append(ssid_data)
        print("final ssid data", ssid_data_list)
        # ssid_data = [[['ssid_idx=0 ssid=ssid_wpa2_2g security=WPA2 password=something bssid=68:7d:b4:5f:5c:30'],
        #               ['ssid_idx=1 ssid=ssid_wpa2_5g security=WPA2 password=something bssid=68:7d:b4:5f:5c:3e']],
        #              [['ssid_idx=0 ssid=ssid_wpa2_2g security=WPA2 password=something bssid=14:16:9d:53:58:c0'],
        #               ['ssid_idx=1 ssid=ssid_wpa2_5g security=WPA2 password=something bssid=14:16:9d:53:58:ce']]]
        lf_tools.create_non_meh_dut(ssid_data=ssid_data_list)
        table1 = [["show ap summary", "done"], ["Configure wlan", "done"], ["configure channel/width", "done"],
                  ["ap admin state up", "done"], ["checking admin state of ap", "done"],
                  ["sniffer capture", "done"],
                  ["client connect", "don"], ["roam", "done"], ["sniffer verification", "done"],
                  ["iteration", "done"],
                  ["table display", "done"], ["check in controller client connected", "done"],
                  ["Bring down unused band before start of testcase", "done"]]
        tab1 = lf_reports.table2(table=table1)
        allure.attach(name="skeleton of code", body=str(tab1))