Throw exception on non unique mocked method

This patch will make a mock throw an exception when multiple matchers can be applied to a invoke. When allowing this, results of tests are not predictable. refs sebastianbergmann#4255
jaapio · Jun 1, 2020 · 1ce8bfb · 1ce8bfb
1 parent 61f7c75
commit 1ce8bfb
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 22 deletions.
diff --git a/src/Framework/MockObject/InvocationHandler.php b/src/Framework/MockObject/InvocationHandler.php
@@ -113,30 +113,11 @@ public function expects(InvocationOrder $rule): InvocationMocker
     public function invoke(Invocation $invocation)
     {
         $exception      = null;
-        $hasReturnValue = false;
         $returnValue    = null;
+        $match          = $this->findMatcher($invocation);
 
-        foreach ($this->matchers as $match) {
-            try {
-                if ($match->matches($invocation)) {
-                    $value = $match->invoked($invocation);
-
-                    if (!$hasReturnValue) {
-                        $returnValue    = $value;
-                        $hasReturnValue = true;
-                    }
-                }
-            } catch (\Exception $e) {
-                $exception = $e;
-            }
-        }
-
-        if ($exception !== null) {
-            throw $exception;
-        }
-
-        if ($hasReturnValue) {
-            return $returnValue;
+        if ($match !== null) {
+            return $match->invoked($invocation);
         }
 
         if (!$this->returnValueGeneration) {
@@ -160,6 +141,28 @@ public function invoke(Invocation $invocation)
         return $invocation->generateReturnValue();
     }
 
+    private function findMatcher(Invocation $invocation): ?Matcher
+    {
+        $result = [];
+        foreach ($this->matchers as $matcher) {
+            if ($matcher->matches($invocation)) {
+                $result[] = $matcher;
+            }
+        }
+
+        if (count($result) > 1) {
+            throw new ExpectationFailedException(
+                sprintf(
+                    'Non unique mocked method invocation: %s::%s',
+                    $invocation->getClassName(),
+                    $invocation->getMethodName()
+                )
+            );
+        }
+
+        return current($result) ?: null;
+    }
+
     public function matches(Invocation $invocation): bool
     {
         foreach ($this->matchers as $matcher) {

diff --git a/tests/unit/Framework/MockObject/Builder/InvocationMockerTest.php b/tests/unit/Framework/MockObject/Builder/InvocationMockerTest.php
@@ -9,6 +9,7 @@
  * file that was distributed with this source code.
  */
 use PHPUnit\Framework\Constraint\IsEqual;
+use PHPUnit\Framework\ExpectationFailedException;
 use PHPUnit\Framework\MockObject\Builder\InvocationMocker;
 use PHPUnit\Framework\MockObject\IncompatibleReturnValueException;
 use PHPUnit\Framework\MockObject\InvocationHandler;
@@ -257,4 +258,26 @@ public function testWillReturnAlreadyInstantiatedStubs(): void
         $this->assertSame('foo', $mock->foo());
         $this->assertSame($mock, $mock->bar());
     }
+
+    public function testMultipleWithParametersWillReturnLatestDefined(): void
+    {
+        $mock = $this->getMockBuilder(stdClass::class)
+            ->setMethods(['foo'])
+            ->getMock();
+
+        $mock->expects($this->any())
+            ->method('foo')
+            ->with('bar')
+            ->willReturn('first');
+
+        $mock->expects($this->any())
+            ->method('foo')
+            ->with('foo')
+            ->willReturn('second');
+
+       $this->expectException(ExpectationFailedException::class);
+       $this->getExpectedExceptionMessage('Non unique mocked method invocation: stdClass::foo');
+
+        $mock->foo('bar');
+    }
 }