test: fix lex runtime and mediastore payload tests

Author: kuhe

clients/client-lex-runtime-service/test/LexRuntimeService.spec.ts

@@ -1,36 +1,32 @@
-import { HttpRequest } from "@smithy/protocol-http";
-import { SerializeMiddleware } from "@smithy/types";
-import { describe, expect, test as it } from "vitest";
+import { requireRequestsFrom } from "@aws-sdk/aws-util-test/src";
+import { Readable } from "node:stream";
+import { describe, test as it } from "vitest";
 
 import { LexRuntimeService } from "../src/LexRuntimeService";
 
 describe("@aws-sdk/client-lex-runtime-service", () => {
   describe("PostContent", () => {
     it("should contain correct x-amz-content-sha256 header", async () => {
-      const validator: SerializeMiddleware<any, any> = (next) => (args) => {
-        // middleware intercept the request and return it early
-        const request = args.request as HttpRequest;
-        expect(request.headers).to.have.property("x-amz-content-sha256", "UNSIGNED-PAYLOAD");
-        return Promise.resolve({ output: {} as any, response: {} as any });
-      };
       const client = new LexRuntimeService({
         region: "us-west-2",
         credentials: {
           accessKeyId: "CLIENT_TEST",
           secretAccessKey: "CLIENT_TEST",
         },
       });
-      client.middlewareStack.add(validator, {
-        step: "serialize",
-        name: "endpointValidator",
-        priority: "low",
+
+      requireRequestsFrom(client).toMatch({
+        headers: {
+          "x-amz-content-sha256": "UNSIGNED-PAYLOAD",
+        },
       });
+
       return await client.postContent({
         botAlias: "alias",
         botName: "bot",
         userId: "user",
         contentType: "text/plain; charset=utf-8",
-        inputStream: "hello world!",
+        inputStream: Readable.from(new Uint8Array([1, 2, 3, 4, 5, 6, 7, 8])),
       });
     });
   });

clients/client-mediastore-data/test/MediaStoreData.spec.ts

@@ -1,33 +1,29 @@
-import { HttpRequest } from "@smithy/protocol-http";
-import { SerializeMiddleware } from "@smithy/types";
-import { describe, expect, test as it } from "vitest";
+import { requireRequestsFrom } from "@aws-sdk/aws-util-test/src";
+import { Readable } from "node:stream";
+import { describe, test as it } from "vitest";
 
 import { MediaStoreData } from "../src/MediaStoreData";
 
 describe("@aws-sdk/client-mediastore-data", () => {
   describe("PutObject", () => {
     it("should contain correct x-amz-content-sha256 header", async () => {
-      const validator: SerializeMiddleware<any, any> = (next) => (args) => {
-        // middleware intercept the request and return it early
-        const request = args.request as HttpRequest;
-        expect(request.headers).to.have.property("x-amz-content-sha256", "UNSIGNED-PAYLOAD");
-        return Promise.resolve({ output: {} as any, response: {} as any });
-      };
       const client = new MediaStoreData({
         region: "us-west-2",
         credentials: {
           accessKeyId: "CLIENT_TEST",
           secretAccessKey: "CLIENT_TEST",
         },
       });
-      client.middlewareStack.add(validator, {
-        step: "serialize",
-        name: "endpointValidator",
-        priority: "low",
+
+      requireRequestsFrom(client).toMatch({
+        headers: {
+          "x-amz-content-sha256": "UNSIGNED-PAYLOAD",
+        },
       });
+
       return await client.putObject({
         Path: "foo.avi",
-        Body: "binary body",
+        Body: Readable.from(new Uint8Array([1, 2, 3, 4, 5, 6, 7, 8])),
       });
     });
   });