.

Author: mredolatti

splitio/push/sse.py

@@ -80,7 +80,7 @@ def build(self):
 class SSEClient(object):
     """SSE Client implementation."""
 
-    _DEFAULT_HEADERS = {'Accept': 'text/event-stream'}
+    _DEFAULT_HEADERS = {'accept': 'text/event-stream'}
     _EVENT_SEPARATORS = set([b'\n', b'\r\n'])
 
     def __init__(self, callback):

tests/helpers/mockserver.py

@@ -71,7 +71,8 @@ def do_GET(self):  #pylint:disable=invalid-name
         self.end_headers()
 
         if self._req_queue is not None:
-            self._req_queue.put(self.path)
+            headers = dict(zip(self.headers.keys(), self.headers.values()))
+            self._req_queue.put(Request('GET', self.path, headers, None))
 
         def write_chunk(chunk):
             """Write an event/chunk."""

tests/push/test_splitsse.py

@@ -7,8 +7,7 @@
 from splitio.models.token import Token
 from splitio.push.splitsse import SplitSSEClient
 from splitio.push.sse import SSEEvent
-
-from .mockserver import SSEMockServer
+from tests.helpers.mockserver import SSEMockServer
 
 
 class SSEClientTests(object):
@@ -41,7 +40,9 @@ def handler(event):
         time.sleep(1)
         client.stop()
 
-        assert request_queue.get() == '/event-stream?v=1.1&accessToken=some&channels=chan1,[?occupancy=metrics.publishers]chan2'
+        request = request_queue.get(1)
+        assert request.path == '/event-stream?v=1.1&accessToken=some&channels=chan1,[?occupancy=metrics.publishers]chan2'
+        assert request.headers['accept'] == 'text/event-stream'
 
         assert events == [
             SSEEvent('1', 'message', '1', 'a'),
@@ -74,8 +75,9 @@ def handler(event):
         with pytest.raises(Exception):
             client.stop()
 
-        assert request_queue.get() == ('/event-stream?v=1.1&accessToken=some'
-                                       '&channels=chan1,[?occupancy=metrics.publishers]chan2')
+        request = request_queue.get(1)
+        assert request.path == '/event-stream?v=1.1&accessToken=some&channels=chan1,[?occupancy=metrics.publishers]chan2'
+        assert request.headers['accept'] == 'text/event-stream'
 
         server.publish(SSEMockServer.VIOLENT_REQUEST_END)
         server.stop()

tests/push/test_sse.py

@@ -4,7 +4,7 @@
 import threading
 import pytest
 from splitio.push.sse import SSEClient, SSEEvent
-from .mockserver import SSEMockServer
+from tests.helpers.mockserver import SSEMockServer
 
 
 class SSEClientTests(object):