Add support for interfaces

Author: akondas

spec/Proget/Tests/PHPStan/PhpSpec/FooSpec.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace spec\Proget\Tests\PHPStan\PhpSpec;
+
+use Proget\Tests\PHPStan\PhpSpec\Baz;
+use Proget\Tests\PHPStan\PhpSpec\Foo;
+use PhpSpec\ObjectBehavior;
+
+class FooSpec extends ObjectBehavior
+{
+    public function let(Baz $baz): void
+    {
+        $this->beConstructedWith($baz);
+    }
+
+    public function it_is_initializable(): void
+    {
+        $this->shouldHaveType(Foo::class);
+    }
+
+    public function it_should_make_baz(Baz $baz): void
+    {
+        $baz->make()->willReturn(123);
+
+        $this->makeBaz()->shouldBe(123);
+    }
+}

src/DependencyInjection/PhpSpecDynamicMethodReturnTypeExtensionCreator.php

@@ -6,7 +6,7 @@
 
 use Nette\DI\CompilerExtension;
 use Nette\DI\ServiceDefinition;
-use Proget\PHPStan\PhpSpec\Locator\NonFinalClassLocator;
+use Proget\PHPStan\PhpSpec\Locator\SourceClassLocator;
 use Proget\PHPStan\PhpSpec\Reflection\CollaboratorDynamicMethodReturnTypeExtension;
 
 final class PhpSpecDynamicMethodReturnTypeExtensionCreator extends CompilerExtension
@@ -15,7 +15,7 @@ public function beforeCompile()
     {
         $builder = $this->getContainerBuilder();
         $workingDir = $this->getContainerBuilder()->parameters['currentWorkingDirectory'];
-        $classes = (new NonFinalClassLocator())->locate(array_map(function (string $dir) use ($workingDir) {
+        $classes = (new SourceClassLocator())->locate(array_map(function (string $dir) use ($workingDir) {
             return $workingDir.DIRECTORY_SEPARATOR.ltrim($dir, DIRECTORY_SEPARATOR);
         }, $this->getContainerBuilder()->parameters['phpspecSourceFiles']));
 

src/Locator/NonFinalClassLocator.php src/Locator/SourceClassLocator.phpsrc/Locator/NonFinalClassLocator.php renamed to src/Locator/SourceClassLocator.php

@@ -7,7 +7,7 @@
 use Symfony\Component\Finder\Finder;
 use Symfony\Component\Finder\SplFileInfo;
 
-final class NonFinalClassLocator
+final class SourceClassLocator
 {
     public function locate(array $dirs): array
     {
@@ -44,7 +44,7 @@ private function getClassName(SplFileInfo $fileInfo): ?string
                 }
             }
 
-            if ($tokens[$i][0] === T_CLASS) {
+            if ($tokens[$i][0] === T_CLASS || $tokens[$i][0] === T_INTERFACE) {
                 for ($j = $i + 1; $j < $count; ++$j) {
                     if ($tokens[$j] === '{') {
                         return $namespace.$tokens[$i + 2][1];

tests/Baz.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Proget\Tests\PHPStan\PhpSpec;
+
+interface Baz
+{
+    public function make(): int;
+}

tests/Foo.php

@@ -6,8 +6,23 @@
 
 class Foo
 {
+    /**
+     * @var Baz
+     */
+    private $baz;
+
+    public function __construct(Baz $baz)
+    {
+        $this->baz = $baz;
+    }
+
     public function foo(): string
     {
         return '';
     }
+
+    public function makeBaz(): int
+    {
+        return $this->baz->make();
+    }
 }