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_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()
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()
