Refactor tests to use async store updates

- Updated various test methods to use async store update methods (UpdateAsync) instead of synchronous updates (Update) for better handling of asynchronous operations.
- Changed test component inheritance from StoreComponent to StoreComponentWithUtilities to leverage additional utility methods.
- Ensured that all relevant assertions and actions in tests are compatible with async execution.
- Adjusted middleware and dependency injection tests to accommodate async patterns.

Author: mashrulhaque

samples/EasyAppDev.Blazor.Store.Sample/EasyAppDev.Blazor.Store.Sample.csproj

@@ -15,4 +15,10 @@
     <ProjectReference Include="..\..\src\EasyAppDev.Blazor.Store\EasyAppDev.Blazor.Store.csproj" />
   </ItemGroup>
 
+  <!-- Exclude diagnostics page in Release builds (diagnostics is DEBUG-only) -->
+  <ItemGroup Condition="'$(Configuration)' != 'Debug'">
+    <Content Remove="Pages/Diagnostics.razor" />
+    <Compile Remove="Pages/Diagnostics.razor" />
+  </ItemGroup>
+
 </Project>

samples/EasyAppDev.Blazor.Store.Sample/Pages/ComprehensiveDemo.razor

@@ -1,7 +1,8 @@
 @page "/comprehensive-demo"
-@inherits StoreComponent<ComprehensiveDemoState>
 @using System.Collections.Immutable
 @using EasyAppDev.Blazor.Store.AsyncActions
+@using EasyAppDev.Blazor.Store.Blazor
+@inherits StoreComponentWithUtilities<ComprehensiveDemoState>
 
 <PageTitle>Comprehensive Demo - All Async Features</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Pages/DebounceDemo.razor

@@ -1,6 +1,7 @@
 @page "/debounce-demo"
 @using EasyAppDev.Blazor.Store.Sample.State
-@inherits StoreComponent<DebounceState>
+@using EasyAppDev.Blazor.Store.Blazor
+@inherits StoreComponentWithUtilities<DebounceState>
 
 <PageTitle>Debounce & Throttle Demo</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Pages/Diagnostics.razor

@@ -1,4 +1,5 @@
 @page "/diagnostics"
+@using EasyAppDev.Blazor.Store.Diagnostics.Components
 
 <PageTitle>Store Diagnostics</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Pages/ExecuteAsyncDemo.razor

@@ -1,7 +1,8 @@
 @page "/execute-async-demo"
-@inherits StoreComponent<UserManagementState>
 @using EasyAppDev.Blazor.Store.Sample.State
 @using EasyAppDev.Blazor.Store.AsyncActions
+@using EasyAppDev.Blazor.Store.Blazor
+@inherits StoreComponentWithUtilities<UserManagementState>
 
 <PageTitle>ExecuteAsync Demo</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Pages/FormValidation.razor

@@ -1,6 +1,7 @@
 @page "/form-validation"
-@inherits StoreComponent<FormValidationState>
 @using EasyAppDev.Blazor.Store.Sample.State
+@using EasyAppDev.Blazor.Store.Blazor
+@inherits StoreComponentWithUtilities<FormValidationState>
 
 <PageTitle>Form Validation Demo</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Pages/LazyLoadDemo.razor

@@ -1,6 +1,7 @@
 @page "/lazy-load-demo"
-@inherits StoreComponent<ProductCatalogState>
 @using EasyAppDev.Blazor.Store.Sample.State
+@using EasyAppDev.Blazor.Store.Blazor
+@inherits StoreComponentWithUtilities<ProductCatalogState>
 
 <PageTitle>LazyLoad Demo</PageTitle>
 

samples/EasyAppDev.Blazor.Store.Sample/Program.cs

@@ -6,7 +6,9 @@
 using EasyAppDev.Blazor.Store.Blazor;
 using EasyAppDev.Blazor.Store.Persistence;
 using Microsoft.JSInterop;
+#if DEBUG
 using EasyAppDev.Blazor.Store.Diagnostics;
+#endif
 using EasyAppDev.Blazor.Store.AsyncActions;
 using EasyAppDev.Blazor.Store.Utilities;
 
@@ -16,8 +18,10 @@
 
 builder.Services.AddScoped(sp => new HttpClient { BaseAddress = new Uri(builder.HostEnvironment.BaseAddress) });
 
-// Register diagnostics service - only active in DEBUG builds
+// Register diagnostics service - only available in DEBUG builds
+#if DEBUG
 builder.Services.AddStoreDiagnostics();
+#endif
 
 // Register utility services required by StoreComponent
 // Note: AddStoreUtilities() registers IDebounceManager, IThrottleManager, and ILazyCache
@@ -34,50 +38,44 @@
 // ============================================================================
 builder.Services.AddStore(
     new CounterState(0),
-    (store, sp) => store.WithDefaults(sp, "Counter Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Counter Store"));
 
 // ============================================================================
 // Debounce Store - Demonstrates debounce and throttle functionality
 // ============================================================================
 builder.Services.AddStore(
     new DebounceState(),
-    (store, sp) => store.WithDefaults(sp, "Debounce Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Debounce Store"));
 
 // ============================================================================
 // Todo Store - Demonstrates immutable collections (ImmutableList)
 // ============================================================================
 builder.Services.AddStore(
     TodoState.Empty,
-    (store, sp) => store.WithDefaults(sp, "Todo Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Todo Store"));
 
 // ============================================================================
 // User Profile Store - Demonstrates async actions and loading states
 // ============================================================================
 builder.Services.AddStore(
     UserProfileState.Empty,
-    (store, sp) => store.WithDefaults(sp, "User Profile Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "User Profile Store"));
 
 // ============================================================================
 // AsyncData Demo Store - Demonstrates AsyncData<T> wrapper pattern
 // Simplifies async state from 20+ lines to 1 property!
 // ============================================================================
 builder.Services.AddStore(
     AsyncDataDemoState.Initial,
-    (store, sp) => store.WithDefaults(sp, "AsyncData Demo Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "AsyncData Demo Store"));
 
 // ============================================================================
 // User Management Store - Demonstrates ExecuteAsync helper
 // Automatic error handling - reduces try-catch from 12+ lines to 5 lines!
 // ============================================================================
 builder.Services.AddStore(
     UserManagementState.Initial,
-    (store, sp) => store.WithDefaults(sp, "User Management Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "User Management Store"));
 
 // ============================================================================
 // Shopping Cart Store - Demonstrates persistence with LocalStorage
@@ -87,8 +85,7 @@
 builder.Services.AddStore(
     ShoppingCartState.Empty,
     (store, sp) => store.WithDefaults(sp, "Shopping Cart Store")
-                        .WithPersistence(sp, "shopping-cart-state")
-                        .WithDiagnosticsIfAvailable(sp));
+                        .WithPersistence(sp, "shopping-cart-state"));
 
 // ============================================================================
 // Theme Store - Demonstrates selector optimization with SelectorStoreComponent
@@ -98,17 +95,15 @@
 builder.Services.AddStore(
     ThemeState.Default,
     (store, sp) => store.WithDefaults(sp, "Theme Store")
-                        .WithPersistence(sp, "theme-state")
-                        .WithDiagnosticsIfAvailable(sp));
+                        .WithPersistence(sp, "theme-state"));
 
 // ============================================================================
 // Product Catalog Store - Demonstrates LazyLoad with caching
 // Automatic caching with request deduplication - no manual cache management!
 // ============================================================================
 builder.Services.AddStore(
     new ProductCatalogState(),
-    (store, sp) => store.WithDefaults(sp, "Product Catalog Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Product Catalog Store"));
 
 // ============================================================================
 // Comprehensive Demo Store - Demonstrates ALL async helpers working together
@@ -117,8 +112,7 @@
 // ============================================================================
 builder.Services.AddStore(
     ComprehensiveDemoState.Initial,
-    (store, sp) => store.WithDefaults(sp, "Comprehensive Demo Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Comprehensive Demo Store"));
 
 // ============================================================================
 // Form Validation Store - Demonstrates form state management and validation
@@ -127,8 +121,7 @@
 // ============================================================================
 builder.Services.AddStore(
     new FormValidationState(),
-    (store, sp) => store.WithDefaults(sp, "Form Validation Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Form Validation Store"));
 
 // ============================================================================
 // Cross-Store Demo Stores - Demonstrates Phase 1 updates
@@ -137,12 +130,10 @@
 // ============================================================================
 builder.Services.AddStore(
     AuthDemoState.Initial,
-    (store, sp) => store.WithDefaults(sp, "Auth Demo Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Auth Demo Store"));
 
 builder.Services.AddStore(
     CartDemoState.Empty,
-    (store, sp) => store.WithDefaults(sp, "Cart Demo Store")
-                        .WithDiagnosticsIfAvailable(sp));
+    (store, sp) => store.WithDefaults(sp, "Cart Demo Store"));
 
 await builder.Build().RunAsync();

samples/EasyAppDev.Blazor.Store.Sample/_Imports.razor

@@ -12,4 +12,3 @@
 @using EasyAppDev.Blazor.Store.Core
 @using EasyAppDev.Blazor.Store.Blazor
 @using EasyAppDev.Blazor.Store.Sample.State
-@using EasyAppDev.Blazor.Store.Diagnostics.Components

src/EasyAppDev.Blazor.Store/Blazor/SelectorStoreComponent.cs

@@ -105,36 +105,6 @@ public abstract class SelectorStoreComponent<TState> : ComponentBase, IDisposabl
     /// </remarks>
     protected abstract object SelectState(TState state);
 
-    /// <summary>
-    /// Updates the state using the provided updater function.
-    /// </summary>
-    /// <param name="updater">Function that transforms current state to new state.</param>
-    /// <param name="action">Optional action name for debugging.</param>
-    protected void UpdateState(Func<TState, TState> updater, string? action = null)
-    {
-        Store.Update(updater, action);
-    }
-
-    /// <summary>
-    /// Updates the state asynchronously.
-    /// </summary>
-    /// <param name="updater">Function that transforms current state to new state.</param>
-    /// <param name="action">Optional action name for debugging.</param>
-    protected Task UpdateStateAsync(Func<TState, TState> updater, string? action = null)
-    {
-        return Store.UpdateAsync(updater, action);
-    }
-
-    /// <summary>
-    /// Updates the state asynchronously using an async updater.
-    /// </summary>
-    /// <param name="asyncUpdater">Async function that transforms state.</param>
-    /// <param name="action">Optional action name for debugging.</param>
-    protected Task UpdateStateAsync(Func<TState, Task<TState>> asyncUpdater, string? action = null)
-    {
-        return Store.UpdateAsync(asyncUpdater, action);
-    }
-
     /// <summary>
     /// Updates the state using a transformation function (Zustand-style).
     /// </summary>