"""The tests for Efergy sensor platform."""
import unittest

import requests_mock

from homeassistant.setup import setup_component

from tests.common import get_test_home_assistant, load_fixture

token = "9p6QGJ7dpZfO3fqPTBk1fyEmjV1cGoLT"
multi_sensor_token = "9r6QGF7dpZfO3fqPTBl1fyRmjV1cGoLT"

ONE_SENSOR_CONFIG = {
    "platform": "efergy",
    "app_token": token,
    "utc_offset": "300",
    "monitored_variables": [
        {"type": "amount", "period": "day"},
        {"type": "instant_readings"},
        {"type": "budget"},
        {"type": "cost", "period": "day", "currency": "$"},
        {"type": "current_values"},
    ],
}

MULTI_SENSOR_CONFIG = {
    "platform": "efergy",
    "app_token": multi_sensor_token,
    "utc_offset": "300",
    "monitored_variables": [{"type": "current_values"}],
}


def mock_responses(mock):
    """Mock responses for Efergy."""
    base_url = "https://engage.efergy.com/mobile_proxy/"
    mock.get(
        f"{base_url}getInstant?token={token}", text=load_fixture("efergy_instant.json"),
    )
    mock.get(
        f"{base_url}getEnergy?token={token}&offset=300&period=day",
        text=load_fixture("efergy_energy.json"),
    )
    mock.get(
        f"{base_url}getBudget?token={token}", text=load_fixture("efergy_budget.json"),
    )
    mock.get(
        f"{base_url}getCost?token={token}&offset=300&period=day",
        text=load_fixture("efergy_cost.json"),
    )
    mock.get(
        f"{base_url}getCurrentValuesSummary?token={token}",
        text=load_fixture("efergy_current_values_single.json"),
    )
    mock.get(
        f"{base_url}getCurrentValuesSummary?token={multi_sensor_token}",
        text=load_fixture("efergy_current_values_multi.json"),
    )


class TestEfergySensor(unittest.TestCase):
    """Tests the Efergy Sensor platform."""

    DEVICES = []

    @requests_mock.Mocker()
    def add_entities(self, devices, mock):
        """Mock add devices."""
        mock_responses(mock)
        for device in devices:
            device.update()
            self.DEVICES.append(device)

    def setUp(self):
        """Initialize values for this test case class."""
        self.hass = get_test_home_assistant()
        self.config = ONE_SENSOR_CONFIG
        self.addCleanup(self.tear_down_cleanup)

    def tear_down_cleanup(self):
        """Stop everything that was started."""
        self.hass.stop()

    @requests_mock.Mocker()
    def test_single_sensor_readings(self, mock):
        """Test for successfully setting up the Efergy platform."""
        mock_responses(mock)
        assert setup_component(self.hass, "sensor", {"sensor": ONE_SENSOR_CONFIG})
        self.hass.block_till_done()

        assert "38.21" == self.hass.states.get("sensor.energy_consumed").state
        assert "1580" == self.hass.states.get("sensor.energy_usage").state
        assert "ok" == self.hass.states.get("sensor.energy_budget").state
        assert "5.27" == self.hass.states.get("sensor.energy_cost").state
        assert "1628" == self.hass.states.get("sensor.efergy_728386").state

    @requests_mock.Mocker()
    def test_multi_sensor_readings(self, mock):
        """Test for multiple sensors in one household."""
        mock_responses(mock)
        assert setup_component(self.hass, "sensor", {"sensor": MULTI_SENSOR_CONFIG})
        self.hass.block_till_done()

        assert "218" == self.hass.states.get("sensor.efergy_728386").state
        assert "1808" == self.hass.states.get("sensor.efergy_0").state
        assert "312" == self.hass.states.get("sensor.efergy_728387").state