fix: update return type of addResult

tflori · Sep 15, 2019 · a925132 · a925132
1 parent 4096939
commit a925132
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/Testing/EntityManagerMock.php b/src/Testing/EntityManagerMock.php
@@ -2,6 +2,7 @@
 
 namespace ORM\Testing;
 
+use Mockery as m;
 use ORM\Entity;
 use ORM\EntityFetcher;
 use ORM\EntityManager;
@@ -56,7 +57,7 @@ public function retrieve($class, array $primaryKey)
      *
      * @param $class
      * @param Entity ...$entities
-     * @return Result
+     * @return Result|m\MockInterface
      * @codeCoverageIgnore proxy method
      */
     public function addResult($class, Entity ...$entities)

diff --git a/src/Testing/MocksEntityManager.php b/src/Testing/MocksEntityManager.php
@@ -129,7 +129,7 @@ function ($str) {
      *
      * @param string $class The class of an Entity
      * @param Entity ...$entities The entities that will be returned
-     * @return EntityFetcherMock\Result
+     * @return EntityFetcherMock\Result|m\MockInterface
      * @codeCoverageIgnore trivial code
      */
     public function ormAddResult($class, Entity ...$entities)