Merge pull request #237 from brambaud/test/entity-base

Test EntityInterface return type

Author: mglaman

tests/src/Type/EntityDynamicReturnTypeTest.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace mglaman\PHPStanDrupal\Tests\Type;
+
+use mglaman\PHPStanDrupal\Tests\AdditionalConfigFilesTrait;
+use PHPStan\Testing\TypeInferenceTestCase;
+
+final class EntityDynamicReturnTypeTest extends TypeInferenceTestCase
+{
+    use AdditionalConfigFilesTrait;
+
+    public function dataFileAsserts(): iterable
+    {
+        yield from $this->gatherAssertTypes(__DIR__ . '/data/entity.php');
+    }
+
+    /**
+     * @dataProvider dataFileAsserts
+     * @param string $assertType
+     * @param string $file
+     * @param mixed ...$args
+     */
+    public function testFileAsserts(
+        string $assertType,
+        string $file,
+        ...$args
+    ): void
+    {
+        $this->assertFileAsserts($assertType, $file, ...$args);
+    }
+}

tests/src/Type/data/entity.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace DrupalEntity;
+
+use Drupal\node\Entity\Node;
+use function PHPStan\Testing\assertType;
+
+assertType('Drupal\node\Entity\Node', $node = Node::create(['type' => 'page', 'title' => 'foo']));
+assertType('Drupal\node\Entity\Node', $node->createDuplicate());
+assertType('Drupal\node\Entity\Node', $node->enforceIsNew());
+assertType('Drupal\node\Entity\Node', $node->enforceIsNew(TRUE));
+assertType('Drupal\node\Entity\Node', $node->enforceIsNew(FALSE));
+assertType('Drupal\node\Entity\Node', $node->setOriginalId(42));
+assertType('Drupal\node\Entity\Node', $node->setOriginalId('42'));
+assertType('Drupal\node\Entity\Node', $node->setOriginalId(NULL));
+assertType('Drupal\node\Entity\Node|null', Node::load(42));
+assertType('Drupal\node\Entity\Node|null', Node::load('42'));
+assertType('array<Drupal\node\Entity\Node>', Node::loadMultiple([42, 29]));
+assertType('array<Drupal\node\Entity\Node>', Node::loadMultiple(['42', '29']));
+assertType('array<Drupal\node\Entity\Node>', Node::loadMultiple(NULL));