Make ReactTestUtils.{replace,modify}Props work with any kind of component

Author: japgolly

test/src/main/scala/japgolly/scalajs/react/test/ReactTestUtils.scala

@@ -189,14 +189,16 @@ object ReactTestUtils {
 
   // ===================================================================================================================
 
-  def modifyProps[P, S, B](c: ScalaComponent[P, S, B, CtorType.Props])(m: ScalaComponent.Mounted[P, S, B], f: P => P): ScalaComponent.Mounted[P, S, B] = {
+  def modifyProps[P, U <: GenericComponent.Unmounted[P, M], M <: GenericComponent.MountedImpure[P, _]]
+      (c: GenericComponent[P, CtorType.Props, U], m: M)(f: P => P): M = {
     val container = m.getDOMNode.parentNode
     val p2 = f(m.props)
     c(p2).renderIntoDOM(container.domCast[org.scalajs.dom.raw.Element])
   }
 
-  def replaceProps[P, S, B](c: ScalaComponent[P, S, B, CtorType.Props])(m: ScalaComponent.Mounted[P, S, B], p: P): ScalaComponent.Mounted[P, S, B] =
-    modifyProps(c)(m, _ => p)
+  def replaceProps[P, U <: GenericComponent.Unmounted[P, M], M <: GenericComponent.MountedImpure[P, _]]
+      (c: GenericComponent[P, CtorType.Props, U], m: M)(p: P): M =
+    modifyProps(c, m)(_ => p)
 
   private val reactDataAttrRegex = """\s+data-react\S*?\s*?=\s*?".*?"""".r
   private val reactTextCommentRegex = """<!-- /?react-text[: ].*?-->""".r

test/src/test/scala/japgolly/scalajs/react/test/TestTest.scala

@@ -257,9 +257,9 @@ object TestTest extends TestSuite {
     'modifyProps {
       ReactTestUtils.withRenderedIntoDocument(CP("start")) { m =>
         assertRendered(m.getDOMNode, "<div>none → start</div>")
-        ReactTestUtils.modifyProps(CP)(m, _ + "ed")
+        ReactTestUtils.modifyProps(CP, m)(_ + "ed")
         assertRendered(m.getDOMNode, "<div>start → started</div>")
-        ReactTestUtils.replaceProps(CP)(m, "done!")
+        ReactTestUtils.replaceProps(CP, m)("done!")
         assertRendered(m.getDOMNode, "<div>started → done!</div>")
       }
     }