From 7916200739f2bccfc2f35f1796a1c22f538bef17 Mon Sep 17 00:00:00 2001
From: saurabh <saurabh.goyal@candelatech.com>
Date: Mon, 13 Sep 2021 20:19:46 +0530
Subject: [PATCH] Added Wpa3 Security Vlan Mode

Signed-off-by: saurabh <saurabh.goyal@candelatech.com>
---
 .../wpa3_security/test_vlan_mode.py           | 72 +++++++++++++++++++
 1 file changed, 72 insertions(+)

diff --git a/tests/e2e/basic/performance_tests/dual_band_performance_test/wpa3_security/test_vlan_mode.py b/tests/e2e/basic/performance_tests/dual_band_performance_test/wpa3_security/test_vlan_mode.py
index e69de29bb..474ca3861 100644
--- a/tests/e2e/basic/performance_tests/dual_band_performance_test/wpa3_security/test_vlan_mode.py
+++ b/tests/e2e/basic/performance_tests/dual_band_performance_test/wpa3_security/test_vlan_mode.py
@@ -0,0 +1,72 @@
+"""
+       Dual Band Performance Test : VLAN Mode
+       pytest -m "performance and dual_band_test and vlan"
+
+
+"""
+
+import os
+import allure
+import pytest
+
+pytestmark = [pytest.mark.performance, pytest.mark.vlan, pytest.mark.dual_band_test]#,
+#              pytest.mark.usefixtures("setup_test_run")]
+
+
+setup_params_general = {
+    "mode": "VLAN",
+    "ssid_modes": {
+        "wpa3_personal": [
+            {"ssid_name": "ssid_wpa3_personal_dual_band", "appliedRadios": ["2G", "5G"], "security_key": "something", "vlan": 100}
+            ]},
+    "rf": {},
+    "radius": False
+}
+
+
+@pytest.mark.dual_band_test
+@pytest.mark.wifi5
+@pytest.mark.wifi6
+@pytest.mark.parametrize(
+    'setup_profiles',
+    [setup_params_general],
+    indirect=True,
+    scope="class"
+)
+@pytest.mark.usefixtures("setup_profiles")
+class TestDualbandPerformanceVlan(object):
+    """
+         pytest -m "performance and dual_band_test and vlan and wpa3_personal and twog  and fiveg"
+    """
+
+    @pytest.mark.wpa3_personal
+    @pytest.mark.twog
+    @pytest.mark.fiveg
+    def test_client_wpa3_personal_wpa3_vlan(self, get_vif_state, lf_tools,
+                                  create_lanforge_chamberview_dut, lf_test, get_configuration):
+        profile_data = setup_params_general["ssid_modes"]["wpa3_personal"]
+        ssid_2G = profile_data[0]["ssid_name"]
+        ssid_5G = profile_data[0]["ssid_name"]
+        dut_name = create_lanforge_chamberview_dut
+        mode = "VLAN"
+        vlan = 100
+        print(lf_tools.dut_idx_mapping)
+        dut_5g = ""
+        dut_2g = ""
+        for i in lf_tools.dut_idx_mapping:
+            if lf_tools.dut_idx_mapping[i][3] == "5G":
+                dut_5g = dut_name + ' ' + lf_tools.dut_idx_mapping[i][0] + ' ' + lf_tools.dut_idx_mapping[i][4]
+                print(dut_5g)
+            if lf_tools.dut_idx_mapping[i][3] == "2G":
+                dut_2g = dut_name + ' ' + lf_tools.dut_idx_mapping[i][0] + ' ' + lf_tools.dut_idx_mapping[i][4]
+                print(dut_2g)
+        if ssid_2G and ssid_5G not in get_vif_state:
+            allure.attach(name="retest,vif state ssid not available:", body=str(get_vif_state))
+            pytest.xfail("SSID's NOT AVAILABLE IN VIF STATE")
+
+        dbpt_obj = lf_test.dualbandperformancetest(mode=mode, ssid_2G=ssid_2G, ssid_5G=ssid_5G,
+                                                   instance_name="dbp_instance_wpa3p_VLAN_p",
+                                                   vlan_id=vlan, dut_5g=dut_5g, dut_2g=dut_2g)
+        report_name = dbpt_obj.report_name[0]['LAST']["response"].split(":::")[1].split("/")[-1]
+        lf_tools.attach_report_graphs(report_name=report_name, pdf_name="Dual Band Performance Wpa3 Vlan Test")
+        assert True
\ No newline at end of file