added aliasing support

robertmarsal · Oct 9, 2012 · 1c909df · 1c909df
1 parent 86e3975
commit 1c909df
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/library/Zend/Code/Generator/ClassGenerator.php b/library/Zend/Code/Generator/ClassGenerator.php
@@ -489,8 +489,11 @@ public function addPropertyFromGenerator(PropertyGenerator $property)
      *
      * @param string $useClass
      */
-    public function addUseClass($useClass)
+    public function addUseClass($useClass, $useClassAlias = null)
     {
+        if (!empty($useClassAlias)) {
+            $useClass .= ' as ' . $useClassAlias;
+        }
         $this->useClasses[] = $useClass;
     }
 

diff --git a/tests/ZendTest/Code/Generator/ClassGeneratorTest.php b/tests/ZendTest/Code/Generator/ClassGeneratorTest.php
@@ -371,10 +371,10 @@ public function testAddUseClass()
         $classGenerator = new ClassGenerator();
         $classGenerator->setName('My\Class');
         $classGenerator->addUseClass('My\First\Use\Class');
-        $classGenerator->addUseClass('My\Second\Use\Class');
+        $classGenerator->addUseClass('My\Second\Use\Class', 'MyAlias');
         $generated = $classGenerator->generate();
 
         $this->assertContains('use My\First\Use\Class;', $generated);
-        $this->assertContains('use My\Second\Use\Class;', $generated);
+        $this->assertContains('use My\Second\Use\Class as MyAlias;', $generated);
     }
 }