stub helper using trait

gistfile1.php

<?php
namespace Rj\CoreBundle\Test\Phpunit\Helper;

/**
 * Usage:
 * 
 * ```php
 * <?php
 *     $fooStub = $this->getStub('FooClass', array(
 *         'barMethod' => $stub
 *     ));
 * ```
 * 
 * More info you could in issue: {@link https://github.com/sebastianbergmann/phpunit/issues/550}
 * 
 * @author Kotlyar Maksim <[email protected]>
 * @since 10/1/12
 */
trait Stub
{
    public function getStub($originalClassName, $methods = array(), array $arguments = array(), $mockClassName = '', $callOriginalConstructor = TRUE, $callOriginalClone = TRUE, $callAutoload = TRUE)
    {
        $mock = $this->getMock(
            $originalClassName,
            array_keys($methods),
            $arguments,
            $mockClassName,
            $callOriginalConstructor,
            $callOriginalClone,
            $callAutoload
        );

        foreach ($methods as $name => $stub) {
            if (false == $stub instanceof \PHPUnit_Framework_MockObject_Stub) {
                $stub = $this->returnValue($stub);
            }

            $mock
                ->expects($this->any())
                ->method($name)
                ->will($stub)
            ;
        }

        return $mock;
    }
}