Test Printer::print()

Author: ondrejmirtes

Diff for: tests/PHPStan/Printer/PrinterTest.php

@@ -42,15 +42,19 @@
 class PrinterTest extends TestCase
 {
 
+	/** @var TypeParser */
+	private $typeParser;
+
 	/** @var PhpDocParser */
 	private $phpDocParser;
 
 	protected function setUp(): void
 	{
 		$usedAttributes = ['lines' => true, 'indexes' => true];
 		$constExprParser = new ConstExprParser(true, true, $usedAttributes);
+		$this->typeParser = new TypeParser($constExprParser, true, $usedAttributes);
 		$this->phpDocParser = new PhpDocParser(
-			new TypeParser($constExprParser, true, $usedAttributes),
+			$this->typeParser,
 			$constExprParser,
 			true,
 			true,
@@ -1190,7 +1194,7 @@ public function testPrintFormatPreserving(string $phpDoc, string $expectedResult
 		);
 	}
 
-	private function unsetAttributes(PhpDocNode $node): PhpDocNode
+	private function unsetAttributes(Node $node): Node
 	{
 		$visitor = new class extends AbstractNodeVisitor {
 
@@ -1213,4 +1217,61 @@ public function enterNode(Node $node)
 		return $traverser->traverse([$node])[0];
 	}
 
+	public function dataPrintType(): iterable
+	{
+		yield [
+			new IdentifierTypeNode('int'),
+			'int',
+		];
+	}
+
+	/**
+	 * @dataProvider dataPrintType
+	 */
+	public function testPrintType(TypeNode $node, string $expectedResult): void
+	{
+		$printer = new Printer();
+		$phpDoc = $printer->print($node);
+		$this->assertSame($expectedResult, $phpDoc);
+
+		$lexer = new Lexer();
+		$this->assertEquals(
+			$this->unsetAttributes($node),
+			$this->unsetAttributes($this->typeParser->parse(new TokenIterator($lexer->tokenize($phpDoc))))
+		);
+	}
+
+	public function dataPrintPhpDocNode(): iterable
+	{
+		yield [
+			new PhpDocNode([
+				new PhpDocTagNode('@param', new ParamTagValueNode(
+					new IdentifierTypeNode('int'),
+					false,
+					'$a',
+					''
+				)),
+			]),
+			'/**
+ * @param int $a
+ */',
+		];
+	}
+
+	/**
+	 * @dataProvider dataPrintPhpDocNode
+	 */
+	public function testPrintPhpDocNode(PhpDocNode $node, string $expectedResult): void
+	{
+		$printer = new Printer();
+		$phpDoc = $printer->print($node);
+		$this->assertSame($expectedResult, $phpDoc);
+
+		$lexer = new Lexer();
+		$this->assertEquals(
+			$this->unsetAttributes($node),
+			$this->unsetAttributes($this->phpDocParser->parse(new TokenIterator($lexer->tokenize($phpDoc))))
+		);
+	}
+
 }