"""The tests for the tplink device tracker platform."""

import os
import pytest

from homeassistant.components.device_tracker.legacy import YAML_DEVICES
from homeassistant.components.tplink.device_tracker import Tplink4DeviceScanner
from homeassistant.const import CONF_PLATFORM, CONF_PASSWORD, CONF_USERNAME, CONF_HOST
import requests_mock


@pytest.fixture(autouse=True)
def setup_comp(hass):
    """Initialize components."""
    yaml_devices = hass.config.path(YAML_DEVICES)
    yield
    if os.path.isfile(yaml_devices):
        os.remove(yaml_devices)


async def test_get_mac_addresses_from_both_bands(hass):
    """Test grabbing the mac addresses from 2.4 and 5 GHz clients pages."""
    with requests_mock.Mocker() as m:
        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]
        ]

        assert tplink.last_results == expected_mac_results