def test_parses_event_json(self): self.parser = parsers.EventStreamJSONParser() headers = {':event-type': 'EventD'} body = (b'{' b' "StringField": "abcde",' b' "IntField": 1234' b'}') parsed = self.parse_event(headers, body) expected = {'EventD': {'StringField': 'abcde', 'IntField': 1234}} self.assertEqual(parsed, expected)
def test_parses_exception_event(self): self.parser = parsers.EventStreamJSONParser() error_code = 'EventException' headers = { ':message-type': 'exception', ':exception-type': error_code, } body = b'{"message": "You did something wrong"}' parsed = self.parse_event(headers, body, status_code=400) expected = { 'Error': { 'Code': error_code, 'Message': 'You did something wrong' } } self.assertEqual(parsed, expected)