Use a callback to handle dynamic getHttpHeader calls in testing

ColdBox · Dec 10, 2024 · ccee9bd · ccee9bd
1 parent 68f723a
commit ccee9bd
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/system/testing/BaseTestCase.cfc b/system/testing/BaseTestCase.cfc
@@ -604,14 +604,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 );
 	}