Don't dispose scope until job completes

src/Diagnostics/Properties/launchSettings.json

@@ -2,7 +2,7 @@
   "profiles": {
     "Diagnostics": {
       "commandName": "Project",
-      "commandLineArgs": "diagnosticjob --milliseconddelayperitem 10 --numberofitems 5000 --throwexceptioningetitemidasync"
+      "commandLineArgs": "diagnosticjob --milliseconddelayperitem 10 --numberofitems 5000"
     }
   }
 }

src/Runly/JobHost.cs

@@ -133,11 +133,10 @@ public static Task RunJobAsync(this IHost host)
 		/// <returns>The <see cref="Task"/> that represents the asynchronous operation.</returns>
 		public static Task RunJobAsync(this IHost host, CancellationToken cancellationToken)
 		{
-			using var scope = host.Services.CreateAsyncScope();
-
+			var scope = host.Services.CreateAsyncScope();
             var action = scope.ServiceProvider.GetRequiredService<IHostAction>();
 
-			return action?.RunAsync(cancellationToken) ?? Task.CompletedTask;
+			return action?.RunAsync(cancellationToken).ContinueWith(async action => await scope.DisposeAsync()) ?? Task.CompletedTask;
 		}
 	}
 }

test/Runly.Tests/Dependencies.cs

@@ -1,8 +1,25 @@
-namespace Runly.Tests
+using System;
+using System.Threading.Tasks;
+
+namespace Runly.Tests
 {
 	public interface IDep1 { }
-	public class Dep1 : IDep1 { }
-	public interface IDep2 { }
+    public class Dep1 : IDep1, IAsyncDisposable
+    {
+        public static bool IsDisposed { get; set; }
+
+        public Dep1()
+        {
+			IsDisposed = false;
+        }
+
+		public ValueTask DisposeAsync()
+		{
+			IsDisposed = true;
+			return ValueTask.CompletedTask;
+		}
+    }
+    public interface IDep2 { }
 	public class Dep2 : IDep2 { }
 	public class Dep3 { }
 	public class Dep4 { }

test/Runly.Tests/Scenarios/Running/Running_a_job.cs

@@ -46,7 +46,15 @@ public async Task should_run_a_job_with_scoped_dependency_in_constructor()
                  })
                 .Build();
 
-            await action.RunJobAsync();
+			Dep1.IsDisposed.Should().BeFalse();
+
+            var run = action.RunJobAsync();
+
+            Dep1.IsDisposed.Should().BeFalse();
+
+            await run;
+
+			Dep1.IsDisposed.Should().BeTrue();
         }
     }
 }