Fix test of run delay const law in tasks

tpetricek · Nov 30, 2012 · 3513a81 · 3513a81
1 parent 32cb925
commit 3513a81
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/tests/FSharpx.Tests/TaskTests.fs b/tests/FSharpx.Tests/TaskTests.fs
@@ -134,20 +134,17 @@ let ``while``() =
     Task.run t |> ignore
     Assert.AreEqual(0, !i)
 
+open FsCheck
+open FsCheck.NUnit
+
+type TaskGen =
+    static member TaskArb =
+        Arb.generate |> Gen.map Task.returnM |> Arb.fromGen
+
 [<Test>]
-let ``run delay with obj task``() =    
+let ``run delay law``() =
+    Arb.register<TaskGen>() |> ignore
     let task = Task.TaskBuilder(continuationOptions = TaskContinuationOptions.ExecuteSynchronously)
-    let objTask = Task.Factory.StartNew(fun () -> 0 :> obj)
-    let c = konst objTask
-    let delayed = task.Delay c
-    let result = task.Run delayed
-    Assert.AreSame(objTask,result)
-
-//open FsCheck.NUnit
-//
-//[<Test>]
-//let ``run delay law``() =    
-//    let task = Task.TaskBuilder(continuationOptions = TaskContinuationOptions.ExecuteSynchronously)
-//    fsCheck "run delay law" (fun a -> (task.Run << task.Delay << konst) a = a)
+    fsCheck "run delay law" (fun a -> (task.Run << task.Delay << konst) a = a)
 
 #endif