small refactorings.

src/DotJEM.Web.Scheduler.Test/MyFrameworkClassTest.cs → src/DotJEM.Web.Scheduler.Test/ScheduledTaskTest.cs

@@ -5,15 +5,6 @@
 
 namespace DotJEM.Web.Scheduler.Test;
 
-public class WebTaskSchedulerTest
-{
-    [Test]
-    public void SayHello_ReturnsHello()
-    {
-
-    }
-}
-
 public class ScheduledTaskTest
 {
     [Test]
@@ -62,17 +53,4 @@ await Task.WhenAll(
 
         Assert.That(executed, Is.EqualTo(1));
     }
-}
-internal static class TaskExt
-{
-    public static void FireAndForget(this Task task)
-    {
-
-    }
-    public static Func<T, Task> ToAsync<T>(this Action<T> action)
-    {
-#pragma warning disable CS1998
-        return async arg => action(arg);
-#pragma warning restore CS1998
-    }
 }

src/DotJEM.Web.Scheduler.Test/TaskExt.cs

@@ -0,0 +1,15 @@
+namespace DotJEM.Web.Scheduler.Test;
+
+internal static class TaskExt
+{
+    public static void FireAndForget(this Task task)
+    {
+
+    }
+    public static Func<T, Task> ToAsync<T>(this Action<T> action)
+    {
+#pragma warning disable CS1998
+        return async arg => action(arg);
+#pragma warning restore CS1998
+    }
+}

src/DotJEM.Web.Scheduler.Test/WebTaskSchedulerTest.cs

@@ -0,0 +1,21 @@
+using NUnit.Framework;
+
+namespace DotJEM.Web.Scheduler.Test;
+
+public class WebTaskSchedulerTest
+{
+    [Test]
+    public async Task SayHello_ReturnsHello()
+    {
+        WebTaskScheduler scheduler = new WebTaskScheduler();
+        IScheduledTask? task = null;
+        bool called = false;
+        task = scheduler.ScheduleTask("Foo", timedOut => { called = true; }, TimeSpan.FromMilliseconds(500));
+        await Task.WhenAny(task.WhenCompleted(), Task.Delay(TimeSpan.FromSeconds(1)));
+        task.Dispose();
+
+
+        Assert.IsTrue(called);
+
+    }
+}

src/DotJEM.Web.Scheduler/IScheduledTask.cs

@@ -71,5 +71,5 @@ public interface IScheduledTask : IDisposable
     /// 
     /// </summary>
     /// <returns></returns>
-    TaskAwaiter<int> GetAwaiter();
+    Task<int> WhenCompleted();
 }

src/DotJEM.Web.Scheduler/ScheduledTask.cs

@@ -94,9 +94,11 @@ public void Start()
         }
     }
 
-    /// <summary>Gets an awaiter used to await this <see cref="ScheduledTask" />, the task is signaled when the task runs to completion and is disposed.</summary>
-    /// <returns>An awaiter instance.</returns>
-    public TaskAwaiter<int> GetAwaiter() => completeCompletionSource.Task.GetAwaiter();
+    /// <summary>
+    /// 
+    /// </summary>
+    /// <returns></returns>
+    public Task<int> WhenCompleted() => completeCompletionSource.Task;
 
     /// <summary>
     /// Registers the next call for the scheduled task onto the threadpool.