add withRenderedAsync

japgolly · Dec 29, 2024 · e6bc306 · e6bc306
1 parent 9b8b79c
commit e6bc306
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/library/testUtil/src/main/scala/japgolly/scalajs/react/test/ReactTestUtils2.scala b/library/testUtil/src/main/scala/japgolly/scalajs/react/test/ReactTestUtils2.scala
@@ -129,4 +129,12 @@ trait ReactTestUtils2 extends japgolly.scalajs.react.test.internal.ReactTestUtil
     F.flatMap(F.pure(ReactTestUtils2.withReactRoot.setup(implicitly, new WithDsl.Cleanup)))(
       root => F.map(actAsync(root.render(unmounted)))(_ => root.selectFirstChild())
     )
+
+  def withRenderedAsync[F[_], A](
+    unmounted: A
+  )(use: TestDomWithRoot => F[Unit]
+  )(implicit F: Async[F], renderable: Renderable[A]): F[Unit] =
+    F.flatMap(renderAsync(unmounted)) { d =>
+      F.finallyRun(use(d), actAsync(d.unmount()))
+    }
 }