Add tplink Archer C7 device tracking support for 5Ghz networks

2026-09-28 02:18:10 -04:00 · 2016-08-21 13:09:44 -07:00
parent 49998272db
commit 2b4980ae5d
2 changed files with 84 additions and 11 deletions
@@ -0,0 +1,69 @@
+"""The tests for the tplink device tracker platform."""
+
+import os
+import unittest
+
+from homeassistant.components import device_tracker
+from homeassistant.components.device_tracker.tplink import Tplink4DeviceScanner
+from homeassistant.const import (CONF_PLATFORM, CONF_PASSWORD, CONF_USERNAME,
+                                 CONF_HOST)
+import requests_mock
+
+from tests.common import get_test_home_assistant
+
+
+class TestTplink4DeviceScanner(unittest.TestCase):
+    """Tests for the Tplink4DeviceScanner class."""
+
+    def setUp(self):  # pylint: disable=invalid-name
+        """Setup things to be run when tests are started."""
+        self.hass = get_test_home_assistant()
+
+    def tearDown(self):  # pylint: disable=invalid-name
+        """Stop everything that was started."""
+        try:
+            os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
+        except FileNotFoundError:
+            pass
+
+    @requests_mock.mock()
+    def test_get_mac_addresses_from_both_bands(self, m):
+        """
+        Test grabbing the mac addresses from both 2.4 and 5 GHz clients pages.
+        """
+        conf_dict = {
+            CONF_PLATFORM: 'tplink',
+            CONF_HOST: 'fake_host',
+            CONF_USERNAME: 'fake_user',
+            CONF_PASSWORD: 'fake_pass'
+        }
+
+        # Mock the token retrieval process
+        FAKE_TOKEN = 'fake_token'
+        fake_auth_token_response = 'window.parent.location.href = ' \
+                                   '"https://a/{}/userRpm/Index.htm";'.format(
+                                       FAKE_TOKEN)
+
+        m.get('http://{}/userRpm/LoginRpm.htm?Save=Save'.format(
+            conf_dict[CONF_HOST]), text=fake_auth_token_response)
+
+        FAKE_MAC_1 = 'CA-FC-8A-C8-BB-53'
+        FAKE_MAC_2 = '6C-48-83-21-46-8D'
+        FAKE_MAC_3 = '77-98-75-65-B1-2B'
+        mac_response_2_4 = '{} {}'.format(FAKE_MAC_1, FAKE_MAC_2)
+        mac_response_5 = '{}'.format(FAKE_MAC_3)
+
+        # Mock the 2.4 GHz clients page
+        m.get('http://{}/{}/userRpm/WlanStationRpm.htm'.format(
+            conf_dict[CONF_HOST], FAKE_TOKEN), text=mac_response_2_4)
+
+        # Mock the 5 GHz clients page
+        m.get('http://{}/{}/userRpm/WlanStationRpm_5g.htm'.format(
+            conf_dict[CONF_HOST], FAKE_TOKEN), text=mac_response_5)
+
+        tplink = Tplink4DeviceScanner(conf_dict)
+
+        expected_mac_results = [mac.replace('-', ':') for mac in
+                                [FAKE_MAC_1, FAKE_MAC_2, FAKE_MAC_3]]
+
+        self.assertEquals(tplink.last_results, expected_mac_results)