tplinkrouterc6u 5.0.2__tar.gz → 5.1.0__tar.gz

{tplinkrouterc6u-5.0.2 → tplinkrouterc6u-5.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tplinkrouterc6u
-Version: 5.0.2
+Version: 5.1.0
 Summary: TP-Link Router API
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -44,6 +44,7 @@ from tplinkrouterc6u import (
     TplinkC1200Router,
     TplinkC5400XRouter,
     TPLinkMRClient,
+    TPLinkEXClient,
     TPLinkDecoClient,
     Connection
 )
@@ -107,6 +108,10 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 | get_ipv4_dhcp_leases |   | Gets IPv4 addresses assigned via DHCP | [[IPv4DHCPLease]](#IPv4DHCPLease) | 
 | set_wifi | wifi: [Connection](#connection), enable: bool | Allow to turn on/of 4 wifi networks |   |
 | send_sms | phone_number: str, message: str | Send sms for LTE routers |   |
+| send_ussd | command: str | Send USSD command for LTE routers | str |
+| get_sms | | Get sms messages from the first page for LTE routers | [[SMS]](#sms) |
+| set_sms_read | sms: [SMS](#sms) | Set sms message read from the first page for LTE routers |   |
+| delete_sms | sms: [SMS](#sms) | Delete sms message from the first page for LTE routers |   |
 | reboot |   | reboot router |
 | authorize |   | authorize for actions |
 | logout |   | logout after all is done |
@@ -210,6 +215,15 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 | lan_ipv4_netmask_address | router LAN gateway IP netmask | ipaddress |
 | remote | router remote | bool, None |
 
+### <a id="sms">SMS</a>
+| Field | Description | Type |
+| --- |---|---|
+| id | message index | int |
+| sender| sender | str |
+| content| sms text | str |
+| received_at| received datetime | datetime |
+| unread| is message unread | bool |
+
 ## Enum
 ### <a id="connection">Connection</a>
 - Connection.HOST_2G - host wifi 2.4G
@@ -257,8 +271,9 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer GX90 v1.0
 - Archer MR200 (v5, v5.3)
 - Archer MR600 (v1, v2, v3)
-- Archer VR2100v v1
+- Archer VR600 v3
 - Archer VR900v
+- Archer VR2100v v1
 - Deco M4 2.0
 - Deco M4R 2.0
 - Deco M5 v3
@@ -270,6 +285,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Deco X60 V3
 - Deco X90
 - Deco XE75 (v1.0, v2.0)
+- EX511 v2.0
 - TD-W9960 (v1, V1.20)
 - TL-MR100 v2.0
 - TL-MR105

{tplinkrouterc6u-5.0.2 → tplinkrouterc6u-5.1.0}/README.md

@@ -22,6 +22,7 @@ from tplinkrouterc6u import (
     TplinkC1200Router,
     TplinkC5400XRouter,
     TPLinkMRClient,
+    TPLinkEXClient,
     TPLinkDecoClient,
     Connection
 )
@@ -85,6 +86,10 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 | get_ipv4_dhcp_leases |   | Gets IPv4 addresses assigned via DHCP | [[IPv4DHCPLease]](#IPv4DHCPLease) | 
 | set_wifi | wifi: [Connection](#connection), enable: bool | Allow to turn on/of 4 wifi networks |   |
 | send_sms | phone_number: str, message: str | Send sms for LTE routers |   |
+| send_ussd | command: str | Send USSD command for LTE routers | str |
+| get_sms | | Get sms messages from the first page for LTE routers | [[SMS]](#sms) |
+| set_sms_read | sms: [SMS](#sms) | Set sms message read from the first page for LTE routers |   |
+| delete_sms | sms: [SMS](#sms) | Delete sms message from the first page for LTE routers |   |
 | reboot |   | reboot router |
 | authorize |   | authorize for actions |
 | logout |   | logout after all is done |
@@ -188,6 +193,15 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 | lan_ipv4_netmask_address | router LAN gateway IP netmask | ipaddress |
 | remote | router remote | bool, None |
 
+### <a id="sms">SMS</a>
+| Field | Description | Type |
+| --- |---|---|
+| id | message index | int |
+| sender| sender | str |
+| content| sms text | str |
+| received_at| received datetime | datetime |
+| unread| is message unread | bool |
+
 ## Enum
 ### <a id="connection">Connection</a>
 - Connection.HOST_2G - host wifi 2.4G
@@ -235,8 +249,9 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer GX90 v1.0
 - Archer MR200 (v5, v5.3)
 - Archer MR600 (v1, v2, v3)
-- Archer VR2100v v1
+- Archer VR600 v3
 - Archer VR900v
+- Archer VR2100v v1
 - Deco M4 2.0
 - Deco M4R 2.0
 - Deco M5 v3
@@ -248,6 +263,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Deco X60 V3
 - Deco X90
 - Deco XE75 (v1.0, v2.0)
+- EX511 v2.0
 - TD-W9960 (v1, V1.20)
 - TL-MR100 v2.0
 - TL-MR105

{tplinkrouterc6u-5.0.2 → tplinkrouterc6u-5.1.0}/setup.py

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

{tplinkrouterc6u-5.0.2 → tplinkrouterc6u-5.1.0}/test/test_client_deco.py

@@ -70,7 +70,7 @@ class TestTPLinkDecoClient(TestCase):
                 "connection_type": "band5", "space_id": "1", "ip": "192.168.68.104", "client_mesh": true,
                 "online": true, "name": "d2lyZWxlc3M0", "enable_priority": false, "remain_time": 0, "owner_id": "",
                 "client_type": "other", "interface": "guest"},
-        {"mac": "56:32:c3:de:ce:f0", "up_speed": 3, "down_speed": 1, "wire_type": "wireless", "access_host": "1",
+        {"mac": "", "up_speed": 3, "down_speed": 1, "wire_type": "wireless", "access_host": "1",
                 "connection_type": "band2_4", "space_id": "1", "ip": "UNKNOWN", "client_mesh": true,
                 "online": true, "name": "d2lyZWxlc3M1", "enable_priority": false, "remain_time": 0, "owner_id": "",
                 "client_type": "other", "interface": "guest"}
@@ -149,7 +149,7 @@ class TestTPLinkDecoClient(TestCase):
         self.assertEqual(status.devices[3].packets_received, None)
         self.assertIsInstance(status.devices[4], Device)
         self.assertEqual(status.devices[4].type, Connection.GUEST_2G)
-        self.assertEqual(status.devices[4].macaddr, '56-32-C3-DE-CE-F0')
+        self.assertEqual(status.devices[4].macaddr, '00-00-00-00-00-00')
         self.assertEqual(status.devices[4].ipaddr, '0.0.0.0')
         self.assertEqual(status.devices[4].hostname, 'wireless5')
         self.assertEqual(status.devices[4].packets_sent, None)

tplinkrouterc6u-5.1.0/test/test_client_ex.py

@@ -0,0 +1,339 @@
+from unittest import main, TestCase
+from macaddress import EUI48
+from ipaddress import IPv4Address
+from tplinkrouterc6u import (
+    TPLinkEXClient,
+    Connection,
+    Firmware,
+    Status,
+    Device,
+    IPv4Reservation,
+    IPv4DHCPLease,
+    IPv4Status,
+    ClientException,
+)
+
+
+class TestTPLinkEXClient(TestCase):
+    def test_firmware(self) -> None:
+        response = ('{"data":{"hardwareVersion":"EX511 v2.0 00000000","modelName":"EX511",'
+                    '"softwareVersion":"0.7.0 3.0.0 v607e.0 Build 240930 Rel.11206n","stack":"0,0,0,0,0,0"},'
+                    '"operation":"go","oid":"DEV2_DEV_INFO","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_firmware()
+
+        self.assertIsInstance(result, Firmware)
+        self.assertEqual(result.hardware_version, 'EX511 v2.0 00000000')
+        self.assertEqual(result.model, 'EX511')
+        self.assertEqual(result.firmware_version, '0.7.0 3.0.0 v607e.0 Build 240930 Rel.11206n')
+
+    def test_get_status_with_5G(self) -> None:
+
+        DEV2_ADT_LAN = ('{"data":[{"MACAddress":"a0:28:84:de:dd:5c","IPAddress":"192.168.4.1","stack":"1,0,0,0,0,0"}],'
+                        '"operation":"gl","oid":"DEV2_ADT_LAN","success":true}')
+        DEV2_ADT_WAN = ('{"data":[{"enable":"1","MACAddr":"BF-75-44-4C-DC-9E","connIPv4Address":"192.168.30.55",'
+                        '"connIPv4Gateway":"192.168.30.1","stack":"1,0,0,0,0,0"}],"operation":"gl",'
+                        '"oid":"DEV2_ADT_WAN","success":true}')
+        DEV2_ADT_WIFI_COMMON = ('{"data":[{"primaryEnable":"1","guestEnable":"0","stack":"1,0,0,0,0,0"},'
+                                '{"primaryEnable":"0","guestEnable":"1","stack":"2,0,0,0,0,0"}],"operation":"gl",'
+                                '"oid":"DEV2_ADT_WIFI_COMMON","success":true}')
+        DEV2_HOST_ENTRY = ('{"data":[{"active":"1","X_TP_LanConnType":"0","physAddress":"66-E2-02-BD-B5-1B",'
+                           '"IPAddress":"192.168.30.10","hostName":"host1","stack":"1,0,0,0,0,0"},'
+                           '{"active":"1","X_TP_LanConnType":"1","physAddress":"F4-A3-86-2D-41-B5",'
+                           '"IPAddress":"192.168.30.11","hostName":"host2","stack":"2,0,0,0,0,0"}],"operation":"gl",'
+                           '"oid":"DEV2_HOST_ENTRY","success":true}')
+        DEV2_MEM_STATUS = ('{"data":{"total":"192780","free":"78400","stack":"0,0,0,0,0,0"},"operation":"go",'
+                           '"oid":"DEV2_MEM_STATUS","success":true}')
+        DEV2_PROC_STATUS = ('{"data":{"CPUUsage":"47","stack":"0,0,0,0,0,0"},"operation":"go",'
+                            '"oid":"DEV2_PROC_STATUS","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                if 'DEV2_ADT_LAN' in data_str:
+                    return 200, DEV2_ADT_LAN
+                elif 'DEV2_ADT_WAN' in data_str:
+                    return 200, DEV2_ADT_WAN
+                elif 'DEV2_ADT_WIFI_COMMON' in data_str:
+                    return 200, DEV2_ADT_WIFI_COMMON
+                elif 'DEV2_HOST_ENTRY' in data_str:
+                    return 200, DEV2_HOST_ENTRY
+                elif 'DEV2_MEM_STATUS' in data_str:
+                    return 200, DEV2_MEM_STATUS
+                elif 'DEV2_PROC_STATUS' in data_str:
+                    return 200, DEV2_PROC_STATUS
+                raise ClientException()
+
+        client = TPLinkEXClientTest('', '')
+        status = client.get_status()
+
+        self.assertIsInstance(status, Status)
+        self.assertEqual(status.wan_macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertIsInstance(status.wan_macaddress, EUI48)
+        self.assertEqual(status.lan_macaddr, 'A0-28-84-DE-DD-5C')
+        self.assertIsInstance(status.lan_macaddress, EUI48)
+        self.assertEqual(status.wan_ipv4_addr, '192.168.30.55')
+        self.assertIsInstance(status.lan_ipv4_address, IPv4Address)
+        self.assertEqual(status.lan_ipv4_addr, '192.168.4.1')
+        self.assertEqual(status.wan_ipv4_gateway, '192.168.30.1')
+        self.assertIsInstance(status.wan_ipv4_address, IPv4Address)
+        self.assertEqual(status.wired_total, 1)
+        self.assertEqual(status.wifi_clients_total, 1)
+        self.assertEqual(status.guest_clients_total, 0)
+        self.assertEqual(status.clients_total, 2)
+        self.assertEqual(status.guest_2g_enable, False)
+        self.assertEqual(status.guest_5g_enable, True)
+        self.assertEqual(status.iot_2g_enable, None)
+        self.assertEqual(status.iot_5g_enable, None)
+        self.assertEqual(status.wifi_2g_enable, True)
+        self.assertEqual(status.wifi_5g_enable, False)
+        self.assertEqual(status.wan_ipv4_uptime, None)
+        self.assertGreaterEqual(status.mem_usage, 0)
+        self.assertLessEqual(status.mem_usage, 1)
+        self.assertGreaterEqual(status.cpu_usage, 0)
+        self.assertLessEqual(status.cpu_usage, 1)
+        self.assertEqual(len(status.devices), 2)
+        self.assertIsInstance(status.devices[0], Device)
+        self.assertEqual(status.devices[0].type, Connection.WIRED)
+        self.assertEqual(status.devices[0].macaddr, '66-E2-02-BD-B5-1B')
+        self.assertIsInstance(status.devices[0].macaddress, EUI48)
+        self.assertEqual(status.devices[0].ipaddr, '192.168.30.10')
+        self.assertIsInstance(status.devices[0].ipaddress, IPv4Address)
+        self.assertEqual(status.devices[0].hostname, 'host1')
+        self.assertEqual(status.devices[0].packets_sent, None)
+        self.assertEqual(status.devices[0].packets_received, None)
+        self.assertIsInstance(status.devices[1], Device)
+        self.assertEqual(status.devices[1].type, Connection.HOST_2G)
+        self.assertEqual(status.devices[1].macaddr, 'F4-A3-86-2D-41-B5')
+        self.assertIsInstance(status.devices[1].macaddress, EUI48)
+        self.assertEqual(status.devices[1].ipaddr, '192.168.30.11')
+        self.assertIsInstance(status.devices[1].ipaddress, IPv4Address)
+        self.assertEqual(status.devices[1].hostname, 'host2')
+        self.assertEqual(status.devices[1].packets_sent, None)  # TODO
+        self.assertEqual(status.devices[1].packets_received, None)  # TODO
+
+    def test_get_ipv4_reservations(self) -> None:
+
+        response = ('{"data":[{"enable":"1","chaddr":"bf:75:44:4c:dc:9e","yiaddr":"192.168.8.21",'
+                    '"stack":"1,1,0,0,0,0"}],"operation":"gl","oid":"DEV2_DHCPV4_POOL_STATICADDR","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_reservations()
+
+        self.assertEqual(len(result), 1)
+        self.assertIsInstance(result[0], IPv4Reservation)
+        self.assertEqual(result[0].macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertEqual(result[0].ipaddr, '192.168.8.21')
+        self.assertEqual(result[0].hostname, '')
+        self.assertEqual(result[0].enabled, True)
+
+    def test_get_ipv4_reservations_no_reservations(self) -> None:
+        response = '''
+[error]0
+
+'''
+        response = '{"data":[],"operation":"gl","oid":"DEV2_DHCPV4_POOL_STATICADDR","success":true}'
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_reservations()
+
+        self.assertEqual(len(result), 0)
+
+    def test_get_ipv4_dhcp_leases_no_leases(self) -> None:
+
+        response = '{"data":[],"operation":"gl","oid":"DEV2_HOST_ENTRY","success":true}'
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_dhcp_leases()
+
+        self.assertEqual(len(result), 0)
+
+    def test_get_ipv4_dhcp_leases(self) -> None:
+
+        response = ('{"data":[{"alias":"","physAddress":"bf:75:44:4c:dc:9e","IPAddress":"192.168.32.175",'
+                    '"addressSource":"Static","leaseTimeRemaining":"85841","X_TP_IPv6Address":"",'
+                    '"X_TP_IPv6LinkLocal":"","layer1Interface":"",'
+                    '"X_TP_Layer2Interface":"Device.WiFi.AccessPoint.1.","vendorClassID":"","clientID":"",'
+                    '"hostName":"name1","interfaceType":"Wi-Fi","X_TP_LanConnType":"1","X_TP_LanConnDev":"br0",'
+                    '"active":"1","IPv4AddressNumberOfEntries":"0","X_TP_Vendor":"","X_TP_ClientType":"Other",'
+                    '"X_TP_DevphyAddress":"","IPv6AddressNumberOfEntries":"0","X_TP_NetworkReadyTime":"0",'
+                    '"stack":"1,0,0,0,0,0"}],"operation":"gl","oid":"DEV2_HOST_ENTRY","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_dhcp_leases()
+
+        self.assertEqual(len(result), 1)
+        self.assertIsInstance(result[0], IPv4DHCPLease)
+        self.assertEqual(result[0].macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertEqual(result[0].ipaddr, '192.168.32.175')
+        self.assertEqual(result[0].hostname, 'name1')
+        self.assertEqual(result[0].lease_time, '23:50:41')
+
+    def test_get_ipv4_dhcp_leases_permanent(self) -> None:
+
+        response = ('{"data":[{"alias":"","physAddress":"bf:75:44:4c:dc:9e","IPAddress":"192.168.32.175",'
+                    '"addressSource":"Static","leaseTimeRemaining":"0","X_TP_IPv6Address":"",'
+                    '"X_TP_IPv6LinkLocal":"","layer1Interface":"","X_TP_Layer2Interface":"Device.WiFi.AccessPoint.1.",'
+                    '"vendorClassID":"","clientID":"","hostName":"name1","interfaceType":"Wi-Fi",'
+                    '"X_TP_LanConnType":"1","X_TP_LanConnDev":"br0","active":"1","IPv4AddressNumberOfEntries":"0",'
+                    '"X_TP_Vendor":"","X_TP_ClientType":"Other","X_TP_DevphyAddress":"",'
+                    '"IPv6AddressNumberOfEntries":"0","X_TP_NetworkReadyTime":"0","stack":"1,0,0,0,0,0"},'
+                    '{"alias":"","physAddress":"a0:28:84:de:dd:5c","IPAddress":"192.168.32.176",'
+                    '"addressSource":"Static","leaseTimeRemaining":"86372","X_TP_IPv6Address":"",'
+                    '"X_TP_IPv6LinkLocal":"","layer1Interface":"","X_TP_Layer2Interface":"Device.WiFi.AccessPoint.1.",'
+                    '"vendorClassID":"","clientID":"","hostName":"name2","interfaceType":"Wi-Fi",'
+                    '"X_TP_LanConnType":"1","X_TP_LanConnDev":"br0","active":"1","IPv4AddressNumberOfEntries":"0",'
+                    '"X_TP_Vendor":"","X_TP_ClientType":"Other","X_TP_DevphyAddress":"",'
+                    '"IPv6AddressNumberOfEntries":"0","X_TP_NetworkReadyTime":"0","stack":"1,0,0,0,0,0"}],'
+                    '"operation":"gl","oid":"DEV2_HOST_ENTRY","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                return 200, response
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_dhcp_leases()
+
+        self.assertEqual(len(result), 2)
+        self.assertEqual(result[0].macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertEqual(result[0].ipaddr, '192.168.32.175')
+        self.assertEqual(result[0].hostname, 'name1')
+        self.assertEqual(result[0].lease_time, 'Permanent')
+        self.assertEqual(result[1].macaddr, 'A0-28-84-DE-DD-5C')
+        self.assertEqual(result[1].ipaddr, '192.168.32.176')
+        self.assertEqual(result[1].hostname, 'name2')
+        self.assertEqual(result[1].lease_time, '23:59:32')
+
+    def test_get_ipv4_status(self) -> None:
+
+        DEV2_ADT_LAN = ('{"data":[{"MACAddress":"bf:75:44:4c:dc:9e","IPAddress":"192.168.5.1",'
+                        '"IPSubnetMask":"255.255.255.0","DHCPv4Enable":"1","stack":"1,0,0,0,0,0"}],'
+                        '"operation":"gl","oid":"DEV2_ADT_LAN","success":true}')
+        DEV2_ADT_WAN = ('{"data":[{"enable":"1","MACAddr":"a0:28:84:de:dd:5c","connIPv4Address":"10.10.11.5",'
+                        '"connIPv4Gateway":"11.11.11.11","name":"ipoe_0_0_d","connIPv4SubnetMask":"1.1.1.1",'
+                        '"connIPv4DnsServer":"8.8.8.8,8.8.4.4","stack":"1,0,0,0,0,0"}],"operation":"gl",'
+                        '"oid":"DEV2_ADT_WAN","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                if 'DEV2_ADT_LAN' in data_str:
+                    return 200, DEV2_ADT_LAN
+                elif 'DEV2_ADT_WAN' in data_str:
+                    return 200, DEV2_ADT_WAN
+                raise ClientException()
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_status()
+
+        self.assertIsInstance(result, IPv4Status)
+        self.assertEqual(result.wan_macaddr, 'A0-28-84-DE-DD-5C')
+        self.assertEqual(result.wan_ipv4_ipaddr, '10.10.11.5')
+        self.assertEqual(result.wan_ipv4_gateway, '11.11.11.11')
+        self.assertEqual(result.wan_ipv4_conntype, 'ipoe_0_0_d')
+        self.assertEqual(result.wan_ipv4_netmask, '1.1.1.1')
+        self.assertEqual(result.wan_ipv4_pridns, '8.8.8.8')
+        self.assertEqual(result.wan_ipv4_snddns, '8.8.4.4')
+        self.assertEqual(result.lan_macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertEqual(result.lan_ipv4_ipaddr, '192.168.5.1')
+        self.assertEqual(result.lan_ipv4_netmask, '255.255.255.0')
+        self.assertEqual(result.lan_ipv4_dhcp_enable, True)
+        self.assertEqual(result.remote, None)
+
+    def test_get_ipv4_status_one_wlan(self) -> None:
+
+        DEV2_ADT_LAN = ('{"data":[{"MACAddress":"bf:75:44:4c:dc:9e","IPAddress":"192.168.5.1",'
+                        '"IPSubnetMask":"255.255.255.0","DHCPv4Enable":"1","stack":"1,0,0,0,0,0"}],'
+                        '"operation":"gl","oid":"DEV2_ADT_LAN","success":true}')
+        DEV2_ADT_WAN = ('{"data":[{"enable":"1","MACAddr":"ba:7a:a4:4a:dc:7e","connIPv4Address":"0.0.0.0",'
+                        '"connIPv4Gateway":"0.0.0.0","name":"ipoe_0_0_d","connIPv4SubnetMask":"0.0.0.0",'
+                        '"connIPv4DnsServer":"0.0.0.0,0.0.0.0","stack":"1,0,0,0,0,0"}],"operation":"gl",'
+                        '"oid":"DEV2_ADT_WAN","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                if 'DEV2_ADT_LAN' in data_str:
+                    return 200, DEV2_ADT_LAN
+                elif 'DEV2_ADT_WAN' in data_str:
+                    return 200, DEV2_ADT_WAN
+                raise ClientException()
+
+        client = TPLinkEXClientTest('', '')
+        result = client.get_ipv4_status()
+
+        self.assertIsInstance(result, IPv4Status)
+        self.assertEqual(result.wan_macaddr, 'BA-7A-A4-4A-DC-7E')
+        self.assertEqual(result.wan_ipv4_ipaddr, '0.0.0.0')
+        self.assertEqual(result.wan_ipv4_gateway, '0.0.0.0')
+        self.assertEqual(result.wan_ipv4_conntype, 'ipoe_0_0_d')
+        self.assertEqual(result.wan_ipv4_netmask, '0.0.0.0')
+        self.assertEqual(result.wan_ipv4_pridns, '0.0.0.0')
+        self.assertEqual(result.wan_ipv4_snddns, '0.0.0.0')
+        self.assertEqual(result.lan_macaddr, 'BF-75-44-4C-DC-9E')
+        self.assertEqual(result.lan_ipv4_ipaddr, '192.168.5.1')
+        self.assertEqual(result.lan_ipv4_netmask, '255.255.255.0')
+        self.assertEqual(result.lan_ipv4_dhcp_enable, True)
+        self.assertEqual(result.remote, None)
+
+    def test_set_wifi(self) -> None:
+        response = '{"success":true, "errorcode":0}'
+
+        check_url = ''
+        check_data = ''
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            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_wifi(Connection.HOST_2G, True)
+
+        self.assertIn('http:///cgi_gdpr?9?_=', check_url)
+        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"}')
+
+
+if __name__ == '__main__':
+    main()