tplinkrouterc6u 5.2.1__tar.gz → 5.4.0__tar.gz

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tplinkrouterc6u
-Version: 5.2.1
+Version: 5.4.0
 Summary: TP-Link Router API
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -303,7 +303,8 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer C7 (v4.0, v5.0)
 - Archer C5400X V1
 - Archer GX90 v1.0
-- Archer MR200 (v5, v5.3)
+- Archer MR200 (v5, v5.3, v6.0)
+- Archer MR550 v1
 - Archer MR600 (v1, v2, v3)
 - Archer VR600 v3
 - Archer VR900v
@@ -327,6 +328,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - TL-MR6500v
 - TL-XDR3010 V2
 - TL-WA3001 v1.0
+- NX510v v1.0
 
 ### Not fully tested Hardware Versions
 - AD7200 V2

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/README.md

@@ -281,7 +281,8 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer C7 (v4.0, v5.0)
 - Archer C5400X V1
 - Archer GX90 v1.0
-- Archer MR200 (v5, v5.3)
+- Archer MR200 (v5, v5.3, v6.0)
+- Archer MR550 v1
 - Archer MR600 (v1, v2, v3)
 - Archer VR600 v3
 - Archer VR900v
@@ -305,6 +306,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - TL-MR6500v
 - TL-XDR3010 V2
 - TL-WA3001 v1.0
+- NX510v v1.0
 
 ### Not fully tested Hardware Versions
 - AD7200 V2

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="tplinkrouterc6u",
-    version="5.2.1",
+    version="5.4.0",
     author="Alex Erohin",
     author_email="alexanderErohin@yandex.ru",
     description="TP-Link Router API",

tplinkrouterc6u-5.4.0/test/test_client_c1200.py

@@ -0,0 +1,259 @@
+from unittest import main, TestCase
+from json import loads
+from tplinkrouterc6u import (
+    TplinkC1200Router,
+    Connection,
+    ClientException,
+)
+from tplinkrouterc6u.common.package_enum import VPN
+
+
+class TestTPLinkC1200Client(TestCase):
+    def test_set_led_on(self) -> None:
+        response_led_general_read = """
+        {
+            "enable": "off",
+            "time_set": "yes",
+            "ledpm_support": "yes"
+        }
+        """
+
+        response_led_general_write = """
+        {
+            "enable": "on",
+            "time_set": "yes",
+            "ledpm_support": "yes"
+        }
+        """
+
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "admin/ledgeneral?form=setting&operation=read":
+                    return loads(response_led_general_read)
+                if path == "admin/ledgeneral?form=setting&operation=write":
+                    self.captured_path = path
+                    return loads(response_led_general_write)
+                raise ClientException()
+
+        client = TPLinkRouterTest("", "")
+
+        client.set_led(True)
+
+        expected_path = "admin/ledgeneral?form=setting&operation=write"
+
+        self.assertEqual(client.captured_path, expected_path)
+
+    def test_set_led_off(self) -> None:
+        response_led_general_read = """
+        {
+            "enable": "on",
+            "time_set": "yes",
+            "ledpm_support": "yes"
+        }
+        """
+
+        response_led_general_write = """
+        {
+            "enable": "off",
+            "time_set": "yes",
+            "ledpm_support": "yes"
+        }
+        """
+
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "admin/ledgeneral?form=setting&operation=read":
+                    return loads(response_led_general_read)
+                if path == "admin/ledgeneral?form=setting&operation=write":
+                    self.captured_path = path
+                    return loads(response_led_general_write)
+                raise ClientException()
+
+        client = TPLinkRouterTest("", "")
+
+        client.set_led(False)
+
+        expected_path = "admin/ledgeneral?form=setting&operation=write"
+
+        self.assertEqual(client.captured_path, expected_path)
+
+    def test_led_status(self) -> None:
+        response_led_general_read = """
+        {
+            "enable": "on",
+            "time_set": "yes",
+            "ledpm_support": "yes"
+        }
+        """
+
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "admin/ledgeneral?form=setting&operation=read":
+                    return loads(response_led_general_read)
+                raise ClientException()
+
+        client = TPLinkRouterTest("", "")
+
+        led_status = client.get_led()
+        self.assertTrue(led_status)
+
+    def test_set_wifi(self) -> None:
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                self.captured_path = path
+                self.captured_data = data
+
+        client = TPLinkRouterTest("", "")
+        client.set_wifi(
+            Connection.HOST_5G,
+            enable=True,
+            ssid="TestSSID",
+            hidden="no",
+            encryption="WPA3-PSK",
+            psk_version="2",
+            psk_cipher="AES",
+            psk_key="testkey123",
+            hwmode="11ac",
+            htmode="VHT20",
+            channel=36,
+            txpower="20",
+            disabled_all="no",
+        )
+
+        expected_data = (
+            "operation=write&enable=on&ssid=TestSSID&hidden=no&encryption=WPA3-PSK&"
+            "psk_version=2&psk_cipher=AES&psk_key=testkey123&hwmode=11ac&"
+            "htmode=VHT20&channel=36&txpower=20&disabled_all=no"
+        )
+        expected_path = f"admin/wireless?form=wireless_5g&{expected_data}"
+
+        self.assertEqual(client.captured_path, expected_path)
+        self.assertEqual(client.captured_data, expected_data)
+
+    def test_vpn_status(self) -> None:
+        response_openvpn_read = """
+        {
+            "enabled": "on",
+            "proto": "udp",
+            "access": "home",
+            "cert_exist": true,
+            "mask": "255.255.255.0",
+            "port": "1194",
+            "serverip": "10.8.0.0"
+        }
+        """
+
+        response_pptp_read = """
+        {
+            "enabled": "off",
+            "unencrypted_access": "on",
+            "samba_access": "on",
+            "netbios_pass": "on",
+            "remoteip": "10.0.0.11-20"
+        }
+        """
+
+        respone_vpnconn_openvpn = """[
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "openvpn", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "openvpn", "key": "7450"}
+        ]"""
+
+        respone_vpnconn_pptpvpn = """[
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"}
+        ]"""
+
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "/admin/openvpn?form=config&operation=read":
+                    return loads(response_openvpn_read)
+                if path == "/admin/pptpd?form=config&operation=read":
+                    return loads(response_pptp_read)
+                if path == "/admin/vpnconn?form=config&operation=list&vpntype=openvpn":
+                    return loads(respone_vpnconn_openvpn)
+                if path == "/admin/vpnconn?form=config&operation=list&vpntype=pptp":
+                    return loads(respone_vpnconn_pptpvpn)
+                raise ClientException()
+
+        client = TPLinkRouterTest("", "")
+
+        vpn_status = client.get_vpn_status()
+        self.assertTrue(vpn_status.openvpn_enable)
+        self.assertFalse(vpn_status.pptpvpn_enable)
+        self.assertEqual(vpn_status.openvpn_clients_total, 2)
+        self.assertEqual(vpn_status.pptpvpn_clients_total, 3)
+
+    def test_set_vpn(self) -> None:
+        response_openvpn_read = """
+        {
+            "enabled": "on",
+            "proto": "udp",
+            "access": "home",
+            "cert_exist": true,
+            "mask": "255.255.255.0",
+            "port": "1194",
+            "serverip": "10.8.0.0"
+        }
+        """
+
+        class TPLinkRouterTest(TplinkC1200Router):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "/admin/openvpn?form=config&operation=read":
+                    return loads(response_openvpn_read)
+                self.captured_path = path
+
+        client = TPLinkRouterTest("", "")
+        client.set_vpn(VPN.OPEN_VPN, True)
+
+        expected_path = (
+            "/admin/openvpn?form=config&operation=write&enabled=on"
+            "&proto=udp&access=home&cert_exist=True"
+            "&mask=255.255.255.0&port=1194&serverip=10.8.0.0"
+        )
+        self.assertEqual(client.captured_path, expected_path)
+
+
+if __name__ == "__main__":
+    main()

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/test/test_client_ex.py

@@ -11,6 +11,8 @@ from tplinkrouterc6u import (
     IPv4DHCPLease,
     IPv4Status,
     ClientException,
+    VPNStatus,
+    VPN,
 )
 
 
@@ -334,6 +336,65 @@ class TestTPLinkEXClient(TestCase):
         self.assertEqual(check_data, '{"data":{"stack":"1,0,0,0,0,0","pstack":"0,0,0,0,0,0",'
                                      '"primaryEnable":"1"},"operation":"so","oid":"DEV2_ADT_WIFI_COMMON"}')
 
+    def test_get_vpn_status(self) -> None:
+        DEV2_OPENVPN = ('{"data":{"enable":"1","stack":"0,0,0,0,0,0"},'
+                        '"operation":"go","oid":"DEV2_OPENVPN","success":true}')
+        DEV2_PPTPVPN = ('{"data":{"enable":"0","stack":"0,0,0,0,0,0"},'
+                        '"operation":"go","oid":"DEV2_PPTPVPN","success":true}')
+        DEV2_OVPN_CLIENT = ('{"data":[{"connAct":"1","stack":"1,0,0,0,0,0"}, {"connAct":"1","stack":"2,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"3,0,0,0,0,0"}, {"connAct":"0","stack":"4,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"5,0,0,0,0,0"}, {"connAct":"0","stack":"6,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"7,0,0,0,0,0"}, {"connAct":"0","stack":"8,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"9,0,0,0,0,0"}, {"connAct":"0","stack":"10,0,0,0,0,0"}],'
+                            '"operation":"gl","oid":"DEV2_OVPN_CLIENT","success":true}')
+        DEV2_PVPN_CLIENT = ('{"data":[{"connAct":"0","stack":"1,0,0,0,0,0"}, {"connAct":"0","stack":"2,0,0,0,0,0"},'
+                            '{"connAct":"1","stack":"3,0,0,0,0,0"}, {"connAct":"0","stack":"4,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"5,0,0,0,0,0"}, {"connAct":"0","stack":"6,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"7,0,0,0,0,0"}, {"connAct":"0","stack":"8,0,0,0,0,0"},'
+                            '{"connAct":"0","stack":"9,0,0,0,0,0"}, {"connAct":"0","stack":"10,0,0,0,0,0"}],'
+                            '"operation":"gl","oid":"DEV2_PVPN_CLIENT","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                if 'DEV2_OPENVPN' in data_str:
+                    return 200, DEV2_OPENVPN
+                elif 'DEV2_PPTPVPN' in data_str:
+                    return 200, DEV2_PPTPVPN
+                elif 'DEV2_OVPN_CLIENT' in data_str:
+                    return 200, DEV2_OVPN_CLIENT
+                elif 'DEV2_PVPN_CLIENT' in data_str:
+                    return 200, DEV2_PVPN_CLIENT
+                raise ClientException()
+
+        client = TPLinkEXClientTest('', '')
+        status = client.get_vpn_status()
+
+        self.assertIsInstance(status, VPNStatus)
+        self.assertEqual(status.openvpn_enable, True)
+        self.assertEqual(status.pptpvpn_enable, False)
+        self.assertEqual(status.openvpn_clients_total, 2)
+        self.assertEqual(status.pptpvpn_clients_total, 1)
+
+    def test_set_vpn(self) -> None:
+        response = '{"success":true, "errorcode":0}'
+
+        check_url = ''
+        check_data = ''
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                nonlocal check_url, check_data
+                check_url = url
+                check_data = data_str
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        client.set_vpn(VPN.OPEN_VPN, True)
+
+        self.assertIn('http:///cgi_gdpr?9?_=', check_url)
+        self.assertEqual(check_data, '{"data":{"stack":"0,0,0,0,0,0","pstack":"0,0,0,0,0,0",'
+                                     '"enable":"1"},"operation":"so","oid":"DEV2_OPENVPN"}')
+
 
 if __name__ == '__main__':
     main()

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u/client/c1200.py

@@ -1,8 +1,11 @@
 from re import search
 from requests import post, Response
+from urllib.parse import urlencode
 from tplinkrouterc6u.common.encryption import EncryptionWrapper
 from tplinkrouterc6u.common.exception import ClientException, AuthorizeError
 from tplinkrouterc6u.client.c5400x import TplinkC5400XRouter
+from tplinkrouterc6u.common.dataclass import VPNStatus
+from tplinkrouterc6u.common.package_enum import VPN
 
 
 class TplinkC1200Router(TplinkC5400XRouter):
@@ -100,3 +103,36 @@ class TplinkC1200Router(TplinkC5400XRouter):
     @staticmethod
     def _get_login_data(crypted_pwd: str) -> str:
         return 'operation=login&password={}'.format(crypted_pwd)
+
+    def get_vpn_status(self) -> VPNStatus:
+        status = VPNStatus()
+
+        values = [
+            self.request("/admin/openvpn?form=config&operation=read", "operation=read"),
+            self.request("/admin/pptpd?form=config&operation=read", "operation=read"),
+            self.request("/admin/vpnconn?form=config&operation=list&vpntype=openvpn",
+                         "operation=list&operation=list&vpntype=openvpn"),
+            self.request("/admin/vpnconn?form=config&operation=list&vpntype=pptp",
+                         "operation=list&operation=list&vpntype=pptp"),
+        ]
+
+        status.openvpn_enable = values[0]['enabled'] == 'on'
+        status.pptpvpn_enable = values[1]['enabled'] == 'on'
+
+        if isinstance(values[2], list):
+            status.openvpn_clients_total = len(values[2])
+            status.pptpvpn_clients_total = len(values[3])
+        else:
+            status.openvpn_clients_total = 0
+            status.pptpvpn_clients_total = 0
+
+        return status
+
+    def set_vpn(self, vpn: VPN, enable: bool) -> None:
+        path = "/admin/{}?form=config&operation=read".format(vpn.lowercase)
+        current_config = self.request(path, "operation=read")
+        current_config['enabled'] = "on" if enable else "off"
+        data = urlencode(current_config)
+        data = "&operation=write&{}".format(data)
+        path = "/admin/{}?form=config{}".format(vpn.lowercase, data)
+        self.request(path, data)

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u/client/c5400x.py

@@ -42,7 +42,7 @@ class TplinkC5400XRouter(TplinkBaseRouter):
         if current_state != enable:
             self.request('admin/ledgeneral?form=setting&operation=write', 'operation=write')
 
-    def get_led(self) -> bool:
+    def get_led(self) -> bool | None:
 
         data = self.request('admin/ledgeneral?form=setting&operation=read', 'operation=read')
         led_status = data.get('enable') if 'enable' in data else None

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u/client/ex.py

@@ -4,8 +4,15 @@ from datetime import timedelta
 from macaddress import EUI48
 from ipaddress import IPv4Address
 from logging import Logger
-from tplinkrouterc6u.common.package_enum import Connection
-from tplinkrouterc6u.common.dataclass import Firmware, Status, Device, IPv4Reservation, IPv4DHCPLease, IPv4Status
+from tplinkrouterc6u.common.package_enum import Connection, VPN
+from tplinkrouterc6u.common.dataclass import (
+    Firmware,
+    Status,
+    Device,
+    IPv4Reservation,
+    IPv4DHCPLease,
+    IPv4Status,
+    VPNStatus)
 from tplinkrouterc6u.common.exception import ClientException, ClientError
 from tplinkrouterc6u.client.mr import TPLinkMRClientBase
 
@@ -293,3 +300,33 @@ class TPLinkEXClient(TPLinkMRClientBase):
             if self._logger:
                 self._logger.debug(error)
             raise ClientException(error)
+
+    def get_vpn_status(self) -> VPNStatus:
+        status = VPNStatus()
+        acts = [
+            self.ActItem(self.ActItem.GET, 'DEV2_OPENVPN', attrs=['enable']),
+            self.ActItem(self.ActItem.GET, 'DEV2_PPTPVPN', attrs=['enable']),
+            self.ActItem(self.ActItem.GL, 'DEV2_OVPN_CLIENT', attrs=['connAct']),
+            self.ActItem(self.ActItem.GL, 'DEV2_PVPN_CLIENT', attrs=['connAct']),
+        ]
+        _, values = self.req_act(acts)
+
+        status.openvpn_enable = values[0]['enable'] == '1'
+        status.pptpvpn_enable = values[1]['enable'] == '1'
+
+        for item in values[2]:
+            if item['connAct'] == '1':
+                status.openvpn_clients_total += 1
+
+        for item in values[3]:
+            if item['connAct'] == '1':
+                status.pptpvpn_clients_total += 1
+
+        return status
+
+    def set_vpn(self, vpn: VPN, enable: bool) -> None:
+        acts = [
+            self.ActItem(self.ActItem.SET, "DEV2_" + vpn.value, attrs=[f'"enable":"{int(enable)}"'])
+        ]
+
+        self.req_act(acts)

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u/client/mr.py

@@ -599,7 +599,7 @@ class TPLinkMRClient(TPLinkMRClientBase):
                 messages.append(
                     SMS(
                         i, item['from'], item['content'], datetime.fromisoformat(item['receivedTime']),
-                        True if item['unread'] == '1' else False
+                        item['unread'] == '1'
                     )
                 )
                 i += 1
@@ -691,8 +691,8 @@ class TPLinkMRClient(TPLinkMRClientBase):
         ]
         _, values = self.req_act(acts)
 
-        status.openvpn_enable = True if values['0']['enable'] == '1' else False
-        status.pptpvpn_enable = True if values['1']['enable'] == '1' else False
+        status.openvpn_enable = values['0']['enable'] == '1'
+        status.pptpvpn_enable = values['1']['enable'] == '1'
 
         for item in values['2']:
             if item['connAct'] == '1':

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u/common/package_enum.py

@@ -49,3 +49,8 @@ class Connection(Enum):
 class VPN(Enum):
     OPEN_VPN = 'OPENVPN'
     PPTP_VPN = 'PPTPVPN'
+
+    @property
+    def lowercase(self) -> str:
+        """Returns the lowercase version of the enum value. Needed for the c1200 router."""
+        return self.value.lower()

{tplinkrouterc6u-5.2.1 → tplinkrouterc6u-5.4.0}/tplinkrouterc6u.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tplinkrouterc6u
-Version: 5.2.1
+Version: 5.4.0
 Summary: TP-Link Router API
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -303,7 +303,8 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer C7 (v4.0, v5.0)
 - Archer C5400X V1
 - Archer GX90 v1.0
-- Archer MR200 (v5, v5.3)
+- Archer MR200 (v5, v5.3, v6.0)
+- Archer MR550 v1
 - Archer MR600 (v1, v2, v3)
 - Archer VR600 v3
 - Archer VR900v
@@ -327,6 +328,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - TL-MR6500v
 - TL-XDR3010 V2
 - TL-WA3001 v1.0
+- NX510v v1.0
 
 ### Not fully tested Hardware Versions
 - AD7200 V2

tplinkrouterc6u-5.2.1/test/test_client_c1200.py

@@ -1,142 +0,0 @@
-from unittest import main, TestCase
-from json import loads
-from tplinkrouterc6u import (
-    TplinkC1200Router,
-    Connection,
-    ClientException
-)
-
-
-class TestTPLinkC1200Client(TestCase):
-
-    def test_set_led_on(self) -> None:
-
-        response_led_general_read = '''
-        {
-            "enable": "off",
-            "time_set": "yes",
-            "ledpm_support": "yes"
-        }
-        '''
-
-        response_led_general_write = '''
-        {
-            "enable": "on",
-            "time_set": "yes",
-            "ledpm_support": "yes"
-        }
-        '''
-
-        class TPLinkRouterTest(TplinkC1200Router):
-            def request(self, path: str, data: str,
-                        ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
-                if path == 'admin/ledgeneral?form=setting&operation=read':
-                    return loads(response_led_general_read)
-                if path == 'admin/ledgeneral?form=setting&operation=write':
-                    self.captured_path = path
-                    return loads(response_led_general_write)
-                raise ClientException()
-
-        client = TPLinkRouterTest('', '')
-
-        client.set_led(True)
-
-        expected_path = "admin/ledgeneral?form=setting&operation=write"
-
-        self.assertEqual(client.captured_path, expected_path)
-
-    def test_set_led_off(self) -> None:
-
-        response_led_general_read = '''
-        {
-            "enable": "on",
-            "time_set": "yes",
-            "ledpm_support": "yes"
-        }
-        '''
-
-        response_led_general_write = '''
-        {
-            "enable": "off",
-            "time_set": "yes",
-            "ledpm_support": "yes"
-        }
-        '''
-
-        class TPLinkRouterTest(TplinkC1200Router):
-            def request(self, path: str, data: str,
-                        ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
-                if path == 'admin/ledgeneral?form=setting&operation=read':
-                    return loads(response_led_general_read)
-                elif path == 'admin/ledgeneral?form=setting&operation=write':
-                    self.captured_path = path
-                    return loads(response_led_general_write)
-                raise ClientException()
-
-        client = TPLinkRouterTest('', '')
-
-        client.set_led(False)
-
-        expected_path = "admin/ledgeneral?form=setting&operation=write"
-
-        self.assertEqual(client.captured_path, expected_path)
-
-    def test_led_status(self) -> None:
-
-        response_led_general_read = '''
-        {
-            "enable": "on",
-            "time_set": "yes",
-            "ledpm_support": "yes"
-        }
-        '''
-
-        class TPLinkRouterTest(TplinkC1200Router):
-            def request(self, path: str, data: str,
-                        ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
-                if path == 'admin/ledgeneral?form=setting&operation=read':
-                    return loads(response_led_general_read)
-                raise ClientException()
-
-        client = TPLinkRouterTest('', '')
-
-        led_status = client.get_led()
-        self.assertTrue(led_status)
-
-    def test_set_wifi(self) -> None:
-
-        class TPLinkRouterTest(TplinkC1200Router):
-            def request(self, path: str, data: str,
-                        ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
-
-                self.captured_path = path
-                self.captured_data = data
-
-        client = TPLinkRouterTest('', '')
-        client.set_wifi(
-            Connection.HOST_5G,
-            enable=True,
-            ssid="TestSSID",
-            hidden="no",
-            encryption="WPA3-PSK",
-            psk_version="2",
-            psk_cipher="AES",
-            psk_key="testkey123",
-            hwmode="11ac",
-            htmode="VHT20",
-            channel=36,
-            txpower="20",
-            disabled_all="no"
-        )
-
-        expected_data = ("operation=write&enable=on&ssid=TestSSID&hidden=no&encryption=WPA3-PSK&"
-                         "psk_version=2&psk_cipher=AES&psk_key=testkey123&hwmode=11ac&"
-                         "htmode=VHT20&channel=36&txpower=20&disabled_all=no")
-        expected_path = f"admin/wireless?form=wireless_5g&{expected_data}"
-
-        self.assertEqual(client.captured_path, expected_path)
-        self.assertEqual(client.captured_data, expected_data)
-
-
-if __name__ == '__main__':
-    main()