| 61 | |
| 62 | class TestEndpointBase(unittest.TestCase): |
| 63 | def setUp(self): |
| 64 | self.op = mock.Mock() |
| 65 | self.op.has_streaming_output = False |
| 66 | self.op.has_event_stream_output = False |
| 67 | self.op.metadata = {'protocol': 'json'} |
| 68 | self.event_emitter = mock.Mock() |
| 69 | self.event_emitter.emit.return_value = [] |
| 70 | self.factory_patch = mock.patch( |
| 71 | 'botocore.parsers.ResponseParserFactory' |
| 72 | ) |
| 73 | self.factory = self.factory_patch.start() |
| 74 | self.endpoint = Endpoint( |
| 75 | 'https://ec2.us-west-2.amazonaws.com/', |
| 76 | endpoint_prefix='ec2', |
| 77 | event_emitter=self.event_emitter, |
| 78 | ) |
| 79 | self.http_session = mock.Mock() |
| 80 | self.http_session.send.return_value = mock.Mock( |
| 81 | status_code=200, |
| 82 | headers={}, |
| 83 | content=b'{"Foo": "bar"}', |
| 84 | ) |
| 85 | self.endpoint.http_session = self.http_session |
| 86 | |
| 87 | def tearDown(self): |
| 88 | self.factory_patch.stop() |