Fix tests

Author: javiercn

src/Components/Components/src/PersistentComponentState.cs

@@ -31,7 +31,6 @@ internal PersistentComponentState(
 
     internal bool PersistingState { get; set; }
 
-    // TODO: Replace IPersistentComponentStateScenario with RestoreContext
     internal void InitializeExistingState(IDictionary<string, byte[]> existingState, RestoreContext context)
     {
         if (_existingState != null)
@@ -89,7 +88,6 @@ public RestoringComponentStateSubscription RegisterOnRestoring(Action callback,
 
         if (options.AllowUpdates)
         {
-            // TODO: Remove the filter from registration
             var registration = new RestoreComponentStateRegistration(callback);
             _registeredRestoringCallbacks.Add(registration);
             return new RestoringComponentStateSubscription(_registeredRestoringCallbacks, registration);

src/Components/Components/test/PersistentStateValueProviderTests.cs

@@ -52,6 +52,9 @@ public void Subscribe_RegistersPersistenceCallback()
             new Dictionary<string, byte[]>(),
             [],
             []);
+
+        InitializeState(state, []);
+
         var provider = new PersistentStateValueProvider(state, NullLogger<PersistentStateValueProvider>.Instance, new ServiceCollection().BuildServiceProvider());
         var renderer = new TestRenderer();
         var component = new TestComponent();
@@ -76,6 +79,8 @@ public void Unsubscribe_RemovesCallbackFromRegisteredCallbacks()
             [],
             []);
 
+        InitializeState(state, []);
+
         var provider = new PersistentStateValueProvider(state, NullLogger<PersistentStateValueProvider>.Instance, new ServiceCollection().BuildServiceProvider());
 
         var renderer = new TestRenderer();
@@ -102,7 +107,7 @@ private static void InitializeState(PersistentComponentState state, List<(Compon
             var key = PersistentStateValueProviderKeyResolver.ComputeKey(item.componentState, item.propertyName);
             dictionary[key] = JsonSerializer.SerializeToUtf8Bytes(item.value, JsonSerializerOptions.Web);
         }
-        state.InitializeExistingState(dictionary, null);
+        state.InitializeExistingState(dictionary, RestoreContext.InitialValue);
     }
 
     private static CascadingParameterInfo CreateCascadingParameterInfo(string propertyName, Type propertyType)