IHostApplicationLifetime for testing

Core/Stateflows.Testing/StateflowsTestClass.cs

@@ -12,6 +12,7 @@
 using Stateflows.Common.Registration.Interfaces;
 using Stateflows.Testing.StateMachines.Sequence;
 using Stateflows.Common.Classes;
+using Stateflows.Testing;
 
 namespace StateMachine.IntegrationTests.Utils
 {
@@ -29,15 +30,22 @@ public abstract class StateflowsTestClass
 
         protected ExecutionSequenceObserver ExecutionSequence => ServiceProvider.GetRequiredService<ExecutionSequenceObserver>();
 
+        private TestingHost testingHost;
+        private TestingHost TestingHost => testingHost ??= ServiceProvider.GetRequiredService<TestingHost>();
+
         public virtual void Initialize()
         {
             ServiceCollection.AddStateflows(b => InitializeStateflows(b));
             ServiceCollection.AddSingleton<IExecutionSequenceBuilder, ExecutionSequence>();
+            ServiceCollection.AddSingleton<TestingHost>();
+            ServiceCollection.AddSingleton<IHostApplicationLifetime>(services => services.GetRequiredService<TestingHost>());
             ServiceCollection.AddLogging(builder => builder.AddConsole());
 
             var hostedServices = ServiceProvider.GetRequiredService<IEnumerable<IHostedService>>();
             Task.WaitAll(hostedServices.Select(s => s.StartAsync(new CancellationToken())).ToArray());
 
+            TestingHost.StartApplication();
+
             ContextValues.GlobalValues.Clear();
             ContextValues.StateValues.Clear();
             ContextValues.SourceStateValues.Clear();
@@ -46,6 +54,8 @@ public virtual void Initialize()
 
         public virtual void Cleanup()
         {
+            TestingHost.StopApplication();
+
             _serviceCollection = null;
             _serviceProvider = null;
         }

Core/Stateflows.Testing/TestingHost.cs

@@ -0,0 +1,28 @@
+using System.Threading;
+using Microsoft.Extensions.Hosting;
+
+namespace Stateflows.Testing
+{
+    internal class TestingHost : IHostApplicationLifetime
+    {
+        private readonly CancellationTokenSource ApplicationStartedToken = new CancellationTokenSource();
+        public CancellationToken ApplicationStarted => ApplicationStartedToken.Token;
+
+        private readonly CancellationTokenSource ApplicationStoppingToken = new CancellationTokenSource();
+        public CancellationToken ApplicationStopping => ApplicationStoppingToken.Token;
+
+        private readonly CancellationTokenSource ApplicationStoppedToken = new CancellationTokenSource();
+        public CancellationToken ApplicationStopped => ApplicationStoppedToken.Token;
+
+        public void StartApplication()
+        {
+            ApplicationStartedToken.Cancel();
+        }
+
+        public void StopApplication()
+        {
+            ApplicationStoppingToken.Cancel();
+            ApplicationStoppedToken.Cancel();
+        }
+    }
+}