Scripts can set response status code

Author: Bernt Røskar Brenna

UnitTests/TestDynamicResponse.cs

@@ -145,20 +145,31 @@ public async Task ScriptsCanHandleEndpointObjects()
         {
             var endpoint = new Endpoint("foo", "bar");
 
-            var obj = (Dictionary <string, string>) endpoint.GetScriptObject("obj", () => new Dictionary<string, string>());
+            endpoint.SetScriptObject("obj", new Dictionary<string, string>());
+            var obj = (Dictionary <string, string>) endpoint.GetScriptObject("obj");
 
             obj["a"] = "b";
             Assert.Equal(
                 "b", 
-                await EvalAsync("using System.Collections.Generic; return ((Dictionary<string, string>)GetScriptObject(\"obj\", null))[\"a\"];", new RequestInfo { Endpoint = endpoint })
+                await EvalAsync("using System.Collections.Generic; return ((Dictionary<string, string>)Endpoint.GetScriptObject(\"obj\"))[\"a\"];", new RequestInfo { Endpoint = endpoint })
             );
 
             obj["a"] = "c";
             Assert.Equal(
                 "c",
-                await EvalAsync("using System.Collections.Generic; return ((Dictionary<string, string>)GetScriptObject(\"obj\", null))[\"a\"];", new RequestInfo { Endpoint = endpoint })
+                await EvalAsync("using System.Collections.Generic; return ((Dictionary<string, string>)Endpoint.GetScriptObject(\"obj\"))[\"a\"];", new RequestInfo { Endpoint = endpoint })
             );
+        }
+
+        [Fact]
+        public async Task ScriptsCanSetResponseStatusCode()
+        {
+            var endpoint = new Endpoint("foo", "bar");
+            var responseCreator = new LiteralDynamicResponseCreator("StatusCode = 102; return \"\";", endpoint);
+            var response = new TestableHttpResponse();
+            await responseCreator.CreateResponseAsync(new TestableHttpRequest("/", null), new byte[0], response, endpoint);
 
+            Assert.Equal(102, (int) response.HttpStatusCode);
         }
     }
 

netmockery/DynamicResponseCreator.cs

@@ -7,6 +7,7 @@
 using System.Diagnostics;
 using System.IO;
 using System.Linq;
+using System.Net;
 using System.Reflection;
 #if !NET462
 using System.Runtime.Loader;
@@ -133,6 +134,17 @@ static public string ExecuteIncludes(string sourceCode, string directory)
         }
 
         public abstract string SourceCode { get; }
+
+        protected override void AfterResponseWritten(RequestInfo requestInfo, IHttpResponseWrapper response)
+        {
+            Debug.Assert(requestInfo != null);
+            Debug.Assert(response != null);
+
+            if (requestInfo.StatusCode != RequestInfo.USE_CONFIGURED_STATUS_CODE)
+            {
+                response.HttpStatusCode = (HttpStatusCode) requestInfo.StatusCode;
+            }
+        }
     }
 
     public class LiteralDynamicResponseCreator : DynamicResponseCreatorBase

netmockery/ResponseCreator.cs

@@ -86,13 +86,16 @@ public interface IResponseCreatorWithFilename
 
     public class RequestInfo
     {
+        public const int USE_CONFIGURED_STATUS_CODE = -1;
+
         private static object _locker = new object();
 
         private DateTime _now = DateTime.MinValue;
 
         public string RequestPath;
         public string QueryString;
         public string RequestBody;
+        public int StatusCode = USE_CONFIGURED_STATUS_CODE;
         public IHeaderDictionary Headers;
         public Endpoint Endpoint;
         public string EndpointDirectory => Endpoint.Directory;
@@ -201,14 +204,15 @@ public void SetStatusCodeFromString(string value)
 
         public override async Task<byte[]> CreateResponseAsync(IHttpRequestWrapper request, byte[] requestBody, IHttpResponseWrapper response, Endpoint endpoint)
         {
-            var responseBody = await GetBodyAndExecuteReplacementsAsync(new RequestInfo
+            var requestInfo = new RequestInfo
             {
                 RequestPath = request.Path.ToString(),
                 QueryString = request.QueryString.ToString(),
-                RequestBody = Encoding.UTF8.GetString(requestBody),                
+                RequestBody = Encoding.UTF8.GetString(requestBody),
                 Headers = request.Headers,
                 Endpoint = endpoint
-            });
+            };
+            var responseBody = await GetBodyAndExecuteReplacementsAsync(requestInfo);
             if (ContentType != null)
             {
                 var contenttype = ContentType;
@@ -217,8 +221,18 @@ public override async Task<byte[]> CreateResponseAsync(IHttpRequestWrapper reque
             }
             response.HttpStatusCode = (HttpStatusCode) StatusCode;
             await response.WriteAsync(responseBody, Encoding);
+
+            AfterResponseWritten(requestInfo, response);
+
             return Encoding.GetBytes(responseBody);
         }
+
+        protected virtual void AfterResponseWritten(RequestInfo requestInfo, IHttpResponseWrapper response)
+        {
+            // extension point, override to add logic in inheritors.
+            // currently used by DynamicResponseCreator in order to let script code override status code
+        }
+
         public string ContentType {
             get { return ReplaceParameterReference(_contentType); }
             set { _contentType = value; }