Fix tests

humanmade · Mar 5, 2024 · 78e568d · 78e568d
1 parent 25ff598
commit 78e568d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test-lambda.ts b/tests/test-lambda.ts
@@ -32,6 +32,7 @@ class TestResponseStream {
 	contentType: string | undefined;
 	body: string | Buffer | undefined;
 	headers: { [key: string]: string } = {};
+	metadata: any;
 
 	setContentType( type: string ): void {
 		this.contentType = type;
@@ -46,7 +47,9 @@ class TestResponseStream {
 		}
 	}
 	end(): void {
-
+		if ( this.metadata.headers['Content-Type'] ) {
+			this.contentType = this.metadata.headers['Content-Type'];
+		}
 	}
 }
 
@@ -63,7 +66,8 @@ global.awslambda = {
 		 * @param stream The response stream.
 		 * @param metadata The metadata for the response.
 		 */
-		from( stream: ResponseStream, metadata ) : ResponseStream {
+		from( stream: TestResponseStream, metadata ) : TestResponseStream {
+			stream.metadata = metadata;
 			return stream;
 		},
 	},