test: update integ test assertions for user agents

aws · Oct 9, 2024 · 306327a · 306327a
1 parent 4ffcd4b
commit 306327a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/packages/middleware-flexible-checksums/src/middleware-flexible-checksums.integ.spec.ts b/packages/middleware-flexible-checksums/src/middleware-flexible-checksums.integ.spec.ts
@@ -137,7 +137,7 @@ describe("middleware-flexible-checksums", () => {
 
           requireRequestsFrom(client).toMatch({
             headers: {
-              "user-agent": new RegExp(`(.*?) m\/${id}$`),
+              "user-agent": new RegExp(`(.*?) m\/${id},E$`),
             },
           });
 

diff --git a/packages/middleware-sdk-s3/src/s3-express/middleware-s3-express.integ.spec.ts b/packages/middleware-sdk-s3/src/s3-express/middleware-s3-express.integ.spec.ts
@@ -93,7 +93,7 @@ describe("middleware-s3-express", () => {
 
       requireRequestsFrom(client).toMatch({
         headers: {
-          "user-agent": /(.*?) m\/J$/,
+          "user-agent": /(.*?) m\/J,E$/,
         },
       });
 

diff --git a/packages/middleware-user-agent/src/check-features.ts b/packages/middleware-user-agent/src/check-features.ts
@@ -38,10 +38,9 @@ export async function checkFeatures(
   config: PreviouslyResolved,
   args: BuildHandlerArguments<any>
 ): Promise<void> {
-  // eslint-disable-next-line
   const request = args.request as IHttpRequest;
 
-  if (request.headers?.["smithy-protocol"] === "rpc-v2-cbor") {
+  if (request?.headers?.["smithy-protocol"] === "rpc-v2-cbor") {
     setFeature(context, "PROTOCOL_RPC_V2_CBOR", "M");
   }
 

diff --git a/packages/middleware-user-agent/src/middleware-user-agent.integ.spec.ts b/packages/middleware-user-agent/src/middleware-user-agent.integ.spec.ts
@@ -36,7 +36,7 @@ describe("middleware-user-agent", () => {
 
       requireRequestsFrom(doc).toMatch({
         headers: {
-          "user-agent": /(.*?) m\/d,O,R$/,
+          "user-agent": /(.*?) m\/d,E,O,R$/,
         },
       });