Clean up mock_coro (#6037)
This commit is contained in:
parent
9c176ad85a
commit
75e41a21c9
12 changed files with 37 additions and 42 deletions
|
@ -298,9 +298,9 @@ class TestHelpersEntityComponent(unittest.TestCase):
|
|||
assert platform2_setup.called
|
||||
|
||||
@patch('homeassistant.helpers.entity_component.EntityComponent'
|
||||
'._async_setup_platform', return_value=mock_coro()())
|
||||
'._async_setup_platform', return_value=mock_coro())
|
||||
@patch('homeassistant.bootstrap.async_setup_component',
|
||||
return_value=mock_coro(True)())
|
||||
return_value=mock_coro(True))
|
||||
def test_setup_does_discovery(self, mock_setup_component, mock_setup):
|
||||
"""Test setup for discovery."""
|
||||
component = EntityComponent(_LOGGER, DOMAIN, self.hass)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue