fix: Moves cascading authentication state to service container (closes #1774)

Author: linkdotnet

CHANGELOG.md

@@ -6,6 +6,9 @@ All notable changes to **bUnit** will be documented in this file. The project ad
 
 ## [Unreleased]
 
+### Changed
+- Registering `AuthenticationState` in the services container rather than as part of the RenderTree. Fixes [#1774](https://github.com/bUnit-dev/bUnit/issues/1774) reported by [@aayjaychan](https://github.com/aayjaychan).
+
 ## [2.0.66] - 2025-11-11
 
 This major release focuses on platform updates and API simplifications.

src/bunit/RemovedInFutureVersionAttribute.cs

@@ -0,0 +1,7 @@
+namespace Bunit;
+
+[AttributeUsage(AttributeTargets.All, Inherited = false)]
+internal sealed class RemovedInFutureVersionAttribute(string todo) : Attribute
+{
+	public string Todo { get; } = todo;
+}

src/bunit/TestDoubles/Authorization/BunitAuthorizationExtensions.cs

@@ -11,9 +11,10 @@ public partial class BunitContext
 	/// an authenticated user, as well as adding the <see cref="CascadingAuthenticationState"/> component to the
 	/// test contexts render tree.
 	/// </summary>
+	[RemovedInFutureVersion("SignOutSessionStateManager should be removed from the container.")]
 	public BunitAuthorizationContext AddAuthorization()
 	{
-		RenderTree.TryAdd<CascadingAuthenticationState>();
+		Services.AddCascadingAuthenticationState();
 		Services.AddSingleton<BunitSignOutSessionStateManager>();
 #pragma warning disable CS0618
 		Services.AddSingleton<SignOutSessionStateManager>(s => s.GetRequiredService<BunitSignOutSessionStateManager>());

src/bunit/TestDoubles/NavigationManager/BunitSignOutSessionStateManager.cs

@@ -7,6 +7,7 @@ namespace Bunit.TestDoubles;
 /// that will help later to assert if the user was logged out
 /// </summary>
 #pragma warning disable CS0618
+[RemovedInFutureVersion("SignOutSessionStateManager is obsolete")]
 public class BunitSignOutSessionStateManager : SignOutSessionStateManager
 #pragma warning restore CS0618
 {

tests/bunit.tests/TestDoubles/Components/StubTest.cs

@@ -1,3 +1,5 @@
+using Microsoft.AspNetCore.Components.Authorization;
+
 namespace Bunit.TestDoubles.Components;
 
 public class StubTest : BunitContext
@@ -25,4 +27,21 @@ public void Test002(string header, string attrValue)
 				ps => ps.ShouldContain(x => x.Key == nameof(Simple1.AttrValue) && attrValue.Equals(x.Value)),
 				ps => ps.Count.ShouldBe(2));
 	}
+
+	[Fact(DisplayName = "Stubbing everything except cut keeps cascading AuthenticationState")]
+	public void Test003()
+	{
+		AddAuthorization();
+		ComponentFactories.AddStub(c => c != typeof(AuthTestComponent));
+		
+		var cut = Render<AuthTestComponent>();
+
+		cut.Instance.Auth.ShouldNotBeNull();
+	}
+
+	private sealed class AuthTestComponent : ComponentBase
+	{
+		[CascadingParameter]
+		public Task<AuthenticationState> Auth { get; set; } = null!;
+	}
 }