Fwolf\Wrapper\PHPUnit\Helper\BuildEasyMockTrait

Complexity

Total Complexity

5

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	2

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	buildEasyMock()	0	26	5

<?php

namespace Fwolf\Wrapper\PHPUnit\Helper;

use PHPUnit\Framework\MockObject\Matcher\AnyInvokedCount;
use PHPUnit\Framework\MockObject\MockBuilder;
use PHPUnit\Framework\MockObject\MockObject;

/**
 * Trait for build simple mock which change method return value
 *
 * @method  MockBuilder getMockBuilder($className)
 * @method  static AnyInvokedCount  any()
 *
 * @copyright   Copyright 2015-2019 Fwolf
 * @license     http://opensource.org/licenses/MIT MIT
 */
trait BuildEasyMockTrait
{
    /**
     * @param string   $className
     * @param string[] $methods {methodName: returnValue}
     * @return  MockObject | object
     */
    public function buildEasyMock($className, array $methods = [])
    {
        $builder = $this->getMockBuilder($className)
            ->setMethods(empty($methods) ? null : array_keys($methods))
            ->disableOriginalConstructor()
        ;

        // Use class name without namespace path to detect abstract or trait
        $shortName = join('', array_slice(explode('\\', $className), -1));
        if ('Trait' == substr($shortName, -5)) {
            $mock = $builder->getMockForTrait();
        } elseif ('Abstract' == substr($shortName, 0, 8)) {
            $mock = $builder->getMockForAbstractClass();
        } else {
            $mock = $builder->getMock();
        }

        foreach ($methods as $method => $returnValue) {
            $mock->expects($this->any())
                ->method($method)
                ->willReturn($returnValue)
            ;
        }

        return $mock;
    }
}