Add new number entity integration (#42735)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
This commit is contained in:
Shulyaka 2020-12-02 15:50:48 +03:00 committed by GitHub
parent 6c9c280bbb
commit f744f7c34e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 406 additions and 1 deletions

View file

@ -0,0 +1,39 @@
"""The tests for the Number component."""
from unittest.mock import MagicMock
from homeassistant.components.number import NumberEntity
class MockNumberEntity(NumberEntity):
"""Mock NumberEntity device to use in tests."""
@property
def max_value(self) -> float:
"""Return the max value."""
return 1.0
@property
def state(self):
"""Return the current value."""
return "0.5"
async def test_step(hass):
"""Test the step calculation."""
number = NumberEntity()
assert number.step == 1.0
number_2 = MockNumberEntity()
assert number_2.step == 0.1
async def test_sync_set_value(hass):
"""Test if async set_value calls sync set_value."""
number = NumberEntity()
number.hass = hass
number.set_value = MagicMock()
await number.async_set_value(42)
assert number.set_value.called
assert number.set_value.call_args[0][0] == 42