core/tests/components/device_tracker/test_tplink.py

"""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."""
        self.hass.stop()
        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 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)