Add test case with union partials (#1147)

rules-tests/CodingStyle/ClassNameImport/ShortNameResolver/Fixture/union_partial_import.php.inc

@@ -0,0 +1,19 @@
+<?php
+
+namespace Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Fixture;
+
+use Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source;
+use Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source\FirstLog;
+use Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source\SecondLog;
+
+final class UnionPartialImport
+{
+    public function run(): Source\FirstLog|Source\SecondLog
+    {
+        if (rand(0, 1)) {
+            return new FirstLog();
+        }
+
+        return new SecondLog();
+    }
+}

rules-tests/CodingStyle/ClassNameImport/ShortNameResolver/ShortNameResolverTest.php

@@ -54,6 +54,13 @@ public function provideData(): Iterator
         yield [__DIR__ . '/Fixture/partial_names.php.inc', [
             'PartialNames' => 'Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Fixture\PartialNames',
         ]];
+
+        yield [__DIR__ . '/Fixture/union_partial_import.php.inc', [
+            'UnionPartialImport' => 'Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Fixture\UnionPartialImport',
+            'rand' => 'rand',
+            'FirstLog' => 'Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source\FirstLog',
+            'SecondLog' => 'Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source\SecondLog',
+        ]];
     }
 
     private function createFileFromFilePath(string $filePath): File

rules-tests/CodingStyle/ClassNameImport/ShortNameResolver/Source/FirstLog.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source;
+
+final class FirstLog
+{
+
+}

rules-tests/CodingStyle/ClassNameImport/ShortNameResolver/Source/SecondLog.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Tests\CodingStyle\ClassNameImport\ShortNameResolver\Source;
+
+final class SecondLog
+{
+
+}