Fix warning in test

This commit is contained in:
Paulus Schoutsen 2016-11-22 08:50:28 -08:00 committed by Paulus Schoutsen
parent 8e776b4dc0
commit 00019b9ff0

View file

@ -292,7 +292,7 @@ class TestHelpersEntityComponent(unittest.TestCase):
assert platform2_setup.called
@patch('homeassistant.helpers.entity_component.EntityComponent'
'._async_setup_platform')
'._async_setup_platform', return_value=mock_coro()())
@patch('homeassistant.bootstrap.async_setup_component',
return_value=mock_coro(True)())
def test_setup_does_discovery(self, mock_setup_component, mock_setup):