"""NuHeat component tests."""
import unittest
from unittest.mock import patch

from homeassistant.components import nuheat

from tests.common import MockDependency, get_test_home_assistant

VALID_CONFIG = {
    "nuheat": {"username": "warm", "password": "feet", "devices": "thermostat123"}
}


class TestNuHeat(unittest.TestCase):
    """Test the NuHeat component."""

    def setUp(self):  # pylint: disable=invalid-name
        """Initialize the values for this test class."""
        self.hass = get_test_home_assistant()
        self.config = VALID_CONFIG

    def tearDown(self):  # pylint: disable=invalid-name
        """Teardown this test class. Stop hass."""
        self.hass.stop()

    @MockDependency("nuheat")
    @patch("homeassistant.helpers.discovery.load_platform")
    def test_setup(self, mocked_nuheat, mocked_load):
        """Test setting up the NuHeat component."""
        with patch.object(nuheat, "nuheat", mocked_nuheat):
            nuheat.setup(self.hass, self.config)

        mocked_nuheat.NuHeat.assert_called_with("warm", "feet")
        assert nuheat.DOMAIN in self.hass.data
        assert len(self.hass.data[nuheat.DOMAIN]) == 2
        assert isinstance(
            self.hass.data[nuheat.DOMAIN][0], type(mocked_nuheat.NuHeat())
        )
        assert self.hass.data[nuheat.DOMAIN][1] == "thermostat123"

        mocked_load.assert_called_with(
            self.hass, "climate", nuheat.DOMAIN, {}, self.config
        )