feat: Implement cache remove value policy in emulator (#69)

Author: Mielek

src/Testing/Document/MockCacheRemoveValueProvider.cs

@@ -0,0 +1,39 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using Azure.ApiManagement.PolicyToolkit.Authoring;
+using Azure.ApiManagement.PolicyToolkit.Testing.Emulator.Policies;
+
+namespace Azure.ApiManagement.PolicyToolkit.Testing.Document;
+
+public static class MockCacheRemoveValueProvider
+{
+    public static Setup CacheRemoveValue<T>(this MockPoliciesProvider<T> mock) where T : class =>
+        CacheRemoveValue(mock, (_, _) => true);
+
+    public static Setup CacheRemoveValue<T>(
+        this MockPoliciesProvider<T> mock,
+        Func<GatewayContext, CacheRemoveValueConfig, bool> predicate
+    ) where T : class
+    {
+        var handler = mock.SectionContextProxy.GetHandler<CacheRemoveValueHandler>();
+        return new Setup(predicate, handler);
+    }
+
+    public class Setup
+    {
+        private readonly Func<GatewayContext, CacheRemoveValueConfig, bool> _predicate;
+        private readonly CacheRemoveValueHandler _handler;
+
+        internal Setup(
+            Func<GatewayContext, CacheRemoveValueConfig, bool> predicate,
+            CacheRemoveValueHandler handler)
+        {
+            _predicate = predicate;
+            _handler = handler;
+        }
+
+        public void WithCallback(Action<GatewayContext, CacheRemoveValueConfig> callback) =>
+            _handler.CallbackSetup.Add((_predicate, callback).ToTuple());
+    }
+}

src/Testing/Emulator/Policies/CacheRemoveValueHandler.cs

@@ -5,13 +5,19 @@
 
 namespace Azure.ApiManagement.PolicyToolkit.Testing.Emulator.Policies;
 
-[Section(nameof(IInboundContext))]
+[
+    Section(nameof(IInboundContext)),
+    Section(nameof(IBackendContext)),
+    Section(nameof(IOutboundContext)),
+    Section(nameof(IOnErrorContext))
+]
 internal class CacheRemoveValueHandler : PolicyHandler<CacheRemoveValueConfig>
 {
     public override string PolicyName => nameof(IInboundContext.CacheRemoveValue);
 
     protected override void Handle(GatewayContext context, CacheRemoveValueConfig config)
     {
-        throw new NotImplementedException();
+        var store = context.CacheStore.GetCache(config.CachingType ?? "prefer-external");
+        store?.Remove(config.Key);
     }
 }

test/Test.Testing/Emulator/Policies/CacheRemoveValueTests.cs

@@ -0,0 +1,57 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using Azure.ApiManagement.PolicyToolkit.Authoring;
+using Azure.ApiManagement.PolicyToolkit.Testing;
+using Azure.ApiManagement.PolicyToolkit.Testing.Document;
+
+namespace Test.Emulator.Emulator.Policies;
+
+[TestClass]
+public class CacheRemoveValueTests
+{
+    class SimpleCacheRemoveValue : IDocument
+    {
+        public void Inbound(IInboundContext context)
+        {
+            context.CacheRemoveValue(new CacheRemoveValueConfig() { Key = "key" });
+        }
+    }
+
+    [TestMethod]
+    public void CacheRemoveValue_Callback()
+    {
+        var test = new SimpleCacheRemoveValue().AsTestDocument();
+        var executedCallback = false;
+        test.SetupInbound().CacheRemoveValue().WithCallback((_, _) =>
+        {
+            executedCallback = true;
+        });
+
+        test.RunInbound();
+
+        executedCallback.Should().BeTrue();
+    }
+
+    [TestMethod]
+    public void CacheRemoveValue_SetupCacheRemove()
+    {
+        var test = new SimpleCacheRemoveValue().AsTestDocument();
+        var cacheStore = test.SetupCacheStore().WithInternalCacheValue("key", "value");
+
+        test.RunInbound();
+
+        cacheStore.InternalCache.Should().NotContainKey("key");
+    }
+
+    [TestMethod]
+    public void CacheRemoveValue_SetupCacheRemove_WithExternalCacheSetup()
+    {
+        var test = new SimpleCacheRemoveValue().AsTestDocument();
+        var cacheStore = test.SetupCacheStore().WithExternalCacheSetup().WithExternalCacheValue("key", "value");
+
+        test.RunInbound();
+
+        cacheStore.ExternalCache.Should().NotContainKey("key");
+    }
+}