Rewrite mochad unittest to pytest style (#42164)
This commit is contained in:
Tim Werner
GitHub
parent
9a54c31c34
commit
fcc895f294
2 changed files with 70 additions and 163 deletions
|
|
@ -1,14 +1,12 @@
|
|||
"""The tests for the mochad light platform."""
|
||||
import unittest
|
||||
|
||||
import pytest
|
||||
|
||||
from homeassistant.components import light
|
||||
from homeassistant.components.mochad import light as mochad
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
import tests.async_mock as mock
|
||||
from tests.common import get_test_home_assistant
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
|
|
@ -18,122 +16,51 @@ def pymochad_mock():
|
|||
yield device
|
||||
|
||||
|
||||
class TestMochadSwitchSetup(unittest.TestCase):
|
||||
"""Test the mochad light."""
|
||||
|
||||
PLATFORM = mochad
|
||||
COMPONENT = light
|
||||
THING = "light"
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.addCleanup(self.hass.stop)
|
||||
|
||||
@mock.patch("homeassistant.components.mochad.light.MochadLight")
|
||||
def test_setup_adds_proper_devices(self, mock_light):
|
||||
"""Test if setup adds devices."""
|
||||
good_config = {
|
||||
"mochad": {},
|
||||
"light": {
|
||||
"platform": "mochad",
|
||||
"devices": [{"name": "Light1", "address": "a1"}],
|
||||
},
|
||||
}
|
||||
assert setup_component(self.hass, light.DOMAIN, good_config)
|
||||
@pytest.fixture
|
||||
def light_mock(hass, brightness):
|
||||
"""Mock light."""
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_light", "brightness_levels": brightness}
|
||||
return mochad.MochadLight(hass, controller_mock, dev_dict)
|
||||
|
||||
|
||||
class TestMochadLight(unittest.TestCase):
|
||||
"""Test for mochad light platform."""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_light", "brightness_levels": 32}
|
||||
self.light = mochad.MochadLight(self.hass, controller_mock, dev_dict)
|
||||
|
||||
def teardown_method(self, method):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_name(self):
|
||||
"""Test the name."""
|
||||
assert "fake_light" == self.light.name
|
||||
|
||||
def test_turn_on_with_no_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on()
|
||||
self.light.light.send_cmd.assert_called_once_with("on")
|
||||
|
||||
def test_turn_on_with_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on(brightness=45)
|
||||
self.light.light.send_cmd.assert_has_calls(
|
||||
[mock.call("on"), mock.call("dim 25")]
|
||||
)
|
||||
|
||||
def test_turn_off(self):
|
||||
"""Test turn_off."""
|
||||
self.light.turn_off()
|
||||
self.light.light.send_cmd.assert_called_once_with("off")
|
||||
async def test_setup_adds_proper_devices(hass):
|
||||
"""Test if setup adds devices."""
|
||||
good_config = {
|
||||
"mochad": {},
|
||||
"light": {
|
||||
"platform": "mochad",
|
||||
"devices": [{"name": "Light1", "address": "a1"}],
|
||||
},
|
||||
}
|
||||
assert await async_setup_component(hass, light.DOMAIN, good_config)
|
||||
|
||||
|
||||
class TestMochadLight256Levels(unittest.TestCase):
|
||||
"""Test for mochad light platform."""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_light", "brightness_levels": 256}
|
||||
self.light = mochad.MochadLight(self.hass, controller_mock, dev_dict)
|
||||
|
||||
def teardown_method(self, method):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_turn_on_with_no_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on()
|
||||
self.light.light.send_cmd.assert_called_once_with("xdim 255")
|
||||
|
||||
def test_turn_on_with_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on(brightness=45)
|
||||
self.light.light.send_cmd.assert_called_once_with("xdim 45")
|
||||
|
||||
def test_turn_off(self):
|
||||
"""Test turn_off."""
|
||||
self.light.turn_off()
|
||||
self.light.light.send_cmd.assert_called_once_with("off")
|
||||
@pytest.mark.parametrize(
|
||||
"brightness,expected", [(32, "on"), (256, "xdim 255"), (64, "xdim 63")]
|
||||
)
|
||||
async def test_turn_on_with_no_brightness(light_mock, expected):
|
||||
"""Test turn_on."""
|
||||
light_mock.turn_on()
|
||||
light_mock.light.send_cmd.assert_called_once_with(expected)
|
||||
|
||||
|
||||
class TestMochadLight64Levels(unittest.TestCase):
|
||||
"""Test for mochad light platform."""
|
||||
@pytest.mark.parametrize(
|
||||
"brightness,expected",
|
||||
[
|
||||
(32, [mock.call("on"), mock.call("dim 25")]),
|
||||
(256, [mock.call("xdim 45")]),
|
||||
(64, [mock.call("xdim 11")]),
|
||||
],
|
||||
)
|
||||
async def test_turn_on_with_brightness(light_mock, expected):
|
||||
"""Test turn_on."""
|
||||
light_mock.turn_on(brightness=45)
|
||||
light_mock.light.send_cmd.assert_has_calls(expected)
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_light", "brightness_levels": 64}
|
||||
self.light = mochad.MochadLight(self.hass, controller_mock, dev_dict)
|
||||
|
||||
def teardown_method(self, method):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_turn_on_with_no_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on()
|
||||
self.light.light.send_cmd.assert_called_once_with("xdim 63")
|
||||
|
||||
def test_turn_on_with_brightness(self):
|
||||
"""Test turn_on."""
|
||||
self.light.turn_on(brightness=45)
|
||||
self.light.light.send_cmd.assert_called_once_with("xdim 11")
|
||||
|
||||
def test_turn_off(self):
|
||||
"""Test turn_off."""
|
||||
self.light.turn_off()
|
||||
self.light.light.send_cmd.assert_called_once_with("off")
|
||||
@pytest.mark.parametrize("brightness", [32])
|
||||
async def test_turn_off(light_mock):
|
||||
"""Test turn_off."""
|
||||
light_mock.turn_off()
|
||||
light_mock.light.send_cmd.assert_called_once_with("off")
|
||||
|
|
|
|||
|
|
@ -1,14 +1,11 @@
|
|||
"""The tests for the mochad switch platform."""
|
||||
import unittest
|
||||
|
||||
import pytest
|
||||
|
||||
from homeassistant.components import switch
|
||||
from homeassistant.components.mochad import switch as mochad
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
import tests.async_mock as mock
|
||||
from tests.common import get_test_home_assistant
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
|
|
@ -20,55 +17,38 @@ def pymochad_mock():
|
|||
yield
|
||||
|
||||
|
||||
class TestMochadSwitchSetup(unittest.TestCase):
|
||||
"""Test the mochad switch."""
|
||||
|
||||
PLATFORM = mochad
|
||||
COMPONENT = switch
|
||||
THING = "switch"
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.addCleanup(self.hass.stop)
|
||||
|
||||
@mock.patch("homeassistant.components.mochad.switch.MochadSwitch")
|
||||
def test_setup_adds_proper_devices(self, mock_switch):
|
||||
"""Test if setup adds devices."""
|
||||
good_config = {
|
||||
"mochad": {},
|
||||
"switch": {
|
||||
"platform": "mochad",
|
||||
"devices": [{"name": "Switch1", "address": "a1"}],
|
||||
},
|
||||
}
|
||||
assert setup_component(self.hass, switch.DOMAIN, good_config)
|
||||
@pytest.fixture
|
||||
def switch_mock(hass):
|
||||
"""Mock switch."""
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_switch"}
|
||||
return mochad.MochadSwitch(hass, controller_mock, dev_dict)
|
||||
|
||||
|
||||
class TestMochadSwitch(unittest.TestCase):
|
||||
"""Test for mochad switch platform."""
|
||||
async def test_setup_adds_proper_devices(hass):
|
||||
"""Test if setup adds devices."""
|
||||
good_config = {
|
||||
"mochad": {},
|
||||
"switch": {
|
||||
"platform": "mochad",
|
||||
"devices": [{"name": "Switch1", "address": "a1"}],
|
||||
},
|
||||
}
|
||||
assert await async_setup_component(hass, switch.DOMAIN, good_config)
|
||||
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
controller_mock = mock.MagicMock()
|
||||
dev_dict = {"address": "a1", "name": "fake_switch"}
|
||||
self.switch = mochad.MochadSwitch(self.hass, controller_mock, dev_dict)
|
||||
|
||||
def teardown_method(self, method):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
async def test_name(switch_mock):
|
||||
"""Test the name."""
|
||||
assert "fake_switch" == switch_mock.name
|
||||
|
||||
def test_name(self):
|
||||
"""Test the name."""
|
||||
assert "fake_switch" == self.switch.name
|
||||
|
||||
def test_turn_on(self):
|
||||
"""Test turn_on."""
|
||||
self.switch.turn_on()
|
||||
self.switch.switch.send_cmd.assert_called_once_with("on")
|
||||
async def test_turn_on(switch_mock):
|
||||
"""Test turn_on."""
|
||||
switch_mock.turn_on()
|
||||
switch_mock.switch.send_cmd.assert_called_once_with("on")
|
||||
|
||||
def test_turn_off(self):
|
||||
"""Test turn_off."""
|
||||
self.switch.turn_off()
|
||||
self.switch.switch.send_cmd.assert_called_once_with("off")
|
||||
|
||||
async def test_turn_off(switch_mock):
|
||||
"""Test turn_off."""
|
||||
switch_mock.turn_off()
|
||||
switch_mock.switch.send_cmd.assert_called_once_with("off")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue