Use a callback to handle dynamic getHttpHeader calls in testing (#595)

ColdBox · Jan 20, 2025 · 47a1cd0 · 47a1cd0
1 parent 17cb642
commit 47a1cd0
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/system/testing/BaseTestCase.cfc b/system/testing/BaseTestCase.cfc
@@ -601,14 +601,18 @@ component extends="testbox.system.compat.framework.TestCase" accessors="true" {
 			.each( function( name ){
 				mockedEvent.setValue( arguments.name, params[ arguments.name ] );
 			} );
-		arguments.headers
-			.keyArray()
-			.each( function( name ){
-				mockedEvent
-					.$( "getHTTPHeader" )
-					.$args( arguments.name )
-					.$results( headers[ arguments.name ] );
-			} );
+		mockedEvent
+			.$(
+				method="getHTTPHeader",
+				callback=function( name, defaultValue ){
+					if( headers.keyExists( arguments.name ) ){
+						return headers[ arguments.name ];
+					}
+					if( !isNull( arguments.defaultValue ) ){
+						return arguments.defaultValue;
+					}
+				}
+			);
 		return this.execute( argumentCollection = arguments );
 	}