diff --git a/rules-tests/TypeDeclarationDocblocks/Rector/Class_/AddReturnDocblockDataProviderRector/Fixture/codeception_data_provider_method.php.inc b/rules-tests/TypeDeclarationDocblocks/Rector/Class_/AddReturnDocblockDataProviderRector/Fixture/codeception_data_provider_method.php.inc new file mode 100644 index 00000000000..ede4f21e82c --- /dev/null +++ b/rules-tests/TypeDeclarationDocblocks/Rector/Class_/AddReturnDocblockDataProviderRector/Fixture/codeception_data_provider_method.php.inc @@ -0,0 +1,52 @@ + +----- + diff --git a/rules/TypeDeclarationDocblocks/Enum/TestClassName.php b/rules/TypeDeclarationDocblocks/Enum/TestClassName.php index 8229420985e..9cf2f9b2247 100644 --- a/rules/TypeDeclarationDocblocks/Enum/TestClassName.php +++ b/rules/TypeDeclarationDocblocks/Enum/TestClassName.php @@ -6,5 +6,7 @@ final class TestClassName { - public const string DATA_PROVIDER = 'PHPUnit\Framework\Attributes\DataProvider'; + public const string PHPUNIT_DATA_PROVIDER = 'PHPUnit\Framework\Attributes\DataProvider'; + + public const string CODECEPTION_DATA_PROVIDER = 'Codeception\Attribute\DataProvider'; } diff --git a/rules/TypeDeclarationDocblocks/NodeFinder/DataProviderMethodsFinder.php b/rules/TypeDeclarationDocblocks/NodeFinder/DataProviderMethodsFinder.php index bad7624d35e..667a60ed010 100644 --- a/rules/TypeDeclarationDocblocks/NodeFinder/DataProviderMethodsFinder.php +++ b/rules/TypeDeclarationDocblocks/NodeFinder/DataProviderMethodsFinder.php @@ -15,6 +15,11 @@ final readonly class DataProviderMethodsFinder { + private const array DATA_PROVIDER_ATTRIBUTES = [ + TestClassName::PHPUNIT_DATA_PROVIDER, + TestClassName::CODECEPTION_DATA_PROVIDER, + ]; + public function __construct( private PhpDocInfoFactory $phpDocInfoFactory, private NodeNameResolver $nodeNameResolver @@ -62,7 +67,7 @@ private function findDataProviderAttributes(ClassMethod $classMethod): array foreach ($classMethod->attrGroups as $attrGroup) { foreach ($attrGroup->attrs as $attribute) { - if (! $this->nodeNameResolver->isName($attribute->name, TestClassName::DATA_PROVIDER)) { + if (! $this->nodeNameResolver->isNames($attribute->name, self::DATA_PROVIDER_ATTRIBUTES)) { continue; }