From c6cdc8d628a2ea87a4473e86cd5bdd85888d5f8d Mon Sep 17 00:00:00 2001
From: maliming <6908465+maliming@users.noreply.github.com>
Date: Fri, 10 Jul 2020 16:02:54 +0800
Subject: [PATCH] Add TestMemoryDistributedCache for better test.

---
 .../Volo/Abp/Caching/AbpCachingTestModule.cs  |  6 +-
 .../Abp/Caching/TestMemoryDistributedCache.cs | 61 +++++++++++++++++++
 2 files changed, 66 insertions(+), 1 deletion(-)
 create mode 100644 framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/TestMemoryDistributedCache.cs

diff --git a/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/AbpCachingTestModule.cs b/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/AbpCachingTestModule.cs
index be003431f2..9da6725fa5 100644
--- a/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/AbpCachingTestModule.cs
+++ b/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/AbpCachingTestModule.cs
@@ -1,5 +1,7 @@
 using Microsoft.Extensions.Caching.Distributed;
 using System;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
 using Volo.Abp.Modularity;
 
 namespace Volo.Abp.Caching
@@ -26,6 +28,8 @@ namespace Volo.Abp.Caching
 
                 option.GlobalCacheEntryOptions.SetSlidingExpiration(TimeSpan.FromMinutes(20));
             });
+
+            context.Services.Replace(ServiceDescriptor.Singleton<IDistributedCache, TestMemoryDistributedCache>());
         }
     }
-}
\ No newline at end of file
+}
diff --git a/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/TestMemoryDistributedCache.cs b/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/TestMemoryDistributedCache.cs
new file mode 100644
index 0000000000..69589d97cd
--- /dev/null
+++ b/framework/test/Volo.Abp.Caching.Tests/Volo/Abp/Caching/TestMemoryDistributedCache.cs
@@ -0,0 +1,61 @@
+using System.Collections.Generic;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Caching.Distributed;
+using Microsoft.Extensions.Caching.Memory;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Options;
+using Volo.Abp.DependencyInjection;
+
+namespace Volo.Abp.Caching
+{
+    [DisableConventionalRegistration]
+    public class TestMemoryDistributedCache : MemoryDistributedCache, ICacheSupportsMultipleItems
+    {
+        public TestMemoryDistributedCache(IOptions<MemoryDistributedCacheOptions> optionsAccessor)
+            : base(optionsAccessor)
+        {
+        }
+
+        public TestMemoryDistributedCache(IOptions<MemoryDistributedCacheOptions> optionsAccessor, ILoggerFactory loggerFactory)
+            : base(optionsAccessor, loggerFactory)
+        {
+        }
+
+        public byte[][] GetMany(IEnumerable<string> keys)
+        {
+            var values = new List<byte[]>();
+            foreach (var key in keys)
+            {
+                values.Add(Get(key));
+            }
+            return values.ToArray();
+        }
+
+        public async Task<byte[][]> GetManyAsync(IEnumerable<string> keys, CancellationToken token = default)
+        {
+            var values = new List<byte[]>();
+            foreach (var key in keys)
+            {
+                values.Add(await GetAsync(key, token));
+            }
+            return values.ToArray();
+        }
+
+        public void SetMany(IEnumerable<KeyValuePair<string, byte[]>> items, DistributedCacheEntryOptions options)
+        {
+            foreach (var item in items)
+            {
+                Set(item.Key, item.Value, options);
+            }
+        }
+
+        public async Task SetManyAsync(IEnumerable<KeyValuePair<string, byte[]>> items, DistributedCacheEntryOptions options, CancellationToken token = default)
+        {
+            foreach (var item in items)
+            {
+                await SetAsync(item.Key, item.Value, options, token);
+            }
+        }
+    }
+}