MockObjectExprDetector.php

<?php

declare(strict_types=1);

namespace Rector\PHPUnit\CodeQuality\NodeAnalyser;

use PhpParser\Node\Expr;
use PhpParser\Node\Expr\Array_;
use PhpParser\Node\Expr\CallLike;
use PhpParser\Node\Expr\MethodCall;
use PhpParser\Node\Expr\PropertyFetch;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Stmt\Class_;
use PhpParser\Node\Stmt\ClassMethod;
use Rector\NodeNameResolver\NodeNameResolver;
use Rector\PhpParser\Node\BetterNodeFinder;
use Rector\PHPUnit\CodeQuality\NodeFinder\VariableFinder;

final readonly class MockObjectExprDetector
{
    public function __construct(
        private BetterNodeFinder $betterNodeFinder,
        private NodeNameResolver $nodeNameResolver,
        private VariableFinder $variableFinder,
    ) {
    }

    public function hasMethodCallWithoutExpects(ClassMethod $classMethod): bool
    {
        /** @var array<Expr\MethodCall> $methodCalls */
        $methodCalls = $this->betterNodeFinder->findInstancesOfScoped((array) $classMethod->stmts, [MethodCall::class]);

        foreach ($methodCalls as $methodCall) {
            if (! $this->nodeNameResolver->isName($methodCall->name, 'method')) {
                continue;
            }

            if ($methodCall->var instanceof MethodCall) {
                continue;
            }

            return true;
        }

        return false;
    }

    public function isUsedForMocking(Expr $expr, ClassMethod $classMethod): bool
    {
        if (! $expr instanceof Variable) {
            return false;
        }

        $variableName = $this->nodeNameResolver->getName($expr);

        // to be safe
        if ($variableName === null) {
            return true;
        }

        $relatedVariables = $this->variableFinder->find($classMethod, $variableName);

        // only self variable found, nothing to mock
        if (count($relatedVariables) === 1) {
            return false;
        }

        // find out, how many are used in call likes as args
        /** @var array<Expr\MethodCall> $methodCalls */
        $methodCalls = $this->betterNodeFinder->findInstancesOfScoped((array) $classMethod->stmts, [MethodCall::class]);

        foreach ($methodCalls as $methodCall) {
            if (! $methodCall->var instanceof Variable) {
                continue;
            }

            if ($this->nodeNameResolver->isName($methodCall->var, $variableName)) {
                // variable is being called on, most like mocking, lets skip
                return true;
            }
        }

        return false;
    }

    public function isPropertyUsedForMocking(Class_ $class, string $propertyName): bool
    {
        // find out, how many are used in call likes as args
        /** @var array<Expr\MethodCall> $methodCalls */
        $methodCalls = $this->betterNodeFinder->findInstancesOfScoped($class->getMethods(), [MethodCall::class]);

        foreach ($methodCalls as $methodCall) {
            if (! $methodCall->var instanceof PropertyFetch) {
                continue;
            }

            $propertyFetch = $methodCall->var;
            if ($this->nodeNameResolver->isName($propertyFetch->name, $propertyName)) {
                // variable is being called on, most like mocking, lets skip
                return true;
            }
        }

        return false;
    }

    public function isPropertyMockObjectPassedAsArgument(Class_ $class, string $propertyName): bool
    {
        /** @var array<Expr\CallLike> $callLikes */
        $callLikes = $this->betterNodeFinder->findInstancesOfScoped($class->getMethods(), [CallLike::class]);

        foreach ($callLikes as $callLike) {
            if ($callLike->isFirstClassCallable()) {
                continue;
            }

            foreach ($callLike->getArgs() as $arg) {
                if (! $arg->value instanceof PropertyFetch) {
                    continue;
                }

                $propertyFetch = $arg->value;

                // its used in arg
                if ($this->nodeNameResolver->isName($propertyFetch->name, $propertyName)) {
                    return true;
                }
            }
        }

        /** @var array<Array_> $arrays */
        $arrays = $this->betterNodeFinder->findInstancesOfScoped($class->getMethods(), [Array_::class]);
        foreach ($arrays as $array) {
            foreach ($array->items as $arrayItem) {
                if (! $arrayItem->value instanceof PropertyFetch) {
                    continue;
                }

                $propertyFetch = $arrayItem->value;
                if ($this->nodeNameResolver->isName($propertyFetch->name, $propertyName)) {
                    return true;
                }
            }
        }

        return false;
    }
}