Simulated health check (#116)

Co-authored-by: Richard Pringle <[email protected]>
pm7y · Feb 7, 2023 · ce9ea55 · ce9ea55
1 parent 12ae6af
commit ce9ea55
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/src/AzureEventGridSimulator.Tests/IntegrationTests/BasicTests.cs b/src/AzureEventGridSimulator.Tests/IntegrationTests/BasicTests.cs
@@ -51,4 +51,21 @@ public async Task GivenAValidEvent_WhenPublished_ThenItShouldBeAccepted()
         response.EnsureSuccessStatusCode();
         response.StatusCode.ShouldBe(HttpStatusCode.OK);
     }
+
+    [Fact]
+    public async Task GivenAHealthRequest_ThenItShouldRespondWithOk()
+    {
+        // Arrange
+        var client = _factory.CreateClient(new WebApplicationFactoryClientOptions
+        {
+            BaseAddress = new Uri("https://localhost:60101")
+        });
+
+        // Act
+        var response = await client.GetAsync("/api/health");
+
+        // Assert
+        response.StatusCode.ShouldBe(HttpStatusCode.OK);
+        (await response.Content.ReadAsStringAsync()).ShouldBe("OK");
+    }
 }
diff --git a/src/AzureEventGridSimulator/Controllers/HealthController.cs b/src/AzureEventGridSimulator/Controllers/HealthController.cs
@@ -0,0 +1,14 @@
+using Microsoft.AspNetCore.Mvc;
+
+namespace AzureEventGridSimulator.Controllers;
+
+[Route("/api/health")]
+[ApiController]
+public class HealthController : ControllerBase
+{
+    [HttpGet]
+    public IActionResult Get()
+    {
+        return Ok("OK");
+    }
+}
diff --git a/src/AzureEventGridSimulator/Infrastructure/Middleware/EventGridMiddleware.cs b/src/AzureEventGridSimulator/Infrastructure/Middleware/EventGridMiddleware.cs
@@ -38,6 +38,12 @@ public async Task InvokeAsync(HttpContext context,
             return;
         }
 
+        if (IsHealthRequest(context))
+        {
+            await ValidateHealthRequest(context);
+            return;
+        }
+
         // This is the end of the line.
         await context.WriteErrorResponse(HttpStatusCode.BadRequest, "Request not supported.", null);
     }
@@ -129,6 +135,11 @@ private async Task ValidateNotificationRequest(HttpContext context,
         await _next(context);
     }
 
+    private async Task ValidateHealthRequest(HttpContext context)
+    {
+        await _next(context);
+    }
+
     private static bool IsNotificationRequest(HttpContext context)
     {
         return context.Request.Headers.Keys.Any(k => string.Equals(k, "Content-Type", StringComparison.OrdinalIgnoreCase)) &&
@@ -144,4 +155,10 @@ private static bool IsValidationRequest(HttpContext context)
                context.Request.Query.Keys.Any(k => string.Equals(k, "id", StringComparison.OrdinalIgnoreCase)) &&
                Guid.TryParse(context.Request.Query["id"], out _);
     }
+
+    private static bool IsHealthRequest(HttpContext context)
+    {
+        return context.Request.Method == HttpMethods.Get &&
+            string.Equals(context.Request.Path, "/api/health", StringComparison.OrdinalIgnoreCase);
+    }
 }