45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
"""NuHeat component tests."""
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from unittest.mock import patch
|
||
|
|
from tests.common import get_test_home_assistant, MockDependency
|
||
|
|
|
||
|
|
from homeassistant.components import nuheat
|
||
|
|
|
||
|
|
VALID_CONFIG = {
|
||
|
|
"nuheat": {
|
||
|
|
"username": "warm",
|
||
|
|
"password": "feet",
|
||
|
|
"devices": "thermostat123"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class TestNuHeat(unittest.TestCase):
|
||
|
|
"""Test the NuHeat component."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Initialize the values for this test class."""
|
||
|
|
self.hass = get_test_home_assistant()
|
||
|
|
self.config = VALID_CONFIG
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
"""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."""
|
||
|
|
nuheat.setup(self.hass, self.config)
|
||
|
|
|
||
|
|
mocked_nuheat.NuHeat.assert_called_with("warm", "feet")
|
||
|
|
self.assertIn(nuheat.DOMAIN, self.hass.data)
|
||
|
|
self.assertEquals(2, len(self.hass.data[nuheat.DOMAIN]))
|
||
|
|
self.assertEquals(self.hass.data[nuheat.DOMAIN][0], "thermostat123")
|
||
|
|
self.assertEquals(self.hass.data[nuheat.DOMAIN][1], "thermostat123")
|
||
|
|
|
||
|
|
mocked_load.assert_called_with(
|
||
|
|
self.hass, "climate", nuheat.DOMAIN, {}, self.config
|
||
|
|
)
|