tplinkrouterc6u 4.2.0__py3-none-any.whl → 4.2.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
test/test_client.py CHANGED
@@ -1,7 +1,7 @@
1
- import unittest
2
- import macaddress
3
- import ipaddress
4
- import json
1
+ from unittest import main, TestCase
2
+ from macaddress import EUI48
3
+ from ipaddress import IPv4Address
4
+ from json import loads
5
5
  from tplinkrouterc6u import (
6
6
  TplinkRouter,
7
7
  Connection,
@@ -11,7 +11,7 @@ from tplinkrouterc6u import (
11
11
  )
12
12
 
13
13
 
14
- class TestTPLinkClient(unittest.TestCase):
14
+ class TestTPLinkClient(TestCase):
15
15
  def test_get_status(self) -> None:
16
16
  response_status = '''
17
17
  {
@@ -53,6 +53,12 @@ class TestTPLinkClient(unittest.TestCase):
53
53
  "macaddr": "06:82:9d:2b:8f:c6",
54
54
  "ipaddr": "192.168.1.186",
55
55
  "hostname": "UNKNOWN"
56
+ },
57
+ {
58
+ "wire_type": "2.4G",
59
+ "macaddr": "06:55:9d:2b:8f:a7",
60
+ "ipaddr": "Unknown",
61
+ "hostname": "Unknown"
56
62
  }
57
63
  ],
58
64
  "guest_5g_psk_key": "",
@@ -206,9 +212,9 @@ class TestTPLinkClient(unittest.TestCase):
206
212
  def request(self, path: str, data: str,
207
213
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
208
214
  if path == 'admin/status?form=all&operation=read':
209
- return json.loads(response_status)['data']
215
+ return loads(response_status)['data']
210
216
  elif path == 'admin/wireless?form=statistics':
211
- return json.loads(response_stats)['data']
217
+ return loads(response_stats)['data']
212
218
  raise ClientException()
213
219
 
214
220
  client = TPLinkRouterTest('', '')
@@ -216,18 +222,18 @@ class TestTPLinkClient(unittest.TestCase):
216
222
 
217
223
  self.assertIsInstance(status, Status)
218
224
  self.assertEqual(status.wan_macaddr, 'D6-0B-40-57-DA-60')
219
- self.assertIsInstance(status.wan_macaddress, macaddress.EUI48)
225
+ self.assertIsInstance(status.wan_macaddress, EUI48)
220
226
  self.assertEqual(status.lan_macaddr, '06-E6-97-9E-23-F5')
221
- self.assertIsInstance(status.lan_macaddress, macaddress.EUI48)
227
+ self.assertIsInstance(status.lan_macaddress, EUI48)
222
228
  self.assertEqual(status.wan_ipv4_addr, '192.168.1.100')
223
- self.assertIsInstance(status.lan_ipv4_address, ipaddress.IPv4Address)
229
+ self.assertIsInstance(status.lan_ipv4_address, IPv4Address)
224
230
  self.assertEqual(status.lan_ipv4_addr, '192.168.1.100')
225
231
  self.assertEqual(status.wan_ipv4_gateway, '192.168.1.254')
226
- self.assertIsInstance(status.wan_ipv4_address, ipaddress.IPv4Address)
232
+ self.assertIsInstance(status.wan_ipv4_address, IPv4Address)
227
233
  self.assertEqual(status.wired_total, 2)
228
- self.assertEqual(status.wifi_clients_total, 2)
234
+ self.assertEqual(status.wifi_clients_total, 3)
229
235
  self.assertEqual(status.guest_clients_total, 0)
230
- self.assertEqual(status.clients_total, 4)
236
+ self.assertEqual(status.clients_total, 5)
231
237
  self.assertEqual(status.iot_clients_total, None)
232
238
  self.assertEqual(status.guest_2g_enable, False)
233
239
  self.assertEqual(status.guest_5g_enable, False)
@@ -238,22 +244,22 @@ class TestTPLinkClient(unittest.TestCase):
238
244
  self.assertEqual(status.wan_ipv4_uptime, None)
239
245
  self.assertEqual(status.mem_usage, 0.43)
240
246
  self.assertEqual(status.cpu_usage, 0.28)
241
- self.assertEqual(len(status.devices), 4)
247
+ self.assertEqual(len(status.devices), 5)
242
248
  self.assertIsInstance(status.devices[0], Device)
243
249
  self.assertEqual(status.devices[0].type, Connection.WIRED)
244
250
  self.assertEqual(status.devices[0].macaddr, '3D-24-25-24-30-79')
245
- self.assertIsInstance(status.devices[0].macaddress, macaddress.EUI48)
251
+ self.assertIsInstance(status.devices[0].macaddress, EUI48)
246
252
  self.assertEqual(status.devices[0].ipaddr, '192.168.1.228')
247
- self.assertIsInstance(status.devices[0].ipaddress, ipaddress.IPv4Address)
253
+ self.assertIsInstance(status.devices[0].ipaddress, IPv4Address)
248
254
  self.assertEqual(status.devices[0].hostname, 'SERVER')
249
255
  self.assertEqual(status.devices[0].packets_sent, None)
250
256
  self.assertEqual(status.devices[0].packets_received, None)
251
257
  self.assertIsInstance(status.devices[0], Device)
252
258
  self.assertEqual(status.devices[1].type, Connection.WIRED)
253
259
  self.assertEqual(status.devices[1].macaddr, 'AC-04-D6-25-2A-96')
254
- self.assertIsInstance(status.devices[1].macaddress, macaddress.EUI48)
260
+ self.assertIsInstance(status.devices[1].macaddress, EUI48)
255
261
  self.assertEqual(status.devices[1].ipaddr, '192.168.1.254')
256
- self.assertIsInstance(status.devices[1].ipaddress, ipaddress.IPv4Address)
262
+ self.assertIsInstance(status.devices[1].ipaddress, IPv4Address)
257
263
  self.assertEqual(status.devices[1].hostname, 'UNKNOWN')
258
264
  self.assertEqual(status.devices[1].packets_sent, None)
259
265
  self.assertEqual(status.devices[1].packets_received, None)
@@ -265,12 +271,19 @@ class TestTPLinkClient(unittest.TestCase):
265
271
  self.assertEqual(status.devices[2].packets_sent, 450333)
266
272
  self.assertEqual(status.devices[2].packets_received, 4867482)
267
273
  self.assertIsInstance(status.devices[3], Device)
268
- self.assertEqual(status.devices[3].type, Connection.HOST_5G)
269
- self.assertEqual(status.devices[3].macaddr, '1F-7A-BD-F7-20-0D')
274
+ self.assertEqual(status.devices[3].type, Connection.HOST_2G)
275
+ self.assertEqual(status.devices[3].macaddr, '06-55-9D-2B-8F-A7')
270
276
  self.assertEqual(status.devices[3].ipaddr, '0.0.0.0')
271
- self.assertEqual(status.devices[3].hostname, '')
272
- self.assertEqual(status.devices[3].packets_sent, 134815)
273
- self.assertEqual(status.devices[3].packets_received, 2953078)
277
+ self.assertEqual(status.devices[3].hostname, 'Unknown')
278
+ self.assertEqual(status.devices[3].packets_sent, None)
279
+ self.assertEqual(status.devices[3].packets_received, None)
280
+ self.assertIsInstance(status.devices[4], Device)
281
+ self.assertEqual(status.devices[4].type, Connection.HOST_5G)
282
+ self.assertEqual(status.devices[4].macaddr, '1F-7A-BD-F7-20-0D')
283
+ self.assertEqual(status.devices[4].ipaddr, '0.0.0.0')
284
+ self.assertEqual(status.devices[4].hostname, '')
285
+ self.assertEqual(status.devices[4].packets_sent, 134815)
286
+ self.assertEqual(status.devices[4].packets_received, 2953078)
274
287
 
275
288
  def test_get_status_ax_55(self) -> None:
276
289
  response_status = '''
@@ -321,9 +334,9 @@ class TestTPLinkClient(unittest.TestCase):
321
334
  def request(self, path: str, data: str,
322
335
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
323
336
  if path == 'admin/status?form=all&operation=read':
324
- return json.loads(response_status)['data']
337
+ return loads(response_status)['data']
325
338
  elif path == 'admin/wireless?form=statistics':
326
- return json.loads(response_stats)['data']
339
+ return loads(response_stats)['data']
327
340
  raise ClientException()
328
341
 
329
342
  client = TPLinkRouterTest('', '')
@@ -332,7 +345,7 @@ class TestTPLinkClient(unittest.TestCase):
332
345
  self.assertIsInstance(status, Status)
333
346
  self.assertEqual(status.wan_macaddr, None)
334
347
  self.assertEqual(status.lan_macaddr, '06-E6-97-9E-23-F5')
335
- self.assertIsInstance(status.lan_macaddress, macaddress.EUI48)
348
+ self.assertIsInstance(status.lan_macaddress, EUI48)
336
349
  self.assertEqual(status.wan_ipv4_addr, None)
337
350
  self.assertEqual(status.lan_ipv4_addr, None)
338
351
  self.assertEqual(status.wan_ipv4_gateway, None)
@@ -357,27 +370,27 @@ class TestTPLinkClient(unittest.TestCase):
357
370
  self.assertIsInstance(status.devices[0], Device)
358
371
  self.assertEqual(status.devices[0].type, Connection.WIRED)
359
372
  self.assertEqual(status.devices[0].macaddr, '3D-24-25-24-30-79')
360
- self.assertIsInstance(status.devices[0].macaddress, macaddress.EUI48)
373
+ self.assertIsInstance(status.devices[0].macaddress, EUI48)
361
374
  self.assertEqual(status.devices[0].ipaddr, '192.168.1.228')
362
- self.assertIsInstance(status.devices[0].ipaddress, ipaddress.IPv4Address)
375
+ self.assertIsInstance(status.devices[0].ipaddress, IPv4Address)
363
376
  self.assertEqual(status.devices[0].hostname, 'SERVER')
364
377
  self.assertEqual(status.devices[0].packets_sent, None)
365
378
  self.assertEqual(status.devices[0].packets_received, None)
366
379
  self.assertIsInstance(status.devices[1], Device)
367
380
  self.assertEqual(status.devices[1].type, Connection.WIRED)
368
381
  self.assertEqual(status.devices[1].macaddr, 'AC-04-D6-25-2A-96')
369
- self.assertIsInstance(status.devices[1].macaddress, macaddress.EUI48)
382
+ self.assertIsInstance(status.devices[1].macaddress, EUI48)
370
383
  self.assertEqual(status.devices[1].ipaddr, '192.168.1.254')
371
- self.assertIsInstance(status.devices[1].ipaddress, ipaddress.IPv4Address)
384
+ self.assertIsInstance(status.devices[1].ipaddress, IPv4Address)
372
385
  self.assertEqual(status.devices[1].hostname, 'UNKNOWN')
373
386
  self.assertEqual(status.devices[1].packets_sent, None)
374
387
  self.assertEqual(status.devices[1].packets_received, None)
375
388
  self.assertIsInstance(status.devices[2], Device)
376
389
  self.assertEqual(status.devices[2].type, Connection.HOST_2G)
377
390
  self.assertEqual(status.devices[2].macaddr, '06-82-9D-2B-8F-C6')
378
- self.assertIsInstance(status.devices[2].macaddress, macaddress.EUI48)
391
+ self.assertIsInstance(status.devices[2].macaddress, EUI48)
379
392
  self.assertEqual(status.devices[2].ipaddr, '192.168.1.186')
380
- self.assertIsInstance(status.devices[2].ipaddress, ipaddress.IPv4Address)
393
+ self.assertIsInstance(status.devices[2].ipaddress, IPv4Address)
381
394
  self.assertEqual(status.devices[2].hostname, 'UNKNOWN')
382
395
  self.assertEqual(status.devices[2].packets_sent, None)
383
396
  self.assertEqual(status.devices[2].packets_received, None)
@@ -425,11 +438,14 @@ class TestTPLinkClient(unittest.TestCase):
425
438
  response_game_accelerator = '''
426
439
  {
427
440
  "data": [
428
- {"mac": "06:82:9d:2b:8f:c6", "deviceTag":"2.4G", "isGuest":false, "ip":"192.168.1.186", "deviceName":"name1",
429
- "uploadSpeed":12, "downloadSpeed":77},
430
- {"mac": "fb:90:b8:2a:8a:b1", "deviceTag":"iot_2.4G", "isGuest":false, "ip":"192.168.1.187", "deviceName":"name2"},
431
- {"mac": "54:b3:a2:f7:be:ea", "deviceTag":"iot_5G", "isGuest":false, "ip":"192.168.1.188", "deviceName":"name3"},
432
- {"mac": "3c:ae:e1:83:94:9d", "deviceTag":"iot_6G", "isGuest":false, "ip":"192.168.1.189", "deviceName":"name4"}
441
+ {"mac": "06:82:9d:2b:8f:c6", "deviceTag":"2.4G", "isGuest":false, "ip":"192.168.1.186",
442
+ "deviceName":"name1", "uploadSpeed":12, "downloadSpeed":77},
443
+ {"mac": "fb:90:b8:2a:8a:b1", "deviceTag":"iot_2.4G", "isGuest":false, "ip":"192.168.1.187",
444
+ "deviceName":"name2"},
445
+ {"mac": "54:b3:a2:f7:be:ea", "deviceTag":"iot_5G", "isGuest":false, "ip":"192.168.1.188",
446
+ "deviceName":"name3"},
447
+ {"mac": "3c:ae:e1:83:94:9d", "deviceTag":"iot_6G", "isGuest":false, "ip":"192.168.1.189",
448
+ "deviceName":"name4"}
433
449
  ],
434
450
  "timeout": false,
435
451
  "success": true
@@ -463,11 +479,11 @@ class TestTPLinkClient(unittest.TestCase):
463
479
  def request(self, path: str, data: str,
464
480
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
465
481
  if path == 'admin/status?form=all&operation=read':
466
- return json.loads(response_status)['data']
482
+ return loads(response_status)['data']
467
483
  elif path == 'admin/smart_network?form=game_accelerator':
468
- return json.loads(response_game_accelerator)['data']
484
+ return loads(response_game_accelerator)['data']
469
485
  elif path == 'admin/wireless?form=statistics':
470
- return json.loads(response_stats)['data']
486
+ return loads(response_stats)['data']
471
487
  raise ClientException()
472
488
 
473
489
  client = TPLinkRouterTest('', '')
@@ -476,7 +492,7 @@ class TestTPLinkClient(unittest.TestCase):
476
492
  self.assertIsInstance(status, Status)
477
493
  self.assertEqual(status.wan_macaddr, None)
478
494
  self.assertEqual(status.lan_macaddr, '06-E6-97-9E-23-F5')
479
- self.assertIsInstance(status.lan_macaddress, macaddress.EUI48)
495
+ self.assertIsInstance(status.lan_macaddress, EUI48)
480
496
  self.assertEqual(status.wan_ipv4_addr, None)
481
497
  self.assertEqual(status.lan_ipv4_addr, None)
482
498
  self.assertEqual(status.wan_ipv4_gateway, None)
@@ -501,27 +517,27 @@ class TestTPLinkClient(unittest.TestCase):
501
517
  self.assertIsInstance(status.devices[0], Device)
502
518
  self.assertEqual(status.devices[0].type, Connection.WIRED)
503
519
  self.assertEqual(status.devices[0].macaddr, '3D-24-25-24-30-79')
504
- self.assertIsInstance(status.devices[0].macaddress, macaddress.EUI48)
520
+ self.assertIsInstance(status.devices[0].macaddress, EUI48)
505
521
  self.assertEqual(status.devices[0].ipaddr, '192.168.1.228')
506
- self.assertIsInstance(status.devices[0].ipaddress, ipaddress.IPv4Address)
522
+ self.assertIsInstance(status.devices[0].ipaddress, IPv4Address)
507
523
  self.assertEqual(status.devices[0].hostname, 'SERVER')
508
524
  self.assertEqual(status.devices[0].packets_sent, None)
509
525
  self.assertEqual(status.devices[0].packets_received, None)
510
526
  self.assertIsInstance(status.devices[1], Device)
511
527
  self.assertEqual(status.devices[1].type, Connection.WIRED)
512
528
  self.assertEqual(status.devices[1].macaddr, 'AC-04-D6-25-2A-96')
513
- self.assertIsInstance(status.devices[1].macaddress, macaddress.EUI48)
529
+ self.assertIsInstance(status.devices[1].macaddress, EUI48)
514
530
  self.assertEqual(status.devices[1].ipaddr, '192.168.1.254')
515
- self.assertIsInstance(status.devices[1].ipaddress, ipaddress.IPv4Address)
531
+ self.assertIsInstance(status.devices[1].ipaddress, IPv4Address)
516
532
  self.assertEqual(status.devices[1].hostname, 'UNKNOWN')
517
533
  self.assertEqual(status.devices[1].packets_sent, None)
518
534
  self.assertEqual(status.devices[1].packets_received, None)
519
535
  self.assertIsInstance(status.devices[2], Device)
520
536
  self.assertEqual(status.devices[2].type, Connection.HOST_2G)
521
537
  self.assertEqual(status.devices[2].macaddr, '06-82-9D-2B-8F-C6')
522
- self.assertIsInstance(status.devices[2].macaddress, macaddress.EUI48)
538
+ self.assertIsInstance(status.devices[2].macaddress, EUI48)
523
539
  self.assertEqual(status.devices[2].ipaddr, '192.168.1.186')
524
- self.assertIsInstance(status.devices[2].ipaddress, ipaddress.IPv4Address)
540
+ self.assertIsInstance(status.devices[2].ipaddress, IPv4Address)
525
541
  self.assertEqual(status.devices[2].hostname, 'UNKNOWN')
526
542
  self.assertEqual(status.devices[2].packets_sent, 450333)
527
543
  self.assertEqual(status.devices[2].packets_received, 4867482)
@@ -530,36 +546,36 @@ class TestTPLinkClient(unittest.TestCase):
530
546
  self.assertIsInstance(status.devices[3], Device)
531
547
  self.assertEqual(status.devices[3].type, Connection.IOT_2G)
532
548
  self.assertEqual(status.devices[3].macaddr, 'FB-90-B8-2A-8A-B1')
533
- self.assertIsInstance(status.devices[3].macaddress, macaddress.EUI48)
549
+ self.assertIsInstance(status.devices[3].macaddress, EUI48)
534
550
  self.assertEqual(status.devices[3].ipaddr, '192.168.1.187')
535
- self.assertIsInstance(status.devices[3].ipaddress, ipaddress.IPv4Address)
551
+ self.assertIsInstance(status.devices[3].ipaddress, IPv4Address)
536
552
  self.assertEqual(status.devices[3].hostname, 'name2')
537
553
  self.assertEqual(status.devices[3].packets_sent, None)
538
554
  self.assertEqual(status.devices[3].packets_received, None)
539
555
  self.assertIsInstance(status.devices[4], Device)
540
556
  self.assertEqual(status.devices[4].type, Connection.IOT_5G)
541
557
  self.assertEqual(status.devices[4].macaddr, '54-B3-A2-F7-BE-EA')
542
- self.assertIsInstance(status.devices[4].macaddress, macaddress.EUI48)
558
+ self.assertIsInstance(status.devices[4].macaddress, EUI48)
543
559
  self.assertEqual(status.devices[4].ipaddr, '192.168.1.188')
544
- self.assertIsInstance(status.devices[4].ipaddress, ipaddress.IPv4Address)
560
+ self.assertIsInstance(status.devices[4].ipaddress, IPv4Address)
545
561
  self.assertEqual(status.devices[4].hostname, 'name3')
546
562
  self.assertEqual(status.devices[4].packets_sent, None)
547
563
  self.assertEqual(status.devices[4].packets_received, None)
548
564
  self.assertIsInstance(status.devices[5], Device)
549
565
  self.assertEqual(status.devices[5].type, Connection.IOT_6G)
550
566
  self.assertEqual(status.devices[5].macaddr, '3C-AE-E1-83-94-9D')
551
- self.assertIsInstance(status.devices[5].macaddress, macaddress.EUI48)
567
+ self.assertIsInstance(status.devices[5].macaddress, EUI48)
552
568
  self.assertEqual(status.devices[5].ipaddr, '192.168.1.189')
553
- self.assertIsInstance(status.devices[5].ipaddress, ipaddress.IPv4Address)
569
+ self.assertIsInstance(status.devices[5].ipaddress, IPv4Address)
554
570
  self.assertEqual(status.devices[5].hostname, 'name4')
555
571
  self.assertEqual(status.devices[5].packets_sent, None)
556
572
  self.assertEqual(status.devices[5].packets_received, None)
557
573
  self.assertIsInstance(status.devices[6], Device)
558
574
  self.assertEqual(status.devices[6].type, Connection.HOST_5G)
559
575
  self.assertEqual(status.devices[6].macaddr, '1F-7A-BD-F7-20-0D')
560
- self.assertIsInstance(status.devices[6].macaddress, macaddress.EUI48)
576
+ self.assertIsInstance(status.devices[6].macaddress, EUI48)
561
577
  self.assertEqual(status.devices[6].ipaddr, '0.0.0.0')
562
- self.assertIsInstance(status.devices[6].ipaddress, ipaddress.IPv4Address)
578
+ self.assertIsInstance(status.devices[6].ipaddress, IPv4Address)
563
579
  self.assertEqual(status.devices[6].hostname, '')
564
580
  self.assertEqual(status.devices[6].packets_sent, 134815)
565
581
  self.assertEqual(status.devices[6].packets_received, 2953078)
@@ -596,11 +612,11 @@ class TestTPLinkClient(unittest.TestCase):
596
612
  def request(self, path: str, data: str,
597
613
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
598
614
  if path == 'admin/status?form=all&operation=read':
599
- return json.loads(response_status)['data']
615
+ return loads(response_status)['data']
600
616
  elif path == 'admin/status?form=perf&operation=read':
601
- return json.loads(perf_stats)['data']
617
+ return loads(perf_stats)['data']
602
618
  elif path == 'admin/wireless?form=statistics':
603
- return json.loads(response_stats)['data']
619
+ return loads(response_stats)['data']
604
620
  raise ClientException()
605
621
 
606
622
  client = TPLinkRouterTest('', '')
@@ -609,7 +625,7 @@ class TestTPLinkClient(unittest.TestCase):
609
625
  self.assertIsInstance(status, Status)
610
626
  self.assertEqual(status.wan_macaddr, None)
611
627
  self.assertEqual(status.lan_macaddr, '06-E6-97-9E-23-F5')
612
- self.assertIsInstance(status.lan_macaddress, macaddress.EUI48)
628
+ self.assertIsInstance(status.lan_macaddress, EUI48)
613
629
  self.assertEqual(status.wan_ipv4_addr, None)
614
630
  self.assertEqual(status.lan_ipv4_addr, None)
615
631
  self.assertEqual(status.wan_ipv4_gateway, None)
@@ -679,4 +695,4 @@ class TestTPLinkClient(unittest.TestCase):
679
695
 
680
696
 
681
697
  if __name__ == '__main__':
682
- unittest.main()
698
+ main()
test/test_client_c1200.py CHANGED
@@ -1,12 +1,13 @@
1
- import unittest
2
- import json
1
+ from unittest import main, TestCase
2
+ from json import loads
3
3
  from tplinkrouterc6u import (
4
4
  TplinkC1200Router,
5
5
  Connection,
6
6
  ClientException
7
7
  )
8
8
 
9
- class TestTPLinkC1200Client(unittest.TestCase):
9
+
10
+ class TestTPLinkC1200Client(TestCase):
10
11
 
11
12
  def test_set_led_on(self) -> None:
12
13
 
@@ -30,18 +31,18 @@ class TestTPLinkC1200Client(unittest.TestCase):
30
31
  def request(self, path: str, data: str,
31
32
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
32
33
  if path == 'admin/ledgeneral?form=setting&operation=read':
33
- return json.loads(response_led_general_read)
34
+ return loads(response_led_general_read)
34
35
  if path == 'admin/ledgeneral?form=setting&operation=write':
35
36
  self.captured_path = path
36
- return json.loads(response_led_general_write)
37
+ return loads(response_led_general_write)
37
38
  raise ClientException()
38
39
 
39
40
  client = TPLinkRouterTest('', '')
40
-
41
+
41
42
  client.set_led(True)
42
43
 
43
44
  expected_path = "admin/ledgeneral?form=setting&operation=write"
44
-
45
+
45
46
  self.assertEqual(client.captured_path, expected_path)
46
47
 
47
48
  def test_set_led_off(self) -> None:
@@ -66,10 +67,10 @@ class TestTPLinkC1200Client(unittest.TestCase):
66
67
  def request(self, path: str, data: str,
67
68
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
68
69
  if path == 'admin/ledgeneral?form=setting&operation=read':
69
- return json.loads(response_led_general_read)
70
+ return loads(response_led_general_read)
70
71
  elif path == 'admin/ledgeneral?form=setting&operation=write':
71
72
  self.captured_path = path
72
- return json.loads(response_led_general_write)
73
+ return loads(response_led_general_write)
73
74
  raise ClientException()
74
75
 
75
76
  client = TPLinkRouterTest('', '')
@@ -77,7 +78,7 @@ class TestTPLinkC1200Client(unittest.TestCase):
77
78
  client.set_led(False)
78
79
 
79
80
  expected_path = "admin/ledgeneral?form=setting&operation=write"
80
-
81
+
81
82
  self.assertEqual(client.captured_path, expected_path)
82
83
 
83
84
  def test_led_status(self) -> None:
@@ -89,12 +90,12 @@ class TestTPLinkC1200Client(unittest.TestCase):
89
90
  "ledpm_support": "yes"
90
91
  }
91
92
  '''
92
-
93
+
93
94
  class TPLinkRouterTest(TplinkC1200Router):
94
95
  def request(self, path: str, data: str,
95
96
  ignore_response: bool = False, ignore_errors: bool = False) -> dict | None:
96
97
  if path == 'admin/ledgeneral?form=setting&operation=read':
97
- return json.loads(response_led_general_read)
98
+ return loads(response_led_general_read)
98
99
  raise ClientException()
99
100
 
100
101
  client = TPLinkRouterTest('', '')
@@ -102,7 +103,6 @@ class TestTPLinkC1200Client(unittest.TestCase):
102
103
  led_status = client.get_led()
103
104
  self.assertTrue(led_status)
104
105
 
105
-
106
106
  def test_set_wifi(self) -> None:
107
107
 
108
108
  class TPLinkRouterTest(TplinkC1200Router):
@@ -128,14 +128,15 @@ class TestTPLinkC1200Client(unittest.TestCase):
128
128
  txpower="20",
129
129
  disabled_all="no"
130
130
  )
131
-
131
+
132
132
  expected_data = ("operation=write&enable=on&ssid=TestSSID&hidden=no&encryption=WPA3-PSK&"
133
133
  "psk_version=2&psk_cipher=AES&psk_key=testkey123&hwmode=11ac&"
134
134
  "htmode=VHT20&channel=36&txpower=20&disabled_all=no")
135
135
  expected_path = f"admin/wireless?form=wireless_5g&{expected_data}"
136
-
136
+
137
137
  self.assertEqual(client.captured_path, expected_path)
138
138
  self.assertEqual(client.captured_data, expected_data)
139
139
 
140
+
140
141
  if __name__ == '__main__':
141
- unittest.main()
142
+ main()