Add generic types for ObjectProphecy

src/Prophecy/Prophecy/ObjectProphecy.php

@@ -26,6 +26,9 @@
  * Object prophecy.
  *
  * @author Konstantin Kudryashov <[email protected]>
+ *
+ * @template-covariant T of object
+ * @template-implements ProphecyInterface<T>
  */
 class ObjectProphecy implements ProphecyInterface
 {
@@ -66,6 +69,10 @@ public function __construct(
      * @param string $class
      *
      * @return $this
+     *
+     * @template U
+     * @phpstan-param class-string<U> $class
+     * @phpstan-this-out static<T&U>
      */
     public function willExtend($class)
     {
@@ -80,6 +87,10 @@ public function willExtend($class)
      * @param string $interface
      *
      * @return $this
+     *
+     * @template U
+     * @phpstan-param class-string<U> $class
+     * @phpstan-this-out static<T&U>
      */
     public function willImplement($interface)
     {
@@ -108,6 +119,8 @@ public function willBeConstructedWith(array $arguments = null)
      * @return object
      *
      * @throws \Prophecy\Exception\Prophecy\ObjectProphecyException If double doesn't implement needed interface
+     *
+     * @phpstan-return T
      */
     public function reveal()
     {

src/Prophecy/Prophecy/ProphecyInterface.php

@@ -15,13 +15,17 @@
  * Core Prophecy interface.
  *
  * @author Konstantin Kudryashov <[email protected]>
+ *
+ * @template-covariant T of object
  */
 interface ProphecyInterface
 {
     /**
      * Reveals prophecy object (double) .
      *
      * @return object
+     *
+     * @phpstan-return T
      */
     public function reveal();
 }

src/Prophecy/Prophet.php

@@ -74,6 +74,10 @@ public function __construct(
      * @param null|string $classOrInterface Class or interface name
      *
      * @return ObjectProphecy
+     *
+     * @template T of object
+     * @phpstan-param class-string<T>|null $classOrInterface
+     * @phpstan-return ObjectProphecy<T>
      */
     public function prophesize($classOrInterface = null)
     {