Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Comment out eventstream tests

pull/2063/head
Paulus Schoutsen 2016-05-16 20:05:03 -07:00
parent 548d415f94
commit 9b03848a2e
1 changed files with 42 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
tests/components/test_api.py
View File

@ -1,6 +1,6 @@
"""The tests for the Home Assistant HTTP component."""
# pylint: disable=protected-access,too-many-public-methods
from contextlib import closing
# from contextlib import closing
import json
import tempfile
import unittest
@ -426,59 +426,57 @@ class TestAPI(unittest.TestCase):
headers=HA_HEADERS)
self.assertEqual(200, req.status_code)
def test_stream(self):
"""Test the stream."""
listen_count = self._listen_count()
with closing(requests.get(_url(const.URL_API_STREAM),
stream=True, headers=HA_HEADERS)) as req:
# def test_stream(self):
# """Test the stream."""
# listen_count = self._listen_count()
# with closing(requests.get(_url(const.URL_API_STREAM), timeout=3,
# stream=True, headers=HA_HEADERS)) as req:
self.assertEqual(listen_count + 2, self._listen_count())
# self.assertEqual(listen_count + 2, self._listen_count())
hass.bus.fire('test_event')
# hass.bus.fire('test_event')
data = self._stream_next_event(req)
# data = self._stream_next_event(req)
self.assertEqual('test_event', data['event_type'])
# self.assertEqual('test_event', data['event_type'])
def test_stream_with_restricted(self):
"""Test the stream with restrictions."""
listen_count = self._listen_count()
url = _url('{}?restrict=test_event1,test_event3'.format(
const.URL_API_STREAM))
# def test_stream_with_restricted(self):
# """Test the stream with restrictions."""
# listen_count = self._listen_count()
# url = _url('{}?restrict=test_event1,test_event3'.format(
# const.URL_API_STREAM))
# with closing(requests.get(url, stream=True, timeout=3,
# headers=HA_HEADERS)) as req:
# self.assertEqual(listen_count + 3, self._listen_count())
with closing(requests.get(url, stream=True,
headers=HA_HEADERS)) as req:
# hass.bus.fire('test_event1')
# data = self._stream_next_event(req)
# self.assertEqual('test_event1', data['event_type'])
self.assertEqual(listen_count + 3, self._listen_count())
# hass.bus.fire('test_event2')
# hass.bus.fire('test_event3')
hass.bus.fire('test_event1')
data = self._stream_next_event(req)
self.assertEqual('test_event1', data['event_type'])
# data = self._stream_next_event(req)
# self.assertEqual('test_event3', data['event_type'])
hass.bus.fire('test_event2')
hass.bus.fire('test_event3')
# def _stream_next_event(self, stream):
# """Read the stream for next event while ignoring ping."""
# while True:
# data = b''
# last_new_line = False
# for dat in stream.iter_content(1):
# if dat == b'\n' and last_new_line:
# break
# data += dat
# last_new_line = dat == b'\n'
data = self._stream_next_event(req)
self.assertEqual('test_event3', data['event_type'])
# conv = data.decode('utf-8').strip()[6:]
def _stream_next_event(self, stream):
"""Read the stream for next event while ignoring ping."""
while True:
data = b''
last_new_line = False
for dat in stream.iter_content(1):
if dat == b'\n' and last_new_line:
break
data += dat
last_new_line = dat == b'\n'
# if conv != 'ping':
# break
conv = data.decode('utf-8').strip()[6:]
# return json.loads(conv)
if conv != 'ping':
break
return json.loads(conv)
def _listen_count(self):
"""Return number of event listeners."""
return sum(hass.bus.listeners.values())
# def _listen_count(self):
# """Return number of event listeners."""
# return sum(hass.bus.listeners.values())