Added support for multiple calls of the same function with different args

classes/ExpectationProxy.php

@@ -0,0 +1,44 @@
+<?php
+/**
+ * @author Michał Bundyra (webimpress) <contact@webimpress.com>
+ * @license http://www.wtfpl.net/txt/copying/ WTFPL
+ */
+
+namespace phpmock\mockery;
+
+use Mockery\CompositeExpectation;
+use Mockery\MockInterface;
+use phpmock\integration\MockDelegateFunctionBuilder;
+
+/**
+ * Proxy to CompositeExpectation which clear all expectations created on mock.
+ */
+class ExpectationProxy extends CompositeExpectation
+{
+    private $isCleared = false;
+
+    private $mock;
+
+    public function __construct(MockInterface $mock)
+    {
+        $this->mock = $mock;
+    }
+
+    public function __call($name, array $args)
+    {
+        if (! $this->isCleared) {
+            $callback = function () {
+                $this->_mockery_expectations = [];
+            };
+
+            $bind = $callback->bindTo($this->mock, get_class($this->mock));
+            $bind();
+
+            $this->isCleared = true;
+        }
+
+        $expectation = $this->mock->shouldReceive(MockDelegateFunctionBuilder::METHOD);
+
+        return call_user_func_array([$expectation, $name], $args);
+    }
+}

classes/PHPMockery.php

@@ -45,8 +45,8 @@ public static function mock($namespace, $name)
         $delegateBuilder->build($name);
 
         $mockeryMock = Mockery::mock($delegateBuilder->getFullyQualifiedClassName());
-        $expectation = $mockeryMock->makePartial()->shouldReceive(MockDelegateFunctionBuilder::METHOD);
-        
+        $mockeryMock->makePartial()->shouldReceive(MockDelegateFunctionBuilder::METHOD);
+
         $builder = new MockBuilder();
         $builder->setNamespace($namespace)
                 ->setName($name)
@@ -57,7 +57,7 @@ public static function mock($namespace, $name)
         $disabler = new MockDisabler($mock);
         Mockery::getContainer()->rememberMock($disabler);
 
-        return $expectation;
+        return new ExpectationProxy($mockeryMock);
     }
 
     /**

tests/PHPMockeryTest.php

@@ -88,4 +88,35 @@ private function workaroundMockeryIssue268()
             }
         }
     }
+
+    public function testMockDoubleCalls()
+    {
+        $mock = PHPMockery::mock(__NAMESPACE__, 'min');
+        $mock->twice()
+            ->with(1, 10)
+            ->andReturnValues([0, 11]);
+
+        $this->assertSame(0, min(1, 10));
+        $this->assertSame(11, min(1, 10));
+    }
+
+    public function testMockDoubleCallsWithDifferentArgs()
+    {
+        $mock = PHPMockery::mock(__NAMESPACE__, 'max');
+        $mock->with(0, 0)->andReturn(77);
+        $mock
+            ->once()
+            ->with(1, 10)
+            ->andReturn(0);
+        $mock
+            ->twice()
+            ->with(11, 20)
+            ->andReturn(10, 30);
+
+        $this->assertSame(77, max(0, 0));
+        $this->assertSame(0, max(1, 10));
+        $this->assertSame(10, max(11, 20));
+        $this->assertSame(30, max(11, 20));
+        $this->assertSame(77, max(0, 0));
+    }
 }