chore: fix type errors in integ tests (#6054)

aws · May 2, 2024 · 94d4a10 · 94d4a10
1 parent 003de3f
commit 94d4a10
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 8 deletions.
diff --git a/packages/credential-provider-node/src/credential-provider-node.integ.spec.ts b/packages/credential-provider-node/src/credential-provider-node.integ.spec.ts
@@ -68,14 +68,16 @@ jest.mock("@aws-sdk/client-sso", () => {
   };
 });
 
-let stsSpy: jest.Spied<any> | any | undefined = undefined;
+// This var must be hoisted.
+// eslint-disable-next-line no-var
+var stsSpy: jest.Spied<any> | any | undefined = undefined;
 
 jest.mock("@aws-sdk/client-sts", () => {
   const actual = jest.requireActual("@aws-sdk/client-sts");
 
   const originalSend = actual.STSClient.prototype.send;
 
-  stsSpy = jest.spyOn(actual.STSClient.prototype, "send").mockImplementation(async function (command: any) {
+  stsSpy = jest.spyOn(actual.STSClient.prototype, "send").mockImplementation(async function (this: any, command: any) {
     if (command.constructor.name === "AssumeRoleCommand") {
       return {
         Credentials: {

diff --git a/packages/middleware-eventstream/src/middleware-eventstream.integ.spec.ts b/packages/middleware-eventstream/src/middleware-eventstream.integ.spec.ts
@@ -26,7 +26,7 @@ describe("middleware-eventstream", () => {
           [Symbol.asyncIterator]() {
             return {
               next() {
-                return this;
+                return this as any;
               },
             };
           },
@@ -57,7 +57,7 @@ describe("middleware-eventstream", () => {
           [Symbol.asyncIterator]() {
             return {
               next() {
-                return this;
+                return this as any;
               },
             };
           },
@@ -87,7 +87,7 @@ describe("middleware-eventstream", () => {
           [Symbol.asyncIterator]() {
             return {
               next() {
-                return this;
+                return this as any;
               },
             };
           },

diff --git a/packages/middleware-sdk-sqs/src/middleware-sdk-sqs.integ.spec.ts b/packages/middleware-sdk-sqs/src/middleware-sdk-sqs.integ.spec.ts
@@ -4,8 +4,9 @@ import type { AwsCredentialIdentity } from "@smithy/types";
 import crypto from "crypto";
 import { Readable } from "stream";
 
-import sqsModel from "../../../codegen/sdk-codegen/aws-models/sqs.json";
 import { requireRequestsFrom } from "../../../private/aws-util-test/src";
+
+const sqsModel: any = require("../../../codegen/sdk-codegen/aws-models/sqs.json");
 const useAwsQuery = !!sqsModel.shapes["com.amazonaws.sqs#AmazonSQS"].traits["aws.protocols#awsQuery"];
 
 let hashError = "";

diff --git a/...middleware-sdk-transcribe-streaming/src/middleware-sdk-transcribe-streaming.integ.spec.ts b/...middleware-sdk-transcribe-streaming/src/middleware-sdk-transcribe-streaming.integ.spec.ts
@@ -42,7 +42,7 @@ describe("middleware-sdk-transcribe-streaming", () => {
         MediaEncoding: "pcm",
         AudioStream: {
           [Symbol.asyncIterator]() {
-            return this;
+            return this as any;
           },
         },
       });

diff --git a/packages/middleware-websocket/src/middleware-websocket.integ.spec.ts b/packages/middleware-websocket/src/middleware-websocket.integ.spec.ts
@@ -41,7 +41,7 @@ describe("middleware-websocket", () => {
         ChallengeVersions: "a,b,c",
         LivenessRequestStream: {
           [Symbol.asyncIterator]() {
-            return this;
+            return this as any;
           },
         },
       });