From 1567d0ff6f2e2c3959f5b8b543d84d9e57051a03 Mon Sep 17 00:00:00 2001
From: Ondrej Mirtes <ondrej@mirtes.cz>
Date: Sat, 20 Jun 2020 13:40:31 +0200
Subject: [PATCH] testFileAsserts - testing variable certainty support

---
 .../Analyser/NodeScopeResolverTest.php        | 78 ++++++++++++++-----
 .../data/root-scope-maybe-defined.php         |  9 +++
 tests/PHPStan/Analyser/functions.php          | 10 +++
 3 files changed, 78 insertions(+), 19 deletions(-)

diff --git a/tests/PHPStan/Analyser/NodeScopeResolverTest.php b/tests/PHPStan/Analyser/NodeScopeResolverTest.php
index 43e2d27631..fc03ab6c16 100644
--- a/tests/PHPStan/Analyser/NodeScopeResolverTest.php
+++ b/tests/PHPStan/Analyser/NodeScopeResolverTest.php
@@ -10027,23 +10027,35 @@ public function dataArrayTypehintWithoutNullInPhpDoc(): array
 	 * @dataProvider dataMixedTypehint
 	 * @dataProvider dataVariadics
 	 * @dataProvider dataArrayTypehintWithoutNullInPhpDoc
-	 * @param ConstantStringType $expectedType
-	 * @param Type $actualType
+	 * @param string $assertType
+	 * @param string $file
+	 * @param mixed ...$args
 	 */
 	public function testFileAsserts(
+		string $assertType,
 		string $file,
-		ConstantStringType $expectedType,
-		Type $actualType,
-		int $line
+		...$args
 	): void
 	{
-		$expected = $expectedType->getValue();
-		$actual = $actualType->describe(VerbosityLevel::precise());
-		$this->assertSame(
-			$expected,
-			$actual,
-			sprintf('Expected type %s, got type %s in %s on line %d.', $expected, $actual, $file, $line)
-		);
+		if ($assertType === 'type') {
+			$expectedType = $args[0];
+			$expected = $expectedType->getValue();
+			$actualType = $args[1];
+			$actual = $actualType->describe(VerbosityLevel::precise());
+			$this->assertSame(
+				$expected,
+				$actual,
+				sprintf('Expected type %s, got type %s in %s on line %d.', $expected, $actual, $file, $args[2])
+			);
+		} elseif ($assertType === 'variableCertainty') {
+			$expectedCertainty = $args[0];
+			$actualCertainty = $args[1];
+			$variableName = $args[2];
+			$this->assertTrue(
+				$expectedCertainty->equals($actualCertainty),
+				sprintf('Expected %s, actual certainty of variable $%s is %s', $expectedCertainty->describe(), $variableName, $actualCertainty->describe())
+			);
+		}
 	}
 
 	/**
@@ -10052,8 +10064,8 @@ public function testFileAsserts(
 	 */
 	private function gatherAssertTypes(string $file): array
 	{
-		$types = [];
-		$this->processFile($file, function (Node $node, Scope $scope) use (&$types, $file): void {
+		$asserts = [];
+		$this->processFile($file, function (Node $node, Scope $scope) use (&$asserts, $file): void {
 			if (!$node instanceof Node\Expr\FuncCall) {
 				return;
 			}
@@ -10065,13 +10077,41 @@ private function gatherAssertTypes(string $file): array
 
 			$functionName = $nameNode->toString();
 			if ($functionName === 'PHPStan\\Analyser\\assertType') {
-				$assertTypeFunctionName = 'PHPStan\\Analyser\\assertType';
 				$expectedType = $scope->getType($node->args[0]->value);
 				$actualType = $scope->getType($node->args[1]->value);
+				$assert = ['type', $file, $expectedType, $actualType, $node->getLine()];
 			} elseif ($functionName === 'PHPStan\\Analyser\\assertNativeType') {
-				$assertTypeFunctionName = 'PHPStan\\Analyser\\assertNativeType';
 				$expectedType = $scope->getNativeType($node->args[0]->value);
 				$actualType = $scope->getNativeType($node->args[1]->value);
+				$assert = ['type', $file, $expectedType, $actualType, $node->getLine()];
+			} elseif ($functionName === 'PHPStan\\Analyser\\assertVariableCertainty') {
+				$certainty = $node->args[0]->value;
+				if (!$certainty instanceof StaticCall) {
+					$this->fail(sprintf('First argument of %s() must be TrinaryLogic call', $functionName));
+				}
+				if (!$certainty->class instanceof Node\Name) {
+					$this->fail(sprintf('ERROR: Invalid TrinaryLogic call.'));
+				}
+
+				if ($certainty->class->toString() !== 'PHPStan\\TrinaryLogic') {
+					$this->fail(sprintf('ERROR: Invalid TrinaryLogic call.'));
+				}
+
+				if (!$certainty->name instanceof Node\Identifier) {
+					$this->fail(sprintf('ERROR: Invalid TrinaryLogic call.'));
+				}
+
+				$expectedertaintyValue = TrinaryLogic::{$certainty->name->toString()}();
+				$variable = $node->args[1]->value;
+				if (!$variable instanceof Node\Expr\Variable) {
+					$this->fail(sprintf('ERROR: Invalid assertVariableCertainty call.'));
+				}
+				if (!is_string($variable->name)) {
+					$this->fail(sprintf('ERROR: Invalid assertVariableCertainty call.'));
+				}
+
+				$actualCertaintyValue = $scope->hasVariableType($variable->name);
+				$assert = ['variableCertainty', $file, $expectedertaintyValue, $actualCertaintyValue, $variable->name];
 			} else {
 				return;
 			}
@@ -10079,15 +10119,15 @@ private function gatherAssertTypes(string $file): array
 			if (count($node->args) !== 2) {
 				$this->fail(sprintf(
 					'ERROR: Wrong %s() call on line %d.',
-					$assertTypeFunctionName,
+					$functionName,
 					$node->getLine()
 				));
 			}
 
-			$types[$file . ':' . $node->getLine()] = [$file, $expectedType, $actualType, $node->getLine()];
+			$asserts[$file . ':' . $node->getLine()] = $assert;
 		});
 
-		return $types;
+		return $asserts;
 	}
 
 	public function dataInferPrivatePropertyTypeFromConstructor(): array
diff --git a/tests/PHPStan/Analyser/data/root-scope-maybe-defined.php b/tests/PHPStan/Analyser/data/root-scope-maybe-defined.php
index c723335076..095934198c 100644
--- a/tests/PHPStan/Analyser/data/root-scope-maybe-defined.php
+++ b/tests/PHPStan/Analyser/data/root-scope-maybe-defined.php
@@ -1,8 +1,13 @@
 <?php
 
+use PHPStan\TrinaryLogic;
+
 \PHPStan\Analyser\assertType('mixed', $foo);
+\PHPStan\Analyser\assertVariableCertainty(TrinaryLogic::createMaybe(), $foo);
 
 $bar = 'str';
+\PHPStan\Analyser\assertVariableCertainty(TrinaryLogic::createYes(), $bar);
+\PHPStan\Analyser\assertVariableCertainty(TrinaryLogic::createMaybe(), $baz);
 
 \PHPStan\Analyser\assertType('\'str\'', $bar);
 
@@ -12,3 +17,7 @@
 }
 
 \PHPStan\Analyser\assertType('mixed', $baz);
+
+function () {
+	\PHPStan\Analyser\assertVariableCertainty(TrinaryLogic::createNo(), $foo);
+};
diff --git a/tests/PHPStan/Analyser/functions.php b/tests/PHPStan/Analyser/functions.php
index aa17f0e9f8..a736a4f2e4 100644
--- a/tests/PHPStan/Analyser/functions.php
+++ b/tests/PHPStan/Analyser/functions.php
@@ -2,6 +2,8 @@
 
 namespace PHPStan\Analyser;
 
+use PHPStan\TrinaryLogic;
+
 /**
  * Asserts the static type of a value.
  *
@@ -24,3 +26,11 @@ function assertType(string $type, $value): void // phpcs:ignore
 function assertNativeType(string $type, $value): void // phpcs:ignore
 {
 }
+
+/**
+ * @param TrinaryLogic $certainty
+ * @param mixed $variable
+ */
+function assertVariableCertainty(TrinaryLogic $certainty, $variable): void // phpcs:ignore
+{
+}