diff --git a/tests/components/test_api.py b/tests/components/test_api.py index c8d8fa50ae1..644982e18fc 100644 --- a/tests/components/test_api.py +++ b/tests/components/test_api.py @@ -422,11 +422,11 @@ class TestAPI(unittest.TestCase): with closing(requests.get(_url(const.URL_API_STREAM), stream=True, headers=HA_HEADERS)) as req: - self.assertEqual(listen_count + 1, self._listen_count()) data = self._stream_next_event(req) - self.assertEqual('ping', data) + self.assertEqual(listen_count + 1, self._listen_count()) + hass.bus.fire('test_event') hass.pool.block_till_done() @@ -441,12 +441,11 @@ class TestAPI(unittest.TestCase): 'restrict': 'test_event1,test_event3'}), stream=True, headers=HA_HEADERS)) as req: - self.assertEqual(listen_count + 2, self._listen_count()) - data = self._stream_next_event(req) - self.assertEqual('ping', data) + self.assertEqual(listen_count + 2, self._listen_count()) + hass.bus.fire('test_event1') hass.pool.block_till_done() hass.bus.fire('test_event2')