Skip to content

Commit cad3112

Browse files
committed
[code-quality] Add ConfiguredMockEntityToSetterObjectRector
1 parent 887abad commit cad3112

File tree

5 files changed

+247
-0
lines changed

5 files changed

+247
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
<?php
2+
3+
declare(strict_types=1);
4+
5+
namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector;
6+
7+
use Iterator;
8+
use PHPUnit\Framework\Attributes\DataProvider;
9+
use Rector\Testing\PHPUnit\AbstractRectorTestCase;
10+
11+
final class ConfiguredMockEntityToSetterObjectRectorTest extends AbstractRectorTestCase
12+
{
13+
#[DataProvider('provideData')]
14+
public function test(string $filePath): void
15+
{
16+
$this->doTestFile($filePath);
17+
}
18+
19+
public static function provideData(): Iterator
20+
{
21+
return self::yieldFilesFromDirectory(__DIR__ . '/Fixture');
22+
}
23+
24+
public function provideConfigFilePath(): string
25+
{
26+
return __DIR__ . '/config/configured_rule.php';
27+
}
28+
}
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
<?php
2+
3+
namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Fixture;
4+
5+
use Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured;
6+
7+
final class SomeTest extends \PHPUnit\Framework\TestCase
8+
{
9+
public function test()
10+
{
11+
$mockObject = $this->createConfiguredMock(SomeEntityToBeConfigured::class, [
12+
'getName'=> 'John',
13+
]);
14+
}
15+
}
16+
17+
?>
18+
-----
19+
<?php
20+
21+
namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Fixture;
22+
23+
use Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured;
24+
25+
final class SomeTest extends \PHPUnit\Framework\TestCase
26+
{
27+
public function test()
28+
{
29+
$mockObject = new \Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source\SomeEntityToBeConfigured();
30+
$mockObject->setName('John');
31+
}
32+
}
33+
34+
?>
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
<?php
2+
3+
namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\Source;
4+
5+
final class SomeEntityToBeConfigured
6+
{
7+
private string $name = '';
8+
9+
public function setName($name)
10+
{
11+
$this->name = $name;
12+
}
13+
14+
public function getName(): string
15+
{
16+
return $this->name;
17+
}
18+
}
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
<?php
2+
3+
declare(strict_types=1);
4+
5+
use Rector\Config\RectorConfig;
6+
use Rector\PHPUnit\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector;
7+
8+
return static function (RectorConfig $rectorConfig): void {
9+
$rectorConfig->rule(ConfiguredMockEntityToSetterObjectRector::class);
10+
};
Lines changed: 157 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,157 @@
1+
<?php
2+
3+
declare(strict_types=1);
4+
5+
namespace Rector\PHPUnit\CodeQuality\Rector\Expression;
6+
7+
use PhpParser\Node;
8+
use PhpParser\Node\Arg;
9+
use PhpParser\Node\Expr;
10+
use PhpParser\Node\Expr\Array_;
11+
use PhpParser\Node\Expr\Assign;
12+
use PhpParser\Node\Expr\MethodCall;
13+
use PhpParser\Node\Expr\New_;
14+
use PhpParser\Node\Name\FullyQualified;
15+
use PhpParser\Node\Stmt\Expression;
16+
use PHPStan\Reflection\ReflectionProvider;
17+
use Rector\PhpParser\Node\Value\ValueResolver;
18+
use Rector\PHPUnit\NodeAnalyzer\TestsNodeAnalyzer;
19+
use Rector\Rector\AbstractRector;
20+
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
21+
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
22+
23+
/**
24+
* @see \Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\ConfiguredMockEntityToSetterObjectRector\ConfiguredMockEntityToSetterObjectRectorTest
25+
*/
26+
final class ConfiguredMockEntityToSetterObjectRector extends AbstractRector
27+
{
28+
public function __construct(
29+
private readonly ReflectionProvider $reflectionProvider,
30+
private readonly TestsNodeAnalyzer $testsNodeAnalyzer,
31+
private readonly ValueResolver $valueResolver,
32+
) {
33+
}
34+
35+
public function getRuleDefinition(): RuleDefinition
36+
{
37+
return new RuleDefinition(
38+
'Change createConfigureMock() on Entity/Document object to direct new instance with setters',
39+
[
40+
new CodeSample(
41+
<<<'CODE_SAMPLE'
42+
use PHPUnit\Framework\TestCase;
43+
44+
final class SomeTest extends TestClass
45+
{
46+
public function test()
47+
{
48+
$someObject = $this->createConfiguredMock(SomeObject::class, [
49+
'name' => 'John',
50+
'surname' => 'Doe',
51+
]);
52+
}
53+
}
54+
CODE_SAMPLE
55+
,
56+
<<<'CODE_SAMPLE'
57+
use PHPUnit\Framework\TestCase;
58+
59+
final class SomeTest extends TestClass
60+
{
61+
public function test()
62+
{
63+
$someObject = new SomeObject();
64+
$someObject->setName('John');
65+
$someObject->setSurname('Doe');
66+
}
67+
}
68+
CODE_SAMPLE
69+
),
70+
]
71+
);
72+
}
73+
74+
/**
75+
* @return array<class-string<Node>>
76+
*/
77+
public function getNodeTypes(): array
78+
{
79+
return [Expression::class];
80+
}
81+
82+
/**
83+
* @param Expression $node
84+
* @return Expression[]|null
85+
*/
86+
public function refactor(Node $node): ?array
87+
{
88+
if (! $this->testsNodeAnalyzer->isInTestClass($node)) {
89+
return null;
90+
}
91+
92+
if (! $node->expr instanceof Assign) {
93+
return null;
94+
}
95+
96+
$assign = $node->expr;
97+
if (! $assign->expr instanceof MethodCall) {
98+
return null;
99+
}
100+
101+
$objectVariable = $assign->var;
102+
103+
$methodCall = $assign->expr;
104+
if (! $this->isName($methodCall->name, 'createConfiguredMock')) {
105+
return null;
106+
}
107+
108+
if ($methodCall->isFirstClassCallable()) {
109+
return null;
110+
}
111+
112+
$mockedClassArg = $methodCall->getArgs()[0];
113+
$mockedClassValue = $this->valueResolver->getValue($mockedClassArg->value);
114+
115+
if (! is_string($mockedClassValue)) {
116+
return null;
117+
}
118+
119+
if (! $this->reflectionProvider->hasClass($mockedClassValue)) {
120+
return null;
121+
}
122+
123+
$definedGettersArg = $methodCall->getArgs()[1];
124+
if (! $definedGettersArg->value instanceof Array_) {
125+
return null;
126+
}
127+
128+
$definedGettersArray = $definedGettersArg->value;
129+
130+
$standaloneExpressions = [$node];
131+
132+
$assign->expr = new New_(new FullyQualified($mockedClassValue));
133+
134+
foreach ($definedGettersArray->items as $arrayItem) {
135+
if (! $arrayItem->key instanceof Expr) {
136+
continue;
137+
}
138+
139+
$getterName = $this->valueResolver->getValue($arrayItem->key);
140+
if (! is_string($getterName)) {
141+
continue;
142+
}
143+
144+
// remove "get" prefix
145+
if (! str_starts_with($getterName, 'get')) {
146+
continue;
147+
}
148+
149+
$setterName = 'set' . substr($getterName, 3);
150+
151+
$setterMethodCall = new MethodCall($objectVariable, $setterName, [new Arg($arrayItem->value)]);
152+
$standaloneExpressions[] = new Expression($setterMethodCall);
153+
}
154+
155+
return $standaloneExpressions;
156+
}
157+
}

0 commit comments

Comments
 (0)