| 
5 | 5 | use PhpParser\Node;  | 
6 | 6 | use PhpParser\Node\Expr\MethodCall;  | 
7 | 7 | use PHPStan\Analyser\Scope;  | 
 | 8 | +use PHPStan\Rules\IdentifierRuleError;  | 
8 | 9 | use PHPStan\Rules\Rule;  | 
9 | 10 | use PHPStan\Rules\RuleErrorBuilder;  | 
10 |  | -use PHPUnit\Framework\MockObject\Builder\InvocationMocker;  | 
 | 11 | +use PHPStan\Type\Type;  | 
11 | 12 | use PHPUnit\Framework\MockObject\MockObject;  | 
12 | 13 | use PHPUnit\Framework\MockObject\Stub;  | 
13 | 14 | use function array_filter;  | 
@@ -47,44 +48,58 @@ public function processNode(Node $node, Scope $scope): array  | 
47 | 48 | 			$method = $constantString->getValue();  | 
48 | 49 | 			$type = $scope->getType($node->var);  | 
49 | 50 | 
 
  | 
50 |  | -			if (  | 
51 |  | -				(  | 
52 |  | -					in_array(MockObject::class, $type->getObjectClassNames(), true)  | 
53 |  | -					|| in_array(Stub::class, $type->getObjectClassNames(), true)  | 
54 |  | -				)  | 
55 |  | -				&& !$type->hasMethod($method)->yes()  | 
56 |  | -			) {  | 
57 |  | -				$mockClasses = array_filter($type->getObjectClassNames(), static fn (string $class): bool => $class !== MockObject::class && $class !== Stub::class);  | 
58 |  | -				if (count($mockClasses) === 0) {  | 
59 |  | -					continue;  | 
60 |  | -				}  | 
61 |  | - | 
62 |  | -				$errors[] = RuleErrorBuilder::message(sprintf(  | 
63 |  | -					'Trying to mock an undefined method %s() on class %s.',  | 
64 |  | -					$method,  | 
65 |  | -					implode('&', $mockClasses),  | 
66 |  | -				))->identifier('phpunit.mockMethod')->build();  | 
 | 51 | +			$error = $this->checkCallOnType($type, $method);  | 
 | 52 | +			if ($error !== null) {  | 
 | 53 | +				$errors[] = $error;  | 
67 | 54 | 				continue;  | 
68 | 55 | 			}  | 
69 | 56 | 
 
  | 
70 |  | -			$mockedClassObject = $type->getTemplateType(InvocationMocker::class, 'TMockedClass');  | 
71 |  | -			if ($mockedClassObject->hasMethod($method)->yes()) {  | 
 | 57 | +			if (!$node->var instanceof MethodCall) {  | 
72 | 58 | 				continue;  | 
73 | 59 | 			}  | 
74 | 60 | 
 
  | 
75 |  | -			$classNames = $mockedClassObject->getObjectClassNames();  | 
76 |  | -			if (count($classNames) === 0) {  | 
 | 61 | +			if (!$node->var->name instanceof Node\Identifier) {  | 
77 | 62 | 				continue;  | 
78 | 63 | 			}  | 
79 | 64 | 
 
  | 
80 |  | -			$errors[] = RuleErrorBuilder::message(sprintf(  | 
 | 65 | +			if ($node->var->name->toLowerString() !== 'expects') {  | 
 | 66 | +				continue;  | 
 | 67 | +			}  | 
 | 68 | + | 
 | 69 | +			$varType = $scope->getType($node->var->var);  | 
 | 70 | +			$error = $this->checkCallOnType($varType, $method);  | 
 | 71 | +			if ($error === null) {  | 
 | 72 | +				continue;  | 
 | 73 | +			}  | 
 | 74 | + | 
 | 75 | +			$errors[] = $error;  | 
 | 76 | +		}  | 
 | 77 | + | 
 | 78 | +		return $errors;  | 
 | 79 | +	}  | 
 | 80 | + | 
 | 81 | +	private function checkCallOnType(Type $type, string $method): ?IdentifierRuleError  | 
 | 82 | +	{  | 
 | 83 | +		if (  | 
 | 84 | +			(  | 
 | 85 | +				in_array(MockObject::class, $type->getObjectClassNames(), true)  | 
 | 86 | +				|| in_array(Stub::class, $type->getObjectClassNames(), true)  | 
 | 87 | +			)  | 
 | 88 | +			&& !$type->hasMethod($method)->yes()  | 
 | 89 | +		) {  | 
 | 90 | +			$mockClasses = array_filter($type->getObjectClassNames(), static fn (string $class): bool => $class !== MockObject::class && $class !== Stub::class);  | 
 | 91 | +			if (count($mockClasses) === 0) {  | 
 | 92 | +				return null;  | 
 | 93 | +			}  | 
 | 94 | + | 
 | 95 | +			return RuleErrorBuilder::message(sprintf(  | 
81 | 96 | 				'Trying to mock an undefined method %s() on class %s.',  | 
82 | 97 | 				$method,  | 
83 |  | -				implode('|', $classNames),  | 
 | 98 | +				implode('&', $mockClasses),  | 
84 | 99 | 			))->identifier('phpunit.mockMethod')->build();  | 
85 | 100 | 		}  | 
86 | 101 | 
 
  | 
87 |  | -		return $errors;  | 
 | 102 | +		return null;  | 
88 | 103 | 	}  | 
89 | 104 | 
 
  | 
90 | 105 | }  | 
0 commit comments