Merge branch 'hotfix/4.0.1'

Author: mikaelcom

CHANGELOG.md

@@ -1,5 +1,8 @@
 # CHANGELOG
 
+## 4.0.1 - 2021/02/04
+- Add back `getMainElement` method to `AbstractComponent`
+
 ## 4.0.0 - 2021/02/04
 - issue #11 - Allow class property to be typed
 - Clear, refactor and review code

src/Component/AbstractComponent.php

@@ -39,6 +39,14 @@ public function setMainElement(AbstractElement $element): self
         return $this;
     }
 
+    /**
+     * @return PhpClassElement|PhpFileElement
+     */
+    public function getMainElement()
+    {
+        return $this->mainElement;
+    }
+
     public function addConstantElement(PhpConstantElement $constant): self
     {
         if (!$constant->getClass() instanceof PhpClassElement && $this->mainElement instanceof PhpClassElement) {

tests/Component/PhpClassTest.php

@@ -6,6 +6,7 @@
 
 use WsdlToPhp\PhpGenerator\Component\PhpClass as PhpClassComponent;
 use WsdlToPhp\PhpGenerator\Element\PhpAnnotation as PhpAnnotationElement;
+use WsdlToPhp\PhpGenerator\Element\PhpClass;
 use WsdlToPhp\PhpGenerator\Element\PhpFunctionParameter as PhpFunctionParameterElement;
 use WsdlToPhp\PhpGenerator\Element\PhpProperty;
 use WsdlToPhp\PhpGenerator\Element\PhpProperty as PhpPropertyElement;
@@ -16,6 +17,13 @@
  */
 class PhpClassTest extends AbstractComponent
 {
+    public function testGetMainElementMustBeOfPhpClassElement()
+    {
+        $class = new PhpClassComponent('Foo', true, 'stdClass');
+
+        $this->assertInstanceOf(PhpClass::class, $class->getMainElement());
+    }
+
     public function testSimpleToString()
     {
         $class = new PhpClassComponent('Foo', true, 'stdClass');

tests/Component/PhpFileTest.php

@@ -10,6 +10,7 @@
 use WsdlToPhp\PhpGenerator\Component\PhpInterface as PhpInterfaceComponent;
 use WsdlToPhp\PhpGenerator\Element\PhpAnnotation as PhpAnnotationElement;
 use WsdlToPhp\PhpGenerator\Element\PhpDeclare;
+use WsdlToPhp\PhpGenerator\Element\PhpFile;
 use WsdlToPhp\PhpGenerator\Element\PhpFunctionParameter as PhpFunctionParameterElement;
 use WsdlToPhp\PhpGenerator\Element\PhpProperty;
 use WsdlToPhp\PhpGenerator\Element\PhpProperty as PhpPropertyElement;
@@ -20,6 +21,13 @@
  */
 class PhpFileTest extends AbstractComponent
 {
+    public function testGetMainElementMustBeOfPhpFileElement()
+    {
+        $class = new PhpFileComponent('Foo', true, 'stdClass');
+
+        $this->assertInstanceOf(PhpFile::class, $class->getMainElement());
+    }
+
     public function testSimpleClassToString()
     {
         $file = new PhpFileComponent('Foo');