def test_event_stream_initial_response_no_event(): raw_stream = create_mock_raw_stream(b'') parser = mock.Mock(spec=EventStreamXMLParser) output_shape = mock.Mock() event_stream = EventStream(raw_stream, output_shape, parser, '') with pytest.raises(NoInitialResponseError): event_stream.get_initial_response()
def test_event_stream_initial_response_wrong_type(): raw_stream = create_mock_raw_stream( b"\x00\x00\x00+\x00\x00\x00\x0e4\x8b\xec{\x08event-id\x04\x00", b"\x00\xa0\x0c{'foo':'bar'}\xd3\x89\x02\x85", ) parser = Mock(spec=EventStreamXMLParser) output_shape = Mock() event_stream = EventStream(raw_stream, output_shape, parser, '') event_stream.get_initial_response()
def test_event_stream_initial_response(): raw_stream = create_mock_raw_stream( b'\x00\x00\x00~\x00\x00\x00O\xc5\xa3\xdd\xc6\r:message-type\x07\x00', b'\x05event\x0b:event-type\x07\x00\x10initial-response\r:content-type', b'\x07\x00\ttext/json{"InitialResponse": "sometext"}\xf6\x98$\x83') parser = Mock(spec=EventStreamXMLParser) output_shape = Mock() event_stream = EventStream(raw_stream, output_shape, parser, '') event = event_stream.get_initial_response() headers = { ':message-type': 'event', ':event-type': 'initial-response', ':content-type': 'text/json', } payload = b'{"InitialResponse": "sometext"}' assert event.headers == headers assert event.payload == payload
def test_event_stream_initial_response(): raw_stream = create_mock_raw_stream( b'\x00\x00\x00~\x00\x00\x00O\xc5\xa3\xdd\xc6\r:message-type\x07\x00', b'\x05event\x0b:event-type\x07\x00\x10initial-response\r:content-type', b'\x07\x00\ttext/json{"InitialResponse": "sometext"}\xf6\x98$\x83' ) parser = Mock(spec=EventStreamXMLParser) output_shape = Mock() event_stream = EventStream(raw_stream, output_shape, parser, '') event = event_stream.get_initial_response() headers = { ':message-type': 'event', ':event-type': 'initial-response', ':content-type': 'text/json', } payload = b'{"InitialResponse": "sometext"}' assert event.headers == headers assert event.payload == payload
def test_event_stream_initial_response_no_event(): raw_stream = create_mock_raw_stream(b'') parser = Mock(spec=EventStreamXMLParser) output_shape = Mock() event_stream = EventStream(raw_stream, output_shape, parser, '') event_stream.get_initial_response()