Scripts: SetParam, GetScriptObject

Author: Bernt Røskar Brenna

UnitTests/TestDynamicResponse.cs

@@ -126,6 +126,40 @@ public async Task UsingSystemDiagnosticsDebug()
         {
             Assert.Equal("OK", await EvalAsync("using System.Diagnostics; Debug.Assert(true); return \"OK\";"));
         }
+
+        [Fact]
+        public async Task ScriptsCanSetParams()
+        {
+            var endpoint = new Endpoint("foo", "bar");
+            endpoint.AddParameter(new EndpointParameter { Name = "param", Value = "abc" });
+
+            Assert.Equal("abc", await EvalAsync("return GetParam(\"param\");", new RequestInfo { Endpoint = endpoint }));
+
+            await EvalAsync("SetParam(\"param\", \"def\"); return \"\";", new RequestInfo { Endpoint = endpoint });
+
+            Assert.Equal("def", await EvalAsync("return GetParam(\"param\");", new RequestInfo { Endpoint = endpoint }));
+        }
+
+        [Fact]
+        public async Task ScriptsCanHandleEndpointObjects()
+        {
+            var endpoint = new Endpoint("foo", "bar");
+
+            var obj = (Dictionary <string, string>) endpoint.GetScriptObject("obj", () => new Dictionary<string, string>());
+
+            obj["a"] = "b";
+            Assert.Equal(
+                "b", 
+                await EvalAsync("using System.Collections.Generic; return ((Dictionary<string, string>)GetScriptObject(\"obj\", null))[\"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 })
+            );
+
+        }
     }
 
     public class TestCheckScriptModifications : IDisposable

netmockery/Endpoint.cs

@@ -22,6 +22,7 @@ public class Endpoint
         private string _pathregex;
         private List<Tuple<RequestMatcher, ResponseCreator>> _responses = new List<Tuple<RequestMatcher, ResponseCreator>>();
         private List<EndpointParameter> _parameters = new List<EndpointParameter>();
+        private Dictionary<string, object> _scriptObjects = new Dictionary<string, object>();
         private bool _ruleThatCatchesEveryThingHasBeenAdded = false;        
 
         public Endpoint(string name, string pathregex)
@@ -95,6 +96,16 @@ public EndpointParameter GetParameter(int parameterIndex)
 
         public int ParameterCount => _parameters.Count;
 
+        public object GetScriptObject(string name, Func<object> objectCreator)
+        {
+            if (!_scriptObjects.ContainsKey(name))
+            {
+                _scriptObjects[name] = objectCreator();
+            }
+            return _scriptObjects[name];
+
+        }
+
         public void Add(RequestMatcher requestMatcher, ResponseCreator responseCreator)
         {
             Debug.Assert(requestMatcher != null);

netmockery/ResponseCreator.cs

@@ -86,6 +86,8 @@ public interface IResponseCreatorWithFilename
 
     public class RequestInfo
     {
+        private static object _locker = new object();
+
         private DateTime _now = DateTime.MinValue;
 
         public string RequestPath;
@@ -101,6 +103,16 @@ public string GetParam(string name)
             return Endpoint.GetParameter(name).Value;
         }
 
+        public void SetParam(string name, string value)
+        {
+            lock (_locker)
+            {
+                Endpoint.GetParameter(name).Value = value;
+            }            
+        }
+
+        public object GetScriptObject(string name, Func<object> objectCreator) => Endpoint.GetScriptObject(name, objectCreator);
+
         public void SetStaticNow(DateTime now)
         {
             _now = now;