test: mulitple type specifying extension

Author: Robin Colombier

tests/PHPStan/Analyser/MultipleTypeSpecifyingExtension.neon

@@ -0,0 +1,21 @@
+services:
+	-
+		class: PHPStan\Tests\AssertionClassMethodTypeSpecifyingExtension
+		arguments:
+			nullContext: true
+		tags:
+			- phpstan.typeSpecifier.methodTypeSpecifyingExtension
+	-
+		class: PHPStan\Tests\AssertionClassStaticMethodTypeSpecifyingExtension
+		arguments:
+			nullContext: true
+		tags:
+			- phpstan.typeSpecifier.staticMethodTypeSpecifyingExtension
+	-
+		class: PHPStan\Tests\AssertionClassMethodMultipleTypeSpecifyingExtension
+		tags:
+			- phpstan.typeSpecifier.methodTypeSpecifyingExtension
+	-
+		class: PHPStan\Tests\AssertionClassStaticMethodMultipleTypeSpecifyingExtension
+		tags:
+			- phpstan.typeSpecifier.staticMethodTypeSpecifyingExtension

tests/PHPStan/Analyser/MultipleTypeSpecifyingExtensionTypeInferenceTest.php

@@ -0,0 +1,35 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser;
+
+use PHPStan\Testing\TypeInferenceTestCase;
+
+class MultipleTypeSpecifyingExtensionTypeInferenceTest extends TypeInferenceTestCase
+{
+
+	public function dataTypeSpecifyingExtensionsTrue(): iterable
+	{
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/multiple-type-specifying-extensions-1.php');
+	}
+
+	/**
+	 * @dataProvider dataTypeSpecifyingExtensionsTrue
+	 * @param mixed ...$args
+	 */
+	public function testTypeSpecifyingExtensionsTrue(
+		string $assertType,
+		string $file,
+		...$args,
+	): void
+	{
+		$this->assertFileAsserts($assertType, $file, ...$args);
+	}
+
+	public static function getAdditionalConfigFiles(): array
+	{
+		return [
+			__DIR__ . '/MultipleTypeSpecifyingExtension.neon',
+		];
+	}
+
+}

tests/PHPStan/Analyser/data/multiple-type-specifying-extensions-1.php

@@ -0,0 +1,16 @@
+<?php
+
+use function PHPStan\Testing\assertType;
+
+/** @var string|null $foo */
+$foo = null;
+
+/** @var int|null $bar */
+$bar = null;
+
+(new \PHPStan\Tests\AssertionClass())->assertString($foo);
+\PHPStan\Tests\AssertionClass::assertInt($bar);
+
+assertType('non-empty-string', $foo);
+assertType('int<1, max>', $bar);
+

tests/PHPStan/Tests/AssertionClassMethodMultipleTypeSpecifyingExtension.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Tests;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\Analyser\TypeSpecifierContext;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Accessory\AccessoryNonEmptyStringType;
+use PHPStan\Type\MethodTypeSpecifyingExtension;
+
+class AssertionClassMethodMultipleTypeSpecifyingExtension implements MethodTypeSpecifyingExtension
+{
+
+	public function getClass(): string
+	{
+		return AssertionClass::class;
+	}
+
+	public function isMethodSupported(
+		MethodReflection $methodReflection,
+		MethodCall $node,
+		TypeSpecifierContext $context,
+	): bool
+	{
+		return $methodReflection->getName() === 'assertString';
+	}
+
+	public function specifyTypes(
+		MethodReflection $methodReflection,
+		MethodCall $node,
+		Scope $scope,
+		TypeSpecifierContext $context,
+	): SpecifiedTypes
+	{
+		return new SpecifiedTypes(['$foo' => [$node->getArgs()[0]->value, new AccessoryNonEmptyStringType()]]);
+	}
+
+}

tests/PHPStan/Tests/AssertionClassStaticMethodMultipleTypeSpecifyingExtension.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Tests;
+
+use PhpParser\Node\Expr\StaticCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\Analyser\TypeSpecifierContext;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\IntegerRangeType;
+use PHPStan\Type\StaticMethodTypeSpecifyingExtension;
+
+class AssertionClassStaticMethodMultipleTypeSpecifyingExtension implements StaticMethodTypeSpecifyingExtension
+{
+
+	public function getClass(): string
+	{
+		return AssertionClass::class;
+	}
+
+	public function isStaticMethodSupported(
+		MethodReflection $staticMethodReflection,
+		StaticCall $node,
+		TypeSpecifierContext $context,
+	): bool
+	{
+		return $staticMethodReflection->getName() === 'assertInt';
+	}
+
+	public function specifyTypes(
+		MethodReflection $staticMethodReflection,
+		StaticCall $node,
+		Scope $scope,
+		TypeSpecifierContext $context,
+	): SpecifiedTypes
+	{
+		return new SpecifiedTypes(['$foo' => [$node->getArgs()[0]->value, IntegerRangeType::createAllGreaterThan(0)]]);
+	}
+
+}