Class pretty printing prioritizing

src/Support/PSR2PrettyPrinter.php

@@ -20,17 +20,13 @@ public function __construct($options = [])
     // Fix empty line before class definition
     protected function pStmt_Class(Class_ $node)
     {
-        return PHP_EOL . $this->pClassCommon($node, ' ' . $node->name);
+        return $this->pClassCommon($node, ' ' . $node->name);
     }
 
     // Fix empty line before class definition
     protected function pStmt_ClassMethod(ClassMethod $node)
     {
-        $comments = $node->getComments();
-
-        $ln = $comments ? '' : $this->nl;
-
-        return $ln . $this->pAttrGroups($node->attrGroups)
+        return $this->pAttrGroups($node->attrGroups)
              . $this->pModifiers($node->flags)
              . 'function ' . ($node->byRef ? '&' : '') . $node->name
              . '(' . $this->pMaybeMultiline($node->params) . ')'

tests/Feature/PrettyPrintTest.php

@@ -1,6 +1,7 @@
 <?php
 
 use Archetype\Tests\Support\Facades\TestablePHPFile as PHPFile;
+use PhpParser\BuilderFactory;
 
 test('arrays are beutiful when loaded and rendered', function() {
 	PHPFile::load('app/Models/User.php')
@@ -35,5 +36,19 @@
 test('created class with use statements have beautiful header'/*, function() {
 	PHPFile::make()->class()
 		->use('\TraitX')
+		->assertProperSpacingInClassHeader()
+		->preview();
+}*/);
+
+test('created and modified class without use statements have beautiful header', function() {
+	PHPFile::make()->class(App\Dummy::class)
+		->astQuery()
+		->class()
+		->insertStmt(
+			(new BuilderFactory)->method('myMethod')->getNode()
+		)
+		->commit()
+		->end()
 		->assertProperSpacingInClassHeader();
-}*/);
+});
+