Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions config/sets/phpunit-code-quality.php
Original file line number Diff line number Diff line change
Expand Up @@ -125,6 +125,9 @@
SimplerWithIsInstanceOfRector::class,
DirectInstanceOverMockArgRector::class,

// @test first, enable later
// \Rector\PHPUnit\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector::class,

FinalizeTestCaseClassRector::class,
DeclareStrictTypesTestsRector::class,
WithCallbackIdenticalToStandaloneAssertsRector::class,
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
<?php

declare(strict_types=1);

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector;

use Iterator;
use PHPUnit\Framework\Attributes\DataProvider;
use Rector\Testing\PHPUnit\AbstractRectorTestCase;

final class ConfiguredMockEntityToSetterObjectRectorTest extends AbstractRectorTestCase
{
#[DataProvider('provideData')]
public function test(string $filePath): void
{
$this->doTestFile($filePath);
}

public static function provideData(): Iterator
{
return self::yieldFilesFromDirectory(__DIR__ . '/Fixture');
}

public function provideConfigFilePath(): string
{
return __DIR__ . '/config/configured_rule.php';
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
<?php

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Fixture;

use Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured;

final class SomeTest extends \PHPUnit\Framework\TestCase
{
public function test()
{
$mockObject = $this->createConfiguredMock(SomeEntityToBeConfigured::class, [
'getName'=> 'John',
]);
}
}

?>
-----
<?php

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Fixture;

use Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured;

final class SomeTest extends \PHPUnit\Framework\TestCase
{
public function test()
{
$mockObject = new \Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured();
$mockObject->setName('John');
}
}

?>
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
<?php

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Fixture;

use Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeObjectToBeConfigured;

final class SkipNonEntity extends \PHPUnit\Framework\TestCase
{
public function test()
{
$mockObject = $this->createConfiguredMock(SomeObjectToBeConfigured::class, [
'getName'=> 'John',
]);
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
<?php

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source;

use Doctrine\ORM\Mapping as ORM;

#[ORM\Entity()]
class SomeEntityToBeConfigured
{
private string $name = '';

public function setName($name)
{
$this->name = $name;
}

public function getName(): string
{
return $this->name;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
<?php

namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source;

final class SomeObjectToBeConfigured
{
private string $name = '';

public function setName($name)
{
$this->name = $name;
}

public function getName(): string
{
return $this->name;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
<?php

declare(strict_types=1);

use Rector\Config\RectorConfig;
use Rector\PHPUnit\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector;

return static function (RectorConfig $rectorConfig): void {
$rectorConfig->rule(ConfiguredMockEntityToSetterObjectRector::class);
};
Original file line number Diff line number Diff line change
@@ -0,0 +1,189 @@
<?php

declare(strict_types=1);

namespace Rector\PHPUnit\CodeQuality\Rector\Expression;

use PhpParser\Node\Stmt\Class_;
use PhpParser\Node;
use PhpParser\Node\Arg;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\Array_;
use PhpParser\Node\Expr\Assign;
use PhpParser\Node\Expr\MethodCall;
use PhpParser\Node\Expr\New_;
use PhpParser\Node\Name\FullyQualified;
use PhpParser\Node\Stmt\Expression;
use PHPStan\Reflection\ReflectionProvider;
use Rector\Doctrine\NodeAnalyzer\DoctrineEntityDetector;
use Rector\PhpParser\AstResolver;
use Rector\PhpParser\Node\Value\ValueResolver;
use Rector\PHPUnit\NodeAnalyzer\TestsNodeAnalyzer;
use Rector\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;

/**
* @see \Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\ConfiguredMockEntityToSetterObjectRectorTest
*/
final class ConfiguredMockEntityToSetterObjectRector extends AbstractRector
{
public function __construct(
private readonly ReflectionProvider $reflectionProvider,
private readonly TestsNodeAnalyzer $testsNodeAnalyzer,
private readonly ValueResolver $valueResolver,
private readonly AstResolver $astResolver,
private readonly DoctrineEntityDetector $doctrineEntityDetector,
) {
}

public function getRuleDefinition(): RuleDefinition
{
return new RuleDefinition(
'Change createConfigureMock() on Entity/Document object to direct new instance with setters',
[
new CodeSample(
<<<'CODE_SAMPLE'
use PHPUnit\Framework\TestCase;

final class SomeTest extends TestClass
{
public function test()
{
$someObject = $this->createConfiguredMock(SomeObject::class, [
'name' => 'John',
'surname' => 'Doe',
]);
}
}
CODE_SAMPLE
,
<<<'CODE_SAMPLE'
use PHPUnit\Framework\TestCase;

final class SomeTest extends TestClass
{
public function test()
{
$someObject = new SomeObject();
$someObject->setName('John');
$someObject->setSurname('Doe');
}
}
CODE_SAMPLE
),
]
);
}

/**
* @return array<class-string<Node>>
*/
public function getNodeTypes(): array
{
return [Expression::class];
}

/**
* @param Expression $node
* @return Expression[]|null
*/
public function refactor(Node $node): ?array
{
if (! $this->testsNodeAnalyzer->isInTestClass($node)) {
return null;
}

if (! $node->expr instanceof Assign) {
return null;
}

$assign = $node->expr;
if (! $assign->expr instanceof MethodCall) {
return null;
}

$objectVariable = $assign->var;

$methodCall = $assign->expr;
if (! $this->isName($methodCall->name, 'createConfiguredMock')) {
return null;
}

if ($methodCall->isFirstClassCallable()) {
return null;
}

$mockedClassArg = $methodCall->getArgs()[0];

$doctrineClass = $this->matchDoctrineClassName($mockedClassArg->value);
if (! is_string($doctrineClass)) {
return null;
}

$definedGettersArg = $methodCall->getArgs()[1];
if (! $definedGettersArg->value instanceof Array_) {
return null;
}

$assign->expr = new New_(new FullyQualified($doctrineClass));

$setterExpressions = $this->createEntitySetterExpressions($definedGettersArg->value, $objectVariable);

return array_merge([$node], $setterExpressions);
}

/**
* @return Expression[]
*/
private function createEntitySetterExpressions(Array_ $definedGettersArray, Expr $expr): array
{
$setterExpressions = [];

foreach ($definedGettersArray->items as $arrayItem) {
if (! $arrayItem->key instanceof Expr) {
continue;
}

$getterName = $this->valueResolver->getValue($arrayItem->key);
if (! is_string($getterName)) {
continue;
}

// remove "get" prefix
if (! str_starts_with($getterName, 'get')) {
continue;
}

$setterName = 'set' . substr($getterName, 3);

$setterMethodCall = new MethodCall($expr, $setterName, [new Arg($arrayItem->value)]);
$setterExpressions[] = new Expression($setterMethodCall);
}

return $setterExpressions;
}

private function matchDoctrineClassName(Expr $expr): string|null
{
$mockedClassValue = $this->valueResolver->getValue($expr);
if (! is_string($mockedClassValue)) {
return null;
}

if (! $this->reflectionProvider->hasClass($mockedClassValue)) {
return null;
}

$mockedClass = $this->astResolver->resolveClassFromName($mockedClassValue);
if (! $mockedClass instanceof Class_) {
return null;
}

if (! $this->doctrineEntityDetector->detect($mockedClass)) {
return null;
}

return $mockedClassValue;
}
}
12 changes: 12 additions & 0 deletions stubs/Doctrine/ORM/Mapping/Entity.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
<?php

namespace Doctrine\ORM\Mapping;

/**
* @Annotation
*/
#[\Attribute]
class Entity
{

}