add test to show corrupted items filtered from result

DuendeSoftware · Jan 5, 2024 · 2ce4286 · 2ce4286
1 parent 893b89f
commit 2ce4286
Showing 1 changed file with 80 additions and 1 deletion.
diff --git a/test/IdentityServer.UnitTests/Services/Default/DefaultPersistedGrantServiceTests.cs b/test/IdentityServer.UnitTests/Services/Default/DefaultPersistedGrantServiceTests.cs
@@ -5,6 +5,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Reflection;
 using System.Security.Claims;
 using System.Threading.Tasks;
 using Duende.IdentityServer;
@@ -540,4 +541,82 @@ await _userConsent.StoreUserConsentAsync(new Consent()
         grants.Count().Should().Be(1);
         grants.First().Scopes.Should().Contain(new string[] { "foo1", "foo2", "quux3" });
     }
-}
+
+    [Fact]
+    public async Task GetAllGrantsAsync_should_filter_items_with_corrupt_data_from_result()
+    {
+        var mockStore = new CorruptingPersistedGrantStore(_store) 
+        {
+            ClientIdToCorrupt = "client2"
+        };
+
+        _subject = new DefaultPersistedGrantService(
+           mockStore,
+           new PersistentGrantSerializer(),
+           TestLogger.Create<DefaultPersistedGrantService>());
+
+        await _userConsent.StoreUserConsentAsync(new Consent()
+        {
+            ClientId = "client1",
+            SubjectId = "123",
+            Scopes = new string[] { "foo1", "foo2" }
+        });
+        await _userConsent.StoreUserConsentAsync(new Consent()
+        {
+            ClientId = "client2",
+            SubjectId = "123",
+            Scopes = new string[] { "foo3" }
+        });
+
+        var grants = await _subject.GetAllGrantsAsync("123");
+
+        grants.Count().Should().Be(1);
+        grants.First().Scopes.Should().Contain(new string[] { "foo1", "foo2" });
+    }
+
+    class CorruptingPersistedGrantStore : IPersistedGrantStore
+    {
+        public string ClientIdToCorrupt { get; set; }
+
+        private IPersistedGrantStore _inner;
+
+        public CorruptingPersistedGrantStore(IPersistedGrantStore inner)
+        {
+            _inner = inner;
+        }
+
+        public async Task<IEnumerable<PersistedGrant>> GetAllAsync(PersistedGrantFilter filter)
+        {
+            var items = await _inner.GetAllAsync(filter);
+            if (ClientIdToCorrupt != null)
+            {
+                var itemsToCorrupt = items.Where(x => x.ClientId == ClientIdToCorrupt);
+                foreach(var corruptItem in itemsToCorrupt)
+                {
+                    corruptItem.Data = "corrupt";
+                }
+            }
+            return items;
+        }
+
+        public Task<PersistedGrant> GetAsync(string key)
+        {
+            return _inner.GetAsync(key);
+        }
+
+        public Task RemoveAllAsync(PersistedGrantFilter filter)
+        {
+            return _inner.RemoveAllAsync(filter);
+        }
+
+        public Task RemoveAsync(string key)
+        {
+            return _inner.RemoveAsync(key);
+        }
+
+        public Task StoreAsync(PersistedGrant grant)
+        {
+            return _inner.StoreAsync(grant);
+        }
+    }
+}