Adds location to class

Author: jaapio

src/phpDocumentor/Reflection/Php/Class_.php

@@ -15,6 +15,7 @@
 use phpDocumentor\Reflection\DocBlock;
 use phpDocumentor\Reflection\Element;
 use phpDocumentor\Reflection\Fqsen;
+use phpDocumentor\Reflection\Location;
 
 /**
  * Descriptor representing a Class.
@@ -59,6 +60,11 @@ final class Class_ implements Element
     /** @var Fqsen[] $usedTraits References to traits consumed by this class */
     private $usedTraits = array();
 
+    /**
+     * @var null|Location
+     */
+    private $location;
+
     /**
      * Initializes a number of properties with the given values. Others are initialized by definition.
      *
@@ -67,19 +73,22 @@ final class Class_ implements Element
      * @param Fqsen $parent
      * @param bool $abstract
      * @param bool $final
+     * @param Location|null $location
      */
     public function __construct(
         Fqsen $fqsen,
         DocBlock $docBlock = null,
         Fqsen $parent = null,
         $abstract = false,
-        $final = false
+        $final = false,
+        Location $location = null
     ) {
         $this->fqsen = $fqsen;
         $this->parent = $parent;
         $this->docBlock = $docBlock;
         $this->abstract = $abstract;
         $this->final = $final;
+        $this->location = $location;
     }
 
     /**
@@ -243,4 +252,12 @@ public function getDocBlock()
     {
         return $this->docBlock;
     }
+
+    /**
+     * @return null|Location
+     */
+    public function getLocation()
+    {
+        return $this->location;
+    }
 }

src/phpDocumentor/Reflection/Php/Factory/Class_.php

@@ -16,6 +16,7 @@
 use InvalidArgumentException;
 use phpDocumentor\Reflection\Element;
 use phpDocumentor\Reflection\Fqsen;
+use phpDocumentor\Reflection\Location;
 use phpDocumentor\Reflection\Php\Class_ as ClassElement;
 use phpDocumentor\Reflection\Php\ProjectFactoryStrategy;
 use phpDocumentor\Reflection\Php\StrategyContainer;
@@ -65,7 +66,8 @@ protected function doCreate($object, StrategyContainer $strategies, Context $con
             $docBlock,
             $object->extends ? new Fqsen('\\' . $object->extends) : null,
             $object->isAbstract(),
-            $object->isFinal()
+            $object->isFinal(),
+            new Location($object->getLine())
         );
 
         if (isset($object->implements)) {

tests/unit/phpDocumentor/Reflection/Php/Class_Test.php

@@ -15,6 +15,7 @@
 use Mockery as m;
 use phpDocumentor\Reflection\DocBlock;
 use phpDocumentor\Reflection\Fqsen;
+use phpDocumentor\Reflection\Location;
 
 /**
  * Tests the functionality for the Class_ class.
@@ -52,7 +53,7 @@ protected function setUp()
         $this->fqsen = new Fqsen('\MyClass');
         $this->docBlock = new DocBlock('');
 
-        $this->fixture = new Class_($this->fqsen, $this->docBlock, null, false, false);
+        $this->fixture = new Class_($this->fqsen, $this->docBlock, null, false, false, new Location(1));
     }
 
     /**
@@ -61,10 +62,10 @@ protected function setUp()
      */
     public function testGettingParent()
     {
-        $class = new Class_($this->fqsen, $this->docBlock, null, false, false);
+        $class = new Class_($this->fqsen, $this->docBlock, null, false, false, null);
         $this->assertNull($class->getParent());
 
-        $class = new Class_($this->fqsen, $this->docBlock, $this->parent, false, false);
+        $class = new Class_($this->fqsen, $this->docBlock, $this->parent, false, false, null);
         $this->assertSame($this->parent, $class->getParent());
     }
 

tests/unit/phpDocumentor/Reflection/Php/Factory/Class_Test.php

@@ -264,6 +264,7 @@ private function buildClassMock()
         $classMock->fqsen = new Fqsen('\Space\MyClass');
         $classMock->shouldReceive('isFinal')->andReturn(true);
         $classMock->shouldReceive('isAbstract')->andReturn(true);
+        $classMock->shouldReceive('getLine')->andReturn(1);
         return $classMock;
     }
 }