Modified the empsk_test method

Signed-off-by: jitendra-kushavah <jitendra.kushavah@candelatech.com>

lf_libs/lf_tests.py

@@ -74,6 +74,8 @@ wifi_mobility_test = importlib.import_module("py-scripts.lf_wifi_mobility_test")
 WifiMobility = wifi_mobility_test.WifiMobility
 modify_station = importlib.import_module("py-scripts.modify_station")
 ModifyStation = modify_station.ModifyStation
+station_profile = importlib.import_module("py-json.station_profile")
+StationProfile = station_profile.StationProfile
 
 
 class lf_tests(lf_libs):
@@ -969,25 +971,52 @@ class lf_tests(lf_libs):
         else:
             logging.info("ALL Stations got IP as Expected")
 
-
-
     def empsk_test(self,ssid="[BLANK]", passkey="[BLANK]", security="wpa2", mode="BRIDGE", band="twog",
-                       vlan_id=[None], num_sta=None, scan_ssid=True, client_type=0, pre_cleanup=True,
+                       num_sta=None, scan_ssid=True, client_type=0, pre_cleanup=True,
-                       station_data=["ip", "alias", "mac", "channel", "port type", "security", "ap", "parent dev"],
+                       allure_attach=True, identifier=None, allure_name="station data", dut_data={},
-                       allure_attach=True, identifier=None, allure_name="station data", dut_data={}):
+                        extra_securities = []):
 
-        sta_data = self.client_connect(ssid=ssid, passkey=passkey, security=security,
+        if pre_cleanup:
-                                                   mode=mode, band=band, pre_cleanup=False, num_sta=num_sta,
+            self.pre_cleanup()
-                                                   scan_ssid=True,
+        self.check_band_ap(band=band)
-                                                   station_data=station_data,
+        if self.run_lf:
-                                                   allure_attach=True, dut_data=dut_data, allure_name=allure_name)
+            dut_data = self.run_lf_dut_data()
+        logging.info("DUT Data:\n" + json.dumps(str(dut_data), indent=2))
+        allure.attach(name="DUT Data:\n", body=json.dumps(str(dut_data), indent=2),
+                      attachment_type=allure.attachment_type.JSON)
 
-        logging.info(f"sta_data from client_connect:{sta_data}")
+        dict_all_radios_6g = {"be200_radios": self.be200_radios, "ax210_radios": self.ax210_radios}
-        station_list = []
+        logging.info(f"dict_all_radios_6g:{dict_all_radios_6g}")
-        for sta in sta_data.keys():
+        radio = dict_all_radios_6g['be200_radios'][0]
-            station_list.append(sta)
+        logging.info("creating station profile obj")
-            radio = sta_data[sta]["parent dev"]
+        obj_sta_profile = StationProfile(self.local_realm.lfclient_url, self.local_realm)
+        obj_sta_profile.add_sta_data = {
+            "shelf": 1,
+            "resource": 1,
+            "radio": radio,
+            "sta_name": 'sta0000',
+            "ssid": "OpenWifi-roam",
+            "key": "aaaaaaaa",
+            "mode": 0,
+            "mac": "xx:xx:xx:xx:*:xx",
+            "flags": 1099511628800, # enable wpa3 and wpa2
+            "flags_mask": 0
+        }
+        if extra_securities:
+            logging.info(f"extra_securities are provided:{extra_securities}")
+            if "wpa3" in extra_securities:
+                logging.info("trying to enable wpa3 security also")
+                obj_sta_profile.add_security_extra(security="wpa3")
+        logging.info(f"creating station profile")
+        obj_sta_profile.create(radio=radio, num_stations=1)
+        time.sleep(30)
+        sta = "sta0000"
+        sta_data = self.json_get(_req_url="port/1/1/%s" % sta)
+        self.allure_report_table_format(dict_data=sta_data["interface"], key="Station Data",
+                                        value="Value", name="station data for 2G band")
+        logging.info(f"sta_data::{sta_data}")
 
+        station_list = ["1.1.sta0000"]
         enable_flag = ["use-wpa3"]
         disable_flag = ["wpa2_enable"]
         obj_modify_sta = ModifyStation(_host=self.manager_ip, _port=self.manager_http_port,
@@ -1002,42 +1031,37 @@ class lf_tests(lf_libs):
         obj_modify_sta.set_station()
         logging.info("Successfully changed encryption from WPA2 to WPA3")
         time.sleep(10)
-        stat_data = self.get_station_data(sta_name=station_list, rows=station_data, allure_attach=False)
 
-        sta_table_dict = {}
+        sta_data = self.json_get(_req_url="port/1/1/%s" % sta)
-        sta_table_dict["station name"] = list(stat_data.keys())
+        self.allure_report_table_format(dict_data=sta_data["interface"], key="Station Data",
-        for i in station_data:
+                                        value="Value", name="station data for 6G band")
-            temp_list = []
+
-            for j in station_list:
+        logging.info(f"sta_data after security modification:{sta_data}")
-                temp_list.append(stat_data[j][i])
-            sta_table_dict[i] = temp_list
 
         # pass fail
         pass_fail_sta = []
         for i in range(len(station_list)):
-            if sta_table_dict["ip"][i] == "0.0.0.0" or sta_table_dict["ap"][i] == "Not-Associated":
+            if sta_data["interface"]["ip"] == "0.0.0.0" or sta_data["interface"]["ap"] == "Not-Associated":
                 pass_fail_sta.append("Fail")
             else:
                 pass_fail_sta.append("Pass")
-        sta_table_dict["Pass/Fail"] = pass_fail_sta
+        sta_data["Pass/Fail"] = pass_fail_sta
-        if allure_attach:
-            self.attach_table_allure(data=sta_table_dict, allure_name="station data for 6G band")
 
         for i in range(len(station_list)):
-            if sta_table_dict["ip"][i] == "0.0.0.0":
+            if sta_data["interface"]["ip"] == "0.0.0.0":
                 logging.info("Station did not get an ip")
                 pytest.fail("Station did not get an ip")
             else:
                 logging.info("Station got IP")
 
         for i in range(len(station_list)):
-            if sta_table_dict["ap"][i] == "Not-Associated":
+            if sta_data["interface"]["ap"] == "Not-Associated":
                 logging.info("Station did not associate to AP")
                 pytest.fail("Station did not associate to AP")
             else:
                 logging.info("Station successfully associated to AP")
 
-        return stat_data
+        return sta_data
 
     def band_steering_test(self, ssid="[BLANK]", passkey="[BLANK]", security="wpa2", mode="BRIDGE", band="twog",
                            num_sta=None, scan_ssid=True, client_type=0, pre_cleanup=True,