From 0253666012238edd740feb0e5e9c59d087cf4877 Mon Sep 17 00:00:00 2001
From: Renato Lima <natenho@gmail.com>
Date: Sun, 10 Mar 2024 18:33:57 -0300
Subject: [PATCH] Fix mocks being returned even when the condition script
 throws an error

---
 Mockaco.sln                                   |  2 +-
 .../Request/RequestConditionMatcher.cs        | 17 +++---
 .../Templating/TemplateTransformer.cs         | 32 +++--------
 .../Request/RequestConditionMatcherTest.cs    | 54 +++++++++++++++++++
 .../Templating/TemplateTransformerTest.cs     | 18 +++----
 5 files changed, 82 insertions(+), 41 deletions(-)
 create mode 100644 test/Mockaco.AspNetCore.Tests/Templating/Request/RequestConditionMatcherTest.cs

diff --git a/Mockaco.sln b/Mockaco.sln
index 932cbd4..548d28c 100644
--- a/Mockaco.sln
+++ b/Mockaco.sln
@@ -10,7 +10,7 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "Solution Items", "Solution
 		README.md = README.md
 	EndProjectSection
 EndProject
-Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Mockaco.Tests", "test\Mockaco.AspNetCore.Tests\Mockaco.AspNetCore.Tests.csproj", "{EE57B1B4-29D2-4AE3-8F23-5E622302C30F}"
+Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Mockaco.AspNetCore.Tests", "test\Mockaco.AspNetCore.Tests\Mockaco.AspNetCore.Tests.csproj", "{EE57B1B4-29D2-4AE3-8F23-5E622302C30F}"
 EndProject
 Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Mockaco.AspNetCore", "src\Mockaco.AspNetCore\Mockaco.AspNetCore.csproj", "{7766C592-9887-4162-8B9C-51003ED30335}"
 EndProject
diff --git a/src/Mockaco.AspNetCore/Templating/Request/RequestConditionMatcher.cs b/src/Mockaco.AspNetCore/Templating/Request/RequestConditionMatcher.cs
index 858a674..ec6aefd 100644
--- a/src/Mockaco.AspNetCore/Templating/Request/RequestConditionMatcher.cs
+++ b/src/Mockaco.AspNetCore/Templating/Request/RequestConditionMatcher.cs
@@ -1,8 +1,5 @@
 using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.Logging;
-using System;
-using System.Linq;
-using System.Threading.Tasks;
 
 namespace Mockaco
 {
@@ -43,12 +40,18 @@ public async Task<bool> IsMatch(HttpRequest httpRequest, Mock mock)
             }
 
             await conditionMatcherScriptContext.AttachRouteParameters(httpRequest, mock);
+            try
+            {
+                var template = await _templateTransformer.Transform(mock.RawTemplate, conditionMatcherScriptContext);
 
-            var template = await _templateTransformer.Transform(mock.RawTemplate, conditionMatcherScriptContext);
-
-            var isMatch = template.Request?.Condition ?? true;
+                var isMatch = template.Request?.Condition ?? true;
 
-            return isMatch;
+                return isMatch;
+            }
+            catch (Exception)
+            {
+                return false;
+            }
         }
 
         //TODO Remove redundant code
diff --git a/src/Mockaco.AspNetCore/Templating/TemplateTransformer.cs b/src/Mockaco.AspNetCore/Templating/TemplateTransformer.cs
index 941f8e7..529da81 100644
--- a/src/Mockaco.AspNetCore/Templating/TemplateTransformer.cs
+++ b/src/Mockaco.AspNetCore/Templating/TemplateTransformer.cs
@@ -1,9 +1,7 @@
 using Microsoft.Extensions.Logging;
 using Mono.TextTemplating;
 using Newtonsoft.Json;
-using System;
 using System.Text;
-using System.Threading.Tasks;
 
 namespace Mockaco
 {
@@ -28,12 +26,12 @@ public async Task<Template> TransformAndSetVariables(IRawTemplate rawTemplate, I
         }
 
         public async Task<Template> Transform(IRawTemplate rawTemplate, IScriptContext scriptContext)
-        {            
+        {
             scriptContext.Global.DisableWriting();
 
             var transformedTemplate = await Transform(rawTemplate.Content, scriptContext);
 
-            return JsonConvert.DeserializeObject<Template>(transformedTemplate);           
+            return JsonConvert.DeserializeObject<Template>(transformedTemplate);
         }
 
         private async Task<string> Transform(string input, IScriptContext scriptContext)
@@ -57,23 +55,11 @@ private async Task<string> Transform(string input, IScriptContext scriptContext)
                     case State.Directive:
                         break;
                     case State.Expression:
-
-                        object expressionResult;
-
-                        try
-                        {
-                            expressionResult = await Run(tokeniser.Value, scriptContext);
-                        }
-                        catch (Exception)
-                        {
-                            expressionResult = string.Empty;
-                        }
-
+                        var expressionResult = await Run(tokeniser.Value, scriptContext);
                         output.Append(expressionResult);
-
                         break;
                     case State.Block:
-                        await Run(tokeniser.Value, scriptContext);                             
+                        await Run(tokeniser.Value, scriptContext);
                         break;
                     case State.Helper:
                         break;
@@ -95,21 +81,19 @@ private async Task<string> Transform(string input, IScriptContext scriptContext)
 
         private async Task<object> Run(string code, IScriptContext scriptContext)
         {
-            object result = null;
-
             try
             {
-                result = await _scriptRunnerFactory.Invoke<IScriptContext, object>(scriptContext, code);
+                var result = await _scriptRunnerFactory.Invoke<IScriptContext, object>(scriptContext, code);
 
                 _logger.LogDebug("Processed script {code} with result {result}", code, result);
+
+                return result;
             }
             catch (Exception ex)
             {
                 _logger.LogError(ex, "Processed script {code} with error", code);
+                throw;
             }
-
-            return result;
         }
-
     }
 }
\ No newline at end of file
diff --git a/test/Mockaco.AspNetCore.Tests/Templating/Request/RequestConditionMatcherTest.cs b/test/Mockaco.AspNetCore.Tests/Templating/Request/RequestConditionMatcherTest.cs
new file mode 100644
index 0000000..a5bac6f
--- /dev/null
+++ b/test/Mockaco.AspNetCore.Tests/Templating/Request/RequestConditionMatcherTest.cs
@@ -0,0 +1,54 @@
+using FluentAssertions;
+using Microsoft.AspNetCore.Http;
+using Microsoft.Extensions.Logging;
+using Moq;
+using System.Net;
+
+namespace Mockaco.AspNetCore.Tests.Templating.Request
+{
+    public sealed class RequestConditionMatcherTest
+    {
+        private readonly IGlobalVariableStorage _globalVariableStorage;
+        private readonly ITemplateTransformer _templateTransformer;
+        private readonly IFakerFactory _fakerFactory;
+        private readonly ILogger<RequestConditionMatcher> _logger;
+        private readonly IRequestBodyFactory _requestBodyFactory;
+        private readonly IMockacoContext _mockacoContext;
+
+        public RequestConditionMatcherTest()
+        {
+            _templateTransformer = Moq.Mock.Of<ITemplateTransformer>();
+            _fakerFactory = Moq.Mock.Of<IFakerFactory>();
+            _requestBodyFactory = Moq.Mock.Of<IRequestBodyFactory>();
+            _mockacoContext = Moq.Mock.Of<IMockacoContext>();
+            _globalVariableStorage = Moq.Mock.Of<IGlobalVariableStorage>();
+            _logger = Moq.Mock.Of<ILogger<RequestConditionMatcher>>();
+        }
+
+        [Fact]
+        public async Task Condition_Does_Not_Match_On_Script_Error()
+        {
+            Moq.Mock.Get(_templateTransformer)
+                .Setup(n => n.Transform(It.IsAny<IRawTemplate>(), It.IsAny<IScriptContext>()))
+                .ThrowsAsync(new NotImplementedException());
+
+            Moq.Mock.Get(_mockacoContext).Setup(c => c.Errors).Returns(new List<Error>());
+
+            var conditionMatcher = new RequestConditionMatcher(
+                _templateTransformer, _fakerFactory, _requestBodyFactory, _mockacoContext, _globalVariableStorage, _logger);
+
+            var httpRequest = new Mock<HttpRequest>();
+            httpRequest.Setup(h => h.HttpContext)
+                .Returns(Moq.Mock.Of<HttpContext>(c => c.Request == httpRequest.Object));
+
+            var rawTemplate = Moq.Mock.Of<IRawTemplate>();
+            Moq.Mock.Get(rawTemplate).Setup(r => r.Content).Returns(@"");
+
+            var mock = new Mock("GET", "/ping", rawTemplate, true);
+
+            var isMatch = await conditionMatcher.IsMatch(httpRequest.Object, mock);
+
+            isMatch.Should().Be(false);
+        }
+    }
+}
diff --git a/test/Mockaco.AspNetCore.Tests/Templating/TemplateTransformerTest.cs b/test/Mockaco.AspNetCore.Tests/Templating/TemplateTransformerTest.cs
index 2f2d856..118c569 100644
--- a/test/Mockaco.AspNetCore.Tests/Templating/TemplateTransformerTest.cs
+++ b/test/Mockaco.AspNetCore.Tests/Templating/TemplateTransformerTest.cs
@@ -68,16 +68,16 @@ private static void Assert(Template transformedTemplate)
             transformedTemplate.Response.Headers.Should()
                 .HaveCount(2);
 
-            transformedTemplate.Response.Body["id"].ToString().Should()
+            transformedTemplate.Response.Body["id"]?.ToString().Should()
                 .Be("1");
 
-            transformedTemplate.Response.Body["message"].ToString().Should()
+            transformedTemplate.Response.Body["message"]?.ToString().Should()
                 .Be("Hello world");
 
-            transformedTemplate.Response.Body["createdAt"].Type.Should()
+            transformedTemplate.Response.Body["createdAt"]?.Type.Should()
                .Be(JTokenType.Date);
 
-            transformedTemplate.Response.Body["createdAt"].Value<DateTime>().Should()
+            transformedTemplate.Response.Body["createdAt"]?.Value<DateTime>().Should()
                .Be(new DateTime(2012, 04, 23, 18, 25, 43, 511, DateTimeKind.Utc));
 
             transformedTemplate.Callback.Method.Should()
@@ -98,19 +98,19 @@ private static void Assert(Template transformedTemplate)
             transformedTemplate.Callback.Headers.Should()
                 .HaveCount(2);
 
-            transformedTemplate.Callback.Body["key"].ToString().Should()
+            transformedTemplate.Callback.Body["key"]?.ToString().Should()
                .Be("2");
 
-            transformedTemplate.Callback.Body["key"].Type.Should()
+            transformedTemplate.Callback.Body["key"]?.Type.Should()
                .Be(JTokenType.Integer);
 
-            transformedTemplate.Callback.Body["topic"].ToString().Should()
+            transformedTemplate.Callback.Body["topic"]?.ToString().Should()
                 .Be("Hello callback");
 
-            transformedTemplate.Callback.Body["updatedAt"].Type.Should()
+            transformedTemplate.Callback.Body["updatedAt"]?.Type.Should()
                .Be(JTokenType.Date);
 
-            transformedTemplate.Callback.Body["updatedAt"].Value<DateTime>().Should()
+            transformedTemplate.Callback.Body["updatedAt"]?.Value<DateTime>().Should()
                .Be(new DateTime(2003, 02, 01, 19, 00, 00, DateTimeKind.Utc));
         }