mirror of
https://github.com/danog/psalm-plugin-phpunit.git
synced 2024-12-02 09:27:56 +01:00
342 lines
14 KiB
PHP
342 lines
14 KiB
PHP
<?php declare(strict_types=1);
|
|
namespace Psalm\PhpUnitPlugin\Hooks;
|
|
|
|
use PHPUnit\Framework\TestCase;
|
|
use PhpParser\Node\Stmt\ClassLike;
|
|
use PhpParser\Node\Stmt\ClassMethod;
|
|
use Psalm\Codebase;
|
|
use Psalm\DocComment;
|
|
use Psalm\FileSource;
|
|
use Psalm\IssueBuffer;
|
|
use Psalm\Issue;
|
|
use Psalm\PhpUnitPlugin\Exception\UnsupportedPsalmVersion;
|
|
use Psalm\Plugin\Hook\AfterClassLikeAnalysisInterface;
|
|
use Psalm\Plugin\Hook\AfterClassLikeVisitInterface;
|
|
use Psalm\StatementsSource;
|
|
use Psalm\Storage\ClassLikeStorage;
|
|
use Psalm\Storage\FunctionLikeParameter;
|
|
use Psalm\Storage\MethodStorage;
|
|
use Psalm\Type;
|
|
|
|
class TestCaseHandler implements AfterClassLikeVisitInterface, AfterClassLikeAnalysisInterface
|
|
{
|
|
/**
|
|
* {@inheritDoc}
|
|
*/
|
|
public static function afterClassLikeVisit(
|
|
ClassLike $classNode,
|
|
ClassLikeStorage $classStorage,
|
|
FileSource $statements_source,
|
|
Codebase $codebase,
|
|
array &$file_replacements = []
|
|
) {
|
|
if (!$codebase->classExtends($classStorage->name, TestCase::class)) {
|
|
return;
|
|
}
|
|
|
|
if (self::hasInitializers($classStorage, $classNode)) {
|
|
$classStorage->suppressed_issues[] = 'MissingConstructor';
|
|
}
|
|
}
|
|
|
|
/**
|
|
* {@inheritDoc}
|
|
*/
|
|
public static function afterStatementAnalysis(
|
|
ClassLike $class_node,
|
|
ClassLikeStorage $class_storage,
|
|
StatementsSource $statements_source,
|
|
Codebase $codebase,
|
|
array &$file_replacements = []
|
|
) {
|
|
if (!$codebase->classExtends($class_storage->name, TestCase::class)) {
|
|
return null;
|
|
}
|
|
|
|
/** @var MethodStorage $method_storage */
|
|
foreach ($class_storage->methods as $method_name => $method_storage) {
|
|
if (!$method_storage->location) {
|
|
continue;
|
|
}
|
|
|
|
$stmt_method = $class_node->getMethod($method_name);
|
|
|
|
if (!$stmt_method) {
|
|
throw new \RuntimeException('Failed to find ' . $method_name);
|
|
}
|
|
|
|
$specials = self::getSpecials($stmt_method);
|
|
|
|
$method_id = $class_storage->name . '::' . $method_storage->cased_name;
|
|
|
|
if (!isset($specials['dataProvider'])) {
|
|
continue;
|
|
}
|
|
|
|
foreach ($specials['dataProvider'] as $line => $provider) {
|
|
$provider_method_id = $class_storage->name . '::' . (string) $provider;
|
|
|
|
$provider_docblock_location = clone $method_storage->location;
|
|
$provider_docblock_location->setCommentLine($line);
|
|
|
|
if (!$codebase->methodExists($provider_method_id)) {
|
|
IssueBuffer::accepts(new Issue\UndefinedMethod(
|
|
'Provider method ' . $provider_method_id . ' is not defined',
|
|
$provider_docblock_location,
|
|
$provider_method_id
|
|
));
|
|
|
|
continue;
|
|
}
|
|
|
|
$provider_return_type = $codebase->getMethodReturnType($provider_method_id, $classStorage->name);
|
|
assert(null !== $provider_return_type);
|
|
|
|
$provider_return_type_location = $codebase->getMethodReturnTypeLocation($provider_method_id);
|
|
assert(null !== $provider_return_type_location);
|
|
|
|
$expected_provider_return_type = new Type\Atomic\TIterable([
|
|
Type::combineUnionTypes(Type::getInt(), Type::getString()),
|
|
Type::getArray(),
|
|
]);
|
|
|
|
if (!self::isTypeContainedByType(
|
|
$codebase,
|
|
$provider_return_type,
|
|
new Type\Union([$expected_provider_return_type])
|
|
)) {
|
|
IssueBuffer::accepts(new Issue\InvalidReturnType(
|
|
'Providers must return ' . $expected_provider_return_type->getId()
|
|
. ', ' . $provider_return_type->getId() . ' provided',
|
|
$provider_return_type_location
|
|
));
|
|
|
|
continue;
|
|
}
|
|
|
|
foreach ($provider_return_type->getTypes() as $type) {
|
|
if (!$type->isIterable($codebase)) {
|
|
IssueBuffer::accepts(new Issue\InvalidReturnType(
|
|
'Providers must return ' . $expected_provider_return_type->getId()
|
|
. ', ' . $provider_return_type->getId() . ' provided',
|
|
$provider_return_type_location
|
|
));
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// unionize iterable so that instead of array<int,string>|Traversable<object|int>
|
|
// we get iterable<int|object,string|int>
|
|
|
|
$provider_return_type = self::unionizeIterables($codebase, $provider_return_type);
|
|
|
|
// this is basically a repetition of above $expected_provider_return_type check
|
|
// but older Psalm versions couldn't check iterable descendants (see vimeo/psalm#1359)
|
|
if (!self::isTypeContainedByType(
|
|
$codebase,
|
|
$provider_return_type->type_params[0],
|
|
$expected_provider_return_type->type_params[0]
|
|
) || !self::isTypeContainedByType(
|
|
$codebase,
|
|
$provider_return_type->type_params[1],
|
|
$expected_provider_return_type->type_params[1]
|
|
)) {
|
|
IssueBuffer::accepts(new Issue\InvalidReturnType(
|
|
'Providers must return ' . $expected_provider_return_type->getId()
|
|
. ', ' . $provider_return_type->getId() . ' provided',
|
|
$provider_return_type_location
|
|
));
|
|
continue;
|
|
}
|
|
|
|
$checkParam =
|
|
/** @return void */
|
|
function (
|
|
Type\Union $potential_argument_type,
|
|
FunctionLikeParameter $param,
|
|
int $param_offset
|
|
) use (
|
|
$codebase,
|
|
$method_id,
|
|
$provider_method_id,
|
|
$provider_return_type,
|
|
$provider_docblock_location
|
|
) {
|
|
assert(null !== $param->type);
|
|
if (self::isTypeContainedByType($codebase, $potential_argument_type, $param->type)) {
|
|
// ok
|
|
} elseif (self::canTypeBeContainedByType($codebase, $potential_argument_type, $param->type)) {
|
|
IssueBuffer::accepts(new Issue\PossiblyInvalidArgument(
|
|
'Argument ' . ($param_offset + 1) . ' of ' . $method_id
|
|
. ' expects ' . $param->type->getId() . ', '
|
|
. $potential_argument_type->getId() . ' provided'
|
|
. ' by ' . $provider_method_id . '():(' . $provider_return_type->getId() . ')',
|
|
$provider_docblock_location
|
|
));
|
|
} else {
|
|
IssueBuffer::accepts(new Issue\InvalidArgument(
|
|
'Argument ' . ($param_offset + 1) . ' of ' . $method_id
|
|
. ' expects ' . $param->type->getId() . ', '
|
|
. $potential_argument_type->getId() . ' provided'
|
|
. ' by ' . $provider_method_id . '():(' . $provider_return_type->getId() . ')',
|
|
$provider_docblock_location
|
|
));
|
|
}
|
|
};
|
|
|
|
/** @var Type\Atomic\TArray|Type\Atomic\ObjectLike $dataset_type */
|
|
$dataset_type = $provider_return_type->type_params[1]->getTypes()['array'];
|
|
|
|
if ($dataset_type instanceof Type\Atomic\TArray) {
|
|
// check that all of the required (?) params accept value type
|
|
$potential_argument_type = $dataset_type->type_params[1];
|
|
foreach ($method_storage->params as $param_offset => $param) {
|
|
$checkParam($potential_argument_type, $param, $param_offset);
|
|
}
|
|
} else {
|
|
// iterate over all params checking if corresponding value type is acceptable
|
|
// let's hope properties are sorted in array order
|
|
$potential_argument_types = array_values($dataset_type->properties);
|
|
|
|
if (count($potential_argument_types) < $method_storage->required_param_count) {
|
|
IssueBuffer::accepts(new Issue\TooFewArguments(
|
|
'Too few arguments for ' . $method_id
|
|
. ' - expecting ' . $method_storage->required_param_count
|
|
. ' but saw ' . count($potential_argument_types)
|
|
. ' provided by ' . $provider_method_id . '()'
|
|
. ':(' . $provider_return_type->getId() . ')',
|
|
$provider_docblock_location,
|
|
$method_id
|
|
));
|
|
}
|
|
|
|
|
|
foreach ($method_storage->params as $param_offset => $param) {
|
|
if (!isset($potential_argument_types[$param_offset])) {
|
|
break;
|
|
}
|
|
$potential_argument_type = $potential_argument_types[$param_offset];
|
|
|
|
$checkParam($potential_argument_type, $param, $param_offset);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private static function isTypeContainedByType(
|
|
Codebase $codebase,
|
|
Type\Union $input_type,
|
|
Type\Union $container_type
|
|
): bool {
|
|
if (method_exists($codebase, 'isTypeContainedByType')) {
|
|
return (bool) $codebase->isTypeContainedByType($input_type, $container_type);
|
|
} elseif (class_exists(\Psalm\Internal\Analyzer\TypeAnalyzer::class, true)
|
|
&& method_exists(\Psalm\Internal\Analyzer\TypeAnalyzer::class, 'isContainedBy')) {
|
|
return \Psalm\Internal\Analyzer\TypeAnalyzer::isContainedBy($codebase, $input_type, $container_type);
|
|
} else {
|
|
throw new UnsupportedPsalmVersion();
|
|
}
|
|
}
|
|
|
|
private static function canTypeBeContainedByType(
|
|
Codebase $codebase,
|
|
Type\Union $input_type,
|
|
Type\Union $container_type
|
|
): bool {
|
|
if (method_exists($codebase, 'canTypeBeContainedByType')) {
|
|
return (bool) $codebase->canTypeBeContainedByType($input_type, $container_type);
|
|
} elseif (class_exists(\Psalm\Internal\Analyzer\TypeAnalyzer::class, true)
|
|
&& method_exists(\Psalm\Internal\Analyzer\TypeAnalyzer::class, 'canBeContainedBy')) {
|
|
return \Psalm\Internal\Analyzer\TypeAnalyzer::canBeContainedBy($codebase, $input_type, $container_type);
|
|
} else {
|
|
throw new UnsupportedPsalmVersion();
|
|
}
|
|
}
|
|
|
|
private static function unionizeIterables(Codebase $codebase, Type\Union $iterables): Type\Atomic\TIterable
|
|
{
|
|
/** @var Type\Union[] $key_types */
|
|
$key_types = [];
|
|
|
|
/** @var Type\Union[] $value_types */
|
|
$value_types = [];
|
|
|
|
foreach ($iterables->getTypes() as $type) {
|
|
if (!$type->isIterable($codebase)) {
|
|
throw new \RuntimeException('should be iterable');
|
|
}
|
|
|
|
if ($type instanceof Type\Atomic\TArray) {
|
|
$key_types[] = $type->type_params[0] ?? Type::getMixed();
|
|
$value_types[] = $type->type_params[1] ?? Type::getMixed();
|
|
} elseif ($type instanceof Type\Atomic\ObjectLike) {
|
|
$key_types[] = $type->getGenericKeyType();
|
|
$value_types[] = $type->getGenericValueType();
|
|
} elseif ($type instanceof Type\Atomic\TNamedObject || $type instanceof Type\Atomic\TIterable) {
|
|
$iterable_key_type = null;
|
|
$iterable_value_type = null;
|
|
|
|
\Psalm\Internal\Analyzer\Statements\Block\ForeachAnalyzer::getKeyValueParamsForTraversableObject(
|
|
$type,
|
|
$codebase,
|
|
$iterable_key_type,
|
|
$iterable_value_type
|
|
);
|
|
$key_types[] = $iterable_key_type ?? Type::getMixed();
|
|
$value_types[] = $iterable_value_type ?? Type::getMixed();
|
|
} else {
|
|
throw new \RuntimeException('unexpected type');
|
|
}
|
|
}
|
|
|
|
$combine = function (Type\Union $a, Type\Union $b) use ($codebase): Type\Union {
|
|
return Type::combineUnionTypes($a, $b, $codebase);
|
|
};
|
|
|
|
return new Type\Atomic\TIterable([
|
|
array_reduce($key_types, $combine, new Type\Union([])),
|
|
array_reduce($value_types, $combine, new Type\Union([]))
|
|
]);
|
|
}
|
|
|
|
|
|
private static function hasInitializers(ClassLikeStorage $storage, ClassLike $stmt): bool
|
|
{
|
|
if (isset($storage->methods['setup'])) {
|
|
return true;
|
|
}
|
|
|
|
foreach ($storage->methods as $method => $_) {
|
|
$stmt_method = $stmt->getMethod($method);
|
|
if (!$stmt_method) {
|
|
throw new \RuntimeException('Failed to find ' . $method);
|
|
}
|
|
if (self::isBeforeInitializer($stmt_method)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private static function isBeforeInitializer(ClassMethod $method): bool
|
|
{
|
|
$specials = self::getSpecials($method);
|
|
return isset($specials['before']);
|
|
}
|
|
|
|
/** @return array<string, array<int,string>> */
|
|
private static function getSpecials(ClassMethod $method): array
|
|
{
|
|
$docblock = $method->getDocComment();
|
|
|
|
if ($docblock) {
|
|
$parsed_comment = DocComment::parse((string)$docblock->getReformattedText(), $docblock->getLine());
|
|
if (isset($parsed_comment['specials'])) {
|
|
return $parsed_comment['specials'];
|
|
}
|
|
}
|
|
return [];
|
|
}
|
|
}
|