From 85d1968ce5f26f87cace5f67df0c1b8a07398148 Mon Sep 17 00:00:00 2001
From: delvedor <tommydelved@gmail.com>
Date: Thu, 16 Dec 2021 17:56:01 +0100
Subject: [PATCH] Updated transport dependency

---
 package.json                 |  2 +-
 test/utils/MockConnection.ts | 24 ++++++++++++++++++------
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/package.json b/package.json
index 6a15cc1a7..8e74947ae 100644
--- a/package.json
+++ b/package.json
@@ -80,7 +80,7 @@
     "xmlbuilder2": "^3.0.2"
   },
   "dependencies": {
-    "@elastic/transport": "^8.0.0-beta.2",
+    "@elastic/transport": "^8.1.0-beta.1",
     "tslib": "^2.3.0"
   },
   "tap": {
diff --git a/test/utils/MockConnection.ts b/test/utils/MockConnection.ts
index 3bb508fed..19af3dd54 100644
--- a/test/utils/MockConnection.ts
+++ b/test/utils/MockConnection.ts
@@ -24,7 +24,9 @@ import {
   ConnectionRequestParams,
   ConnectionRequestOptions,
   ConnectionRequestResponse,
-  errors
+  errors,
+  ConnectionRequestOptionsAsStream,
+  ConnectionRequestResponseAsStream
 } from '@elastic/transport'
 const {
   ConnectionError,
@@ -32,7 +34,9 @@ const {
 } = errors
 
 export class MockConnection extends BaseConnection {
-  request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse> {
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse>
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptionsAsStream): Promise<ConnectionRequestResponseAsStream>
+  async request (params: ConnectionRequestParams, options: any): Promise<any> {
     return new Promise((resolve, reject) => {
       const body = JSON.stringify({ hello: 'world' })
       const statusCode = setStatusCode(params.path)
@@ -49,7 +53,9 @@ export class MockConnection extends BaseConnection {
 }
 
 export class MockConnectionTimeout extends BaseConnection {
-  request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse> {
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse>
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptionsAsStream): Promise<ConnectionRequestResponseAsStream>
+  async request (params: ConnectionRequestParams, options: any): Promise<any> {
     return new Promise((resolve, reject) => {
       process.nextTick(reject, new TimeoutError('Request timed out'))
     })
@@ -57,7 +63,9 @@ export class MockConnectionTimeout extends BaseConnection {
 }
 
 export class MockConnectionError extends BaseConnection {
-  request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse> {
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse>
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptionsAsStream): Promise<ConnectionRequestResponseAsStream>
+  async request (params: ConnectionRequestParams, options: any): Promise<any> {
     return new Promise((resolve, reject) => {
       process.nextTick(reject, new ConnectionError('kaboom'))
     })
@@ -65,7 +73,9 @@ export class MockConnectionError extends BaseConnection {
 }
 
 export class MockConnectionSniff extends BaseConnection {
-  request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse> {
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse>
+  async request (params: ConnectionRequestParams, options: ConnectionRequestOptionsAsStream): Promise<ConnectionRequestResponseAsStream>
+  async request (params: ConnectionRequestParams, options: any): Promise<any> {
     return new Promise((resolve, reject) => {
       const sniffResult = {
         nodes: {
@@ -106,7 +116,9 @@ export function buildMockConnection (opts: onRequestMock) {
   assert(opts.onRequest, 'Missing required onRequest option')
 
   class MockConnection extends BaseConnection {
-    request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse> {
+    async request (params: ConnectionRequestParams, options: ConnectionRequestOptions): Promise<ConnectionRequestResponse>
+    async request (params: ConnectionRequestParams, options: ConnectionRequestOptionsAsStream): Promise<ConnectionRequestResponseAsStream>
+    async request (params: ConnectionRequestParams, options: any): Promise<any> {
       return new Promise((resolve, reject) => {
         params.headers = { ...this.headers, ...params.headers }
         let { body, statusCode, headers } = opts.onRequest(params)