?

hedgehogqa · Jul 1, 2023 · 4869c93 · 4869c93
1 parent 5697937
commit 4869c93
Showing 1 changed file with 6 additions and 12 deletions.
diff --git a/src/Hedgehog.Xunit/InternalLogic.fs b/src/Hedgehog.Xunit/InternalLogic.fs
@@ -116,19 +116,13 @@ let rec yieldAndCheckReturnValue (x: obj) =
   match x with
   | :? bool        as b -> if not b then TestReturnedFalseException() |> raise
   | :? Task<unit>  as t -> Async.AwaitTask t |> yieldAndCheckReturnValue
-  | :? Task<bool>  as t -> Async.AwaitTask t |> yieldAndCheckReturnValue
   | _ when x <> null && x.GetType().IsGenericType && x.GetType().GetGenericTypeDefinition().IsSubclassOf typeof<Task> ->
-    let genType = x.GetType().GetGenericTypeDefinition()
-    if not genType.ContainsGenericParameters then
-      let t = x :?> Task
-      Async.AwaitTask t |> yieldAndCheckReturnValue
-    else
-      typeof<Async>
-        .GetMethods()
-        .First(fun x -> x.Name = "AwaitTask" && x.IsGenericMethod)
-        .MakeGenericMethod(x.GetType().GetGenericArguments())
-        .Invoke(null, [|x|])
-      |> yieldAndCheckReturnValue
+    typeof<Async>
+      .GetMethods()
+      .First(fun x -> x.Name = "AwaitTask" && x.IsGenericMethod)
+      .MakeGenericMethod(x.GetType().GetGenericArguments())
+      .Invoke(null, [|x|])
+    |> yieldAndCheckReturnValue
   | :? Task        as t -> Async.AwaitTask t |> yieldAndCheckReturnValue
   | :? Async<unit> as a -> Async.RunSynchronously(a, cancellationToken = CancellationToken.None) |> yieldAndCheckReturnValue
   | _ when x <> null && x.GetType().IsGenericType && x.GetType().GetGenericTypeDefinition() = typedefof<Async<_>> ->