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
 {
 }