PhpFile: added addUse() [Closes #38]

nette · Nov 27, 2018 · 9de4e09 · 9de4e09
1 parent 7c5e08e
commit 9de4e09
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/src/PhpGenerator/PhpFile.php b/src/PhpGenerator/PhpFile.php
@@ -77,6 +77,16 @@ public function getNamespaces(): array
 	}
 
 
+	/**
+	 * @return static
+	 */
+	public function addUse(string $name, string $alias = null): self
+	{
+		$this->addNamespace('')->addUse($name, $alias);
+		return $this;
+	}
+
+
 	/**
 	 * Adds declare(strict_types=1) to output.
 	 * @return static

diff --git a/tests/PhpGenerator/PhpFile.phpt b/tests/PhpGenerator/PhpFile.phpt
@@ -78,6 +78,8 @@ sameFile(__DIR__ . '/expected/PhpFile.bracketed.expect', (string) $file);
 
 $file = new PhpFile;
 $file->addClass('A');
+$file->addUse('A')
+	->addUse('B', 'C');
 
 sameFile(__DIR__ . '/expected/PhpFile.globalNamespace.expect', (string) $file);
 

diff --git a/tests/PhpGenerator/expected/PhpFile.globalNamespace.expect b/tests/PhpGenerator/expected/PhpFile.globalNamespace.expect
@@ -1,5 +1,7 @@
 <?php
 
+use B as C;
+
 class A
 {
 }