Fixes CancellationTask overload

Author: TheAngryByrd

src/FsToolkit.ErrorHandling.IcedTasks/CancellableTaskResultCE.fs

@@ -18,7 +18,7 @@ module CancellableTaskResultCE =
     open IcedTasks
 
     /// CancellationToken -> Task<Result<'T, 'Error>>
-    type CancellableTaskResult<'T, 'Error> = CancellationToken -> TaskResult<'T, 'Error>
+    type CancellableTaskResult<'T, 'Error> = CancellableTask<Result<'T, 'Error>>
 
 
     /// The extra data stored in ResumableStateMachine for tasks
@@ -855,6 +855,13 @@ module CancellableTaskResultCE =
                     return Ok r
                 }
 
+
+            member inline _.Source(result: CancellableTask) : CancellableTaskResult<unit, 'Error> =
+                cancellableTask {
+                    let! r = result
+                    return Ok r
+                }
+
             member inline _.Source(result: ColdTask<_>) : CancellableTaskResult<_, _> =
                 fun _ ->
                     // TODO: using `coldTask` results in "internal error: The local field ResumptionDynamicInfo was referenced but not declare" compliation error

tests/FsToolkit.ErrorHandling.IcedTasks.Tests/CancellableTaskResultCE.fs

@@ -203,7 +203,7 @@ module CancellableTaskResultCE =
                         task {
 
                             let ctr =
-                                cancellableTaskResult { return! fun ct -> Task.CompletedTask }
+                                cancellableTaskResult { return! fun (ct : CancellationToken) -> Task.CompletedTask }
 
                             let! actual = ctr CancellationToken.None
                             Expect.equal actual (Ok()) "Should be able to Return! CancellableTask"
@@ -477,10 +477,24 @@ module CancellableTaskResultCE =
 
                             let ctr =
                                 cancellableTaskResult {
-                                    let! someValue = fun ct -> Task.CompletedTask
+                                    let! someValue = fun (ct : CancellationToken) -> Task.CompletedTask
                                     return someValue
                                 }
 
+                            let! actual = ctr CancellationToken.None
+                            Expect.equal actual (Ok data) "Should be able to let! CancellableTask"
+                        }
+                    testCaseTask "do! CancellableTask"
+                    <| fun () ->
+                        task {
+                            let data = ()
+
+                            let ctr =
+                                cancellableTaskResult {
+                                    do! fun (ct : CancellationToken) -> Task.CompletedTask
+                                    // return someValue
+                                }
+
                             let! actual = ctr CancellationToken.None
                             Expect.equal actual (Ok data) "Should be able to let! CancellableTask"
                         }