add fixtures to handle

Author: TomasVotruba

rules-tests/PHPUnit120/Rector/Class_/AllowMockObjectsWithoutExpectationsAttributeRector/Fixture/skip_if_all_tests_methods_define_expectations.php.inc

@@ -0,0 +1,25 @@
+<?php
+
+namespace Rector\PHPUnit\Tests\PHPUnit120\Rector\Class_\AllowMockObjectsWithoutExpectationsAttributeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class SkipIfAllTestsMethodsDefineExpectations extends TestCase
+{
+    private \PHPUnit\Framework\MockObject\MockObject $someMock;
+
+    protected function setUp(): void
+    {
+        $this->someMock = $this->createMock(\stdClass::class);
+    }
+
+    public function testOne()
+    {
+        $this->someMock->method('doSomething')->willReturn('value');
+    }
+
+    public function testTwo()
+    {
+        $this->someMock->method('doSomethingElse')->willReturn('another value');
+    }
+}

rules-tests/PHPUnit120/Rector/Class_/AllowMockObjectsWithoutExpectationsAttributeRector/Fixture/skip_if_mock_not_used_in_2_test_methods.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Rector\PHPUnit\Tests\PHPUnit120\Rector\Class_\AllowMockObjectsWithoutExpectationsAttributeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class SkipIfMockNotUsedIn2TestMethods extends TestCase
+{
+    private \PHPUnit\Framework\MockObject\MockObject $someMock;
+
+    protected function setUp(): void
+    {
+        $this->someMock = $this->createMock(\stdClass::class);
+    }
+
+    public function testOne()
+    {
+    }
+
+    public function testTwo()
+    {
+
+    }
+}

rules-tests/PHPUnit120/Rector/Class_/AllowMockObjectsWithoutExpectationsAttributeRector/Fixture/some_class.php.inc

@@ -15,11 +15,11 @@ final class SomeClass extends TestCase
 
     public function testOne()
     {
+        $this->someMock->method('doSomething')->willReturn('value');
     }
 
     public function testTwo()
     {
-
     }
 }
 
@@ -43,11 +43,11 @@ final class SomeClass extends TestCase
 
     public function testOne()
     {
+        $this->someMock->method('doSomething')->willReturn('value');
     }
 
     public function testTwo()
     {
-
     }
 }
 

rules/PHPUnit120/Rector/Class_/AllowMockObjectsWithoutExpectationsAttributeRector.php

@@ -17,6 +17,7 @@
 use Rector\PHPUnit\Enum\PHPUnitClassName;
 use Rector\PHPUnit\NodeAnalyzer\TestsNodeAnalyzer;
 use Rector\Rector\AbstractRector;
+use Rector\ValueObject\MethodName;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
 use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
 
@@ -44,30 +45,14 @@ public function getNodeTypes(): array
      */
     public function refactor(Node $node): ?Class_
     {
-        if (! $this->testsNodeAnalyzer->isInTestClass($node)) {
+        if ($this->shouldSkipClass($node)) {
             return null;
         }
 
-        // attribute must exist for the rule to work
-        if (! $this->reflectionProvider->hasClass(PHPUnitAttribute::ALLOW_MOCK_OBJECTS_WITHOUT_EXPECTATIONS)) {
-            return null;
-        }
-
-        // already filled
-        if ($this->attributeFinder->hasAttributeByClasses(
-            $node,
-            [PHPUnitAttribute::ALLOW_MOCK_OBJECTS_WITHOUT_EXPECTATIONS]
-        )) {
-            return null;
-        }
-
-        // has mock objects properties and setUp() method?
-
-        if (! $node->getMethod('setUp') instanceof ClassMethod) {
-            return null;
-        }
+        $mockObjectPropertyNames = $this->matchMockObjectPropertyNames($node);
 
-        if (! $this->hasMockObjectProperty($node)) {
+        // there are no mock object properties
+        if ($mockObjectPropertyNames === []) {
             return null;
         }
 
@@ -76,6 +61,9 @@ public function refactor(Node $node): ?Class_
 
         foreach ($node->getMethods() as $classMethod) {
             if ($this->testsNodeAnalyzer->isTestClassMethod($classMethod)) {
+                // is a mock property used in the method?
+                // skip if so
+
                 ++$testMethodCount;
             }
         }
@@ -153,18 +141,50 @@ public function testTwo(): void
 
     }
 
-    private function hasMockObjectProperty(Class_ $class): bool
+    /**
+     * @return string[]
+     */
+    private function matchMockObjectPropertyNames(Class_ $class): array
     {
+        $propertyNames = [];
+
         foreach ($class->getProperties() as $property) {
             if (! $property->type instanceof Name) {
                 continue;
             }
 
-            if ($this->isName($property->type, PHPUnitClassName::MOCK_OBJECT)) {
-                return true;
+            if (! $this->isName($property->type, PHPUnitClassName::MOCK_OBJECT)) {
+                continue;
             }
+
+            $propertyNames[] = $this->getName($property->props[0]);
+        }
+
+        return $propertyNames;
+    }
+
+    private function shouldSkipClass(Class_ $class): bool
+    {
+        if (! $this->testsNodeAnalyzer->isInTestClass($class)) {
+            return true;
         }
 
-        return false;
+        // attribute must exist for the rule to work
+        if (! $this->reflectionProvider->hasClass(PHPUnitAttribute::ALLOW_MOCK_OBJECTS_WITHOUT_EXPECTATIONS)) {
+            return true;
+        }
+
+        // already filled
+        if ($this->attributeFinder->hasAttributeByClasses(
+            $class,
+            [PHPUnitAttribute::ALLOW_MOCK_OBJECTS_WITHOUT_EXPECTATIONS]
+        )) {
+            return true;
+        }
+
+        // has mock objects properties and setUp() method?
+
+        $setupClassMethod = $class->getMethod(MethodName::SET_UP);
+        return ! $setupClassMethod instanceof ClassMethod;
     }
 }