Fix tests

Author: javiercn

src/Components/Server/test/Circuits/HybridCacheCircuitPersistenceProviderTest.cs

@@ -2,6 +2,7 @@
 // The .NET Foundation licenses this file to you under the MIT license.
 
 using Microsoft.AspNetCore.Components.Server.Circuits;
+using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.Caching.Hybrid;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging.Abstractions;
@@ -13,12 +14,13 @@ namespace Microsoft.AspNetCore.Components.Server.Tests.Circuits;
 public class HybridCacheCircuitPersistenceProviderTest
 {
     [Fact]
-    public async Task PersistCircuitAsync_StoresCircuitState()
+    public async Task CanPersistAndRestoreState()
     {
         // Arrange
         var hybridCache = CreateHybridCache();
         var circuitId = TestCircuitIdFactory.CreateTestFactory().CreateCircuitId();
-        var persistedState = new PersistedCircuitState() {
+        var persistedState = new PersistedCircuitState()
+        {
             RootComponents = [1, 2, 3],
             ApplicationState = new Dictionary<string, byte[]> {
                 { "key1", new byte[] { 4, 5, 6 } },
@@ -37,25 +39,6 @@ public async Task PersistCircuitAsync_StoresCircuitState()
         Assert.Equal(persistedState.ApplicationState, result.ApplicationState);
     }
 
-    [Fact]
-    public async Task RestoreCircuitAsync_ReturnsPersistedState_WhenCircuitExists()
-    {
-        // Arrange
-        var hybridCache = CreateHybridCache();
-        var circuitId = TestCircuitIdFactory.CreateTestFactory().CreateCircuitId();
-        var persistedState = new PersistedCircuitState();
-        var provider = CreateProvider(hybridCache);
-        var cacheKey = circuitId.Secret;
-
-        await provider.PersistCircuitAsync(circuitId, persistedState);
-
-        // Act
-        var result = await provider.RestoreCircuitAsync(circuitId);
-
-        // Assert
-        Assert.Same(persistedState, result);
-    }
-
     [Fact]
     public async Task RestoreCircuitAsync_ReturnsNull_WhenCircuitDoesNotExist()
     {
@@ -78,7 +61,15 @@ public async Task RestoreCircuitAsync_RemovesCircuitFromCache()
         // Arrange
         var hybridCache = CreateHybridCache();
         var circuitId = TestCircuitIdFactory.CreateTestFactory().CreateCircuitId();
-        var persistedState = new PersistedCircuitState();
+        var persistedState = new PersistedCircuitState()
+        {
+            RootComponents = [1, 2, 3],
+            ApplicationState = new Dictionary<string, byte[]> {
+                { "key1", new byte[] { 4, 5, 6 } },
+                { "key2", new byte[] { 7, 8, 9 } }
+            }
+        };
+
         var provider = CreateProvider(hybridCache);
         var cacheKey = circuitId.Secret;
 
@@ -89,7 +80,10 @@ public async Task RestoreCircuitAsync_RemovesCircuitFromCache()
         var result2 = await provider.RestoreCircuitAsync(circuitId);
 
         // Assert
-        Assert.Same(persistedState, result1);
+        Assert.NotNull(result1);
+        Assert.Equal(persistedState.RootComponents, result1.RootComponents);
+        Assert.Equal(persistedState.ApplicationState, result1.ApplicationState);
+
         Assert.Null(result2); // Circuit should be removed after first restore
     }