More tests and ability to make a classProperty

Author: quintenmuijser

composer.json

@@ -1,5 +1,6 @@
 {
     "name": "quintenmbusiness/php-ast-code-generation-helper",
+    "keywords": ["ast", "php-parser", "code generation", "ast helper", "php code generator", "code generation tool"],
     "type": "library",
     "license": "MIT",
     "require": {

src/AstGenerationHelper.php

@@ -5,6 +5,7 @@
 namespace quintenmbusiness\PhpAstCodeGenerationHelper\GeneratorHelpers;
 
 use PhpParser\BuilderFactory;
+use PhpParser\Builder\Property;
 
 class ClassGenerationHelper extends MethodGenerationHelper
 {
@@ -15,4 +16,23 @@ public function __construct(BuilderFactory $factory)
     {
         parent::__construct($factory);
     }
+
+    /**
+     * @param string $name
+     * @param null|string $type
+     * @param string $visibility
+     * @return Property
+     */
+    public function createClassProperty(string $name, ?string $type = null, string $visibility = 'public'): Property
+    {
+        $property = $this->factory->property($name);
+
+        if ($type != null) {
+            $property->setType($type);
+        }
+
+        $this->addVisibility($property, $visibility);
+
+        return $property;
+    }
 }

src/GeneratorHelpers/ClassGenerationHelper.php

@@ -0,0 +1,37 @@
+<?php
+
+declare(strict_types=1);
+
+namespace BasicGeneration;
+
+use PhpParser\Node\Expr;
+use PhpParser\BuilderFactory;
+use PHPUnit\Framework\TestCase;
+use PhpParser\Node\Expr\Variable;
+use PhpParser\Node\Stmt\Return_;
+use PHPUnit\Framework\Attributes\Test;
+use quintenmbusiness\PhpAstCodeGenerationHelper\GeneratorHelpers\BasicGenerationHelper;
+
+class ReturnStmtTest extends TestCase
+{
+    /**
+     * @var BasicGenerationHelper
+     */
+    private BasicGenerationHelper $helper;
+
+    protected function setUp(): void
+    {
+        $this->helper = new BasicGenerationHelper(new BuilderFactory());
+    }
+
+    #[Test]
+    public function it_creates_a_return_statement(): void
+    {
+        $variableExpr = new Variable('testVar');
+        $returnStmt = $this->helper->return($variableExpr);
+
+        $this->assertInstanceOf(Return_::class, $returnStmt);
+        $this->assertInstanceOf(Expr::class, $returnStmt->expr);
+        $this->assertSame($variableExpr, $returnStmt->expr);
+    }
+}

tests/BasicGeneration/ReturnStmtTest.php

@@ -0,0 +1,72 @@
+<?php
+
+declare(strict_types=1);
+
+namespace ClassGeneration;
+
+use PhpParser\Builder\Property;
+use PhpParser\BuilderFactory;
+use PHPUnit\Framework\TestCase;
+use PHPUnit\Framework\Attributes\Test;
+use quintenmbusiness\PhpAstCodeGenerationHelper\GeneratorHelpers\ClassGenerationHelper;
+
+class ClassPropertyGenerationTest extends TestCase
+{
+    private ClassGenerationHelper $helper;
+
+    protected function setUp(): void
+    {
+        $this->helper = new ClassGenerationHelper(new BuilderFactory());
+    }
+
+    #[Test]
+    public function create_class_property_with_default_visibility(): void
+    {
+        $property = $this->helper->createClassProperty('testProperty');
+
+        $this->assertInstanceOf(Property::class, $property);
+        $this->assertSame('testProperty', $property->getNode()->props[0]->name->toString());
+        $this->assertTrue($property->getNode()->isPublic());
+        $this->assertNull($property->getNode()->type);
+    }
+
+    #[Test]
+    public function create_class_property_with_protected_visibility(): void
+    {
+        $property = $this->helper->createClassProperty('testProperty', null, 'protected');
+
+        $this->assertInstanceOf(Property::class, $property);
+        $this->assertTrue($property->getNode()->isProtected());
+        $this->assertNull($property->getNode()->type);
+    }
+
+    #[Test]
+    public function create_class_property_with_private_visibility(): void
+    {
+        $property = $this->helper->createClassProperty('testProperty', null, 'private');
+
+        $this->assertInstanceOf(Property::class, $property);
+        $this->assertTrue($property->getNode()->isPrivate());
+        $this->assertNull($property->getNode()->type);
+    }
+
+    #[Test]
+    public function create_class_property_with_type(): void
+    {
+        $property = $this->helper->createClassProperty('testProperty', 'string');
+
+        $this->assertInstanceOf(Property::class, $property);
+        $this->assertSame('string', $property->getNode()->type->toString());
+        $this->assertTrue($property->getNode()->isPublic());
+    }
+
+    #[Test]
+    public function create_class_property_with_type_and_visibility(): void
+    {
+        $property = $this->helper->createClassProperty('testProperty', 'int', 'protected');
+
+        $this->assertInstanceOf(Property::class, $property);
+        $this->assertSame('int', $property->getNode()->type->toString());
+        $this->assertTrue($property->getNode()->isProtected());
+    }
+}