Added EasyMock::spy() to assert a method is called at least once

Author: mnapoli

README.md

@@ -40,6 +40,14 @@ $mock = EasyMock::mock('My\Class', [
 ]);
 ```
 
+What if you want to assert that the method is called once (i.e. `$mock->expect($this->once())`)? Use `spy()` instead:
+
+```php
+$mock = EasyMock::spy('My\Class', [
+ 'sayHello' => 'Hello',
+]);
+```
+
 ### Features
 
 You can mock methods so that they return values:

src/EasyMock.php

@@ -3,7 +3,9 @@
 namespace EasyMock;
 
 use PHPUnit_Framework_MockObject_Generator;
-use PHPUnit_Framework_MockObject_Matcher_AnyInvokedCount;
+use PHPUnit_Framework_MockObject_Matcher_AnyInvokedCount as AnyInvokedCount;
+use PHPUnit_Framework_MockObject_Matcher_Invocation as InvocationMatcher;
+use PHPUnit_Framework_MockObject_Matcher_InvokedAtLeastOnce as InvokedAtLeastOnce;
 use PHPUnit_Framework_MockObject_MockObject as MockObject;
 
 /**
@@ -33,15 +35,43 @@ public static function mock($classname, array $methods = array())
  }
 
  foreach ($methods as $method => $return) {
- self::mockMethod($mock, $method, $return);
+ self::mockMethod($mock, $method, new AnyInvokedCount, $return);
  }
 
  return $mock;
  }
 
- private static function mockMethod(MockObject $mock, $method, $return)
+ /**
+ * Mock the given class by spying on method calls.
+ *
+ * This is the same as EasyMock::mock() except this assert that methods are called at
+ * least once.
+ *
+ * @see mock()
+ *
+ * @param string $classname The class to mock. Can also be an existing mock to mock new methods.
+ * @param array $methods Array of values to return, indexed by the method name.
+ *
+ * @return \PHPUnit_Framework_MockObject_MockObject
+ */
+ public static function spy($classname, array $methods = array())
+ {
+ if ($classname instanceof MockObject) {
+ $mock = $classname;
+ } else {
+ $mock = self::createMock($classname);
+ }
+
+ foreach ($methods as $method => $return) {
+ self::mockMethod($mock, $method, new InvokedAtLeastOnce, $return);
+ }
+
+ return $mock;
+ }
+
+ private static function mockMethod(MockObject $mock, $method, InvocationMatcher $invocation, $return)
  {
- $methodAssertion = $mock->expects(new PHPUnit_Framework_MockObject_Matcher_AnyInvokedCount)
+ $methodAssertion = $mock->expects($invocation)
  ->method($method);
 
  if (is_callable($return)) {

tests/MockClassTest.php

@@ -102,4 +102,14 @@ public function should_mock_new_methods_on_existing_mock()
 
  $this->assertSame('bar', $mock->foo());
  }
+
+ /**
+ * @test
+ */
+ public function should_allow_to_spy_method_calls()
+ {
+ $mock = EasyMock::spy('EasyMock\Test\Fixture\ClassFixture', array(
+ 'foo' => 'bar',
+ ));
+ }
 }