Added async setup method to snapshot test

Author: egil

src/Components/Fixture.cs

@@ -13,9 +13,9 @@ namespace Egil.RazorComponents.Testing
     public class Fixture : FragmentBase
     {
         private Action _setup = NoopTestMethod;
-        private Func<Task> _setupAsync = NoopAsyncTestMethod;
+        private Func<Task> _setupAsync = NoopTestMethodAsync;
         private Action _test = NoopTestMethod;
-        private Func<Task> _testAsync = NoopAsyncTestMethod;
+        private Func<Task> _testAsync = NoopTestMethodAsync;
         private IReadOnlyCollection<Action> _tests = Array.Empty<Action>();
         private IReadOnlyCollection<Func<Task>> _testsAsync = Array.Empty<Func<Task>>();
 
@@ -34,7 +34,7 @@ public class Fixture : FragmentBase
         /// Gets or sets the asynchronous setup action to perform before the <see cref="Test"/> action,
         /// <see cref="TestAsync"/> action and <see cref="Tests"/> and <see cref="TestsAsync"/> actions are invoked.
         /// </summary>
-        [Parameter] public Func<Task> SetupAsync { get => _setupAsync; set => _setupAsync = value ?? NoopAsyncTestMethod; }
+        [Parameter] public Func<Task> SetupAsync { get => _setupAsync; set => _setupAsync = value ?? NoopTestMethodAsync; }
 
         /// <summary>
         /// Gets or sets the first test action to invoke, after the <see cref="Setup"/> action has
@@ -52,7 +52,7 @@ public class Fixture : FragmentBase
         /// Use this to assert against the <see cref="ComponentUnderTest"/> and <see cref="Fragment"/>'s
         /// defined in the <see cref="Fixture"/>.
         /// </summary>
-        [Parameter] public Func<Task> TestAsync { get => _testAsync; set => _testAsync = value ?? NoopAsyncTestMethod; }
+        [Parameter] public Func<Task> TestAsync { get => _testAsync; set => _testAsync = value ?? NoopTestMethodAsync; }
 
         /// <summary>
         /// Gets or sets the test actions to invoke, one at the time, in the order they are placed 
@@ -73,9 +73,5 @@ public class Fixture : FragmentBase
         /// defined in the <see cref="Fixture"/>.
         /// </summary>
         [Parameter] public IReadOnlyCollection<Func<Task>> TestsAsync { get => _testsAsync; set => _testsAsync = value ?? Array.Empty<Func<Task>>(); }
-
-        private static void NoopTestMethod() { }
-
-        private static Task NoopAsyncTestMethod() => Task.CompletedTask;
     }
 }

src/Components/FragmentBase.cs

@@ -9,6 +9,9 @@ namespace Egil.RazorComponents.Testing
     /// </summary>
     public abstract class FragmentBase : IComponent
     {
+        internal static void NoopTestMethod() { }
+        internal static Task NoopTestMethodAsync() => Task.CompletedTask;
+
         /// <summary>
         /// Gets or sets the child content of the fragment.
         /// </summary>

src/Components/SnapshotTest.cs

@@ -16,19 +16,24 @@ namespace Egil.RazorComponents.Testing
     public class SnapshotTest : FragmentBase
     {
         private Action _setup = NoopTestMethod;
+        private Func<Task> _setupAsync = NoopTestMethodAsync;
 
         /// <summary>
         /// A description or name for the test that will be displayed if the test fails.
         /// </summary>
         [Parameter] public string? Description { get; set; }
 
         /// <summary>
-        /// A method to be called <see cref="TestInput"/> component and <see cref="ExpectedOutput"/> component
-        /// is rendered. Use to e.g. setup services that the test input needs to render.
+        /// Gets or sets the setup action to perform before the <see cref="TestInput"/> and <see cref="ExpectedOutput"/>
+        /// is rendered and compared.
         /// </summary>
         [Parameter] public Action Setup { get => _setup; set => _setup = value ?? NoopTestMethod; }
 
-        private static void NoopTestMethod() { }
+        /// <summary>
+        /// Gets or sets the setup action to perform before the <see cref="TestInput"/> and <see cref="ExpectedOutput"/>
+        /// is rendered and compared.
+        /// </summary>
+        [Parameter] public Func<Task> SetupAsync { get => _setupAsync; set => _setupAsync = value ?? NoopTestMethodAsync; }
     }
 
     /// <summary>

src/Components/TestComponentBase.cs

@@ -56,7 +56,7 @@ public async Task RazorTest()
             container.Render(BuildRenderTree);
 
             await ExecuteFixtureTests(container).ConfigureAwait(false);
-            ExecuteSnapshotTests(container);
+            await ExecuteSnapshotTests(container).ConfigureAwait(false);
         }
 
         /// <inheritdoc/>
@@ -148,7 +148,7 @@ private static async Task InvokeFixtureAction(Fixture fixture, Func<Task> action
             }
         }
 
-        private void ExecuteSnapshotTests(ContainerComponent container)
+        private async Task ExecuteSnapshotTests(ContainerComponent container)
         {
             foreach (var (_, snapshot) in container.GetComponents<SnapshotTest>())
             {
@@ -157,6 +157,7 @@ private void ExecuteSnapshotTests(ContainerComponent container)
 
                 var context = _testContextAdapter.ActivateSnapshotTestContext(testData);
                 snapshot.Setup();
+                await snapshot.SetupAsync().ConfigureAwait(false);
                 var actual = context.RenderTestInput();
                 var expected = context.RenderExpectedOutput();
                 actual.MarkupMatches(expected, snapshot.Description);

tests/Components/TestComponentBaseTest/SnapshotTestTest.razor

@@ -0,0 +1,13 @@
+@inherits TestComponentBase
+@using Shouldly
+
+<SnapshotTest Setup="() => Services">
+    <TestInput>
+        <SimpleWithDeps></SimpleWithDeps>
+        <SimpleWithAyncDeps></SimpleWithAyncDeps>
+    </TestInput>
+    <ExpectedOutput>
+        @inject ITestDep testDep
+<p>@testDep.Name</p>
+    </ExpectedOutput>
+</SnapshotTest>