def _query_stream(self, response_type, result, keepAliveToken=DEFAULT_KEEPALIVE_TOKEN): client = Client(retries=0, config={'address': "URI", "stream": True, "response": response_type, "keepAliveToken": keepAliveToken}) client._make_request = MagicMock(return_value=result) return client.query()
def _query_stream(self, response_type, result, processor): client = Client(retries=0, config={'address': "URI", "stream": True, "response": response_type, "processor": processor}) client._make_request = MagicMock(return_value=result) return client.query()