Fix model names with underscores (#41)

* change Avatar to User_Avatar

* fix tests

* remove - from node name

* docblocks

* fix ref

* snapshots

* make unit test to ensure hiphens are handle

* rename User_Avatar to Avatar, it's nicer

* fix docblocks

Author: diogogomeswww

src/GenerateDiagramCommand.php

@@ -2,11 +2,11 @@
 
 namespace BeyondCode\ErdGenerator;
 
-use ReflectionClass;
+use BeyondCode\ErdGenerator\Model as GraphModel;
 use Illuminate\Console\Command;
-use phpDocumentor\GraphViz\Graph;
 use Illuminate\Support\Collection;
-use BeyondCode\ErdGenerator\Model as GraphModel;
+use phpDocumentor\GraphViz\Graph;
+use ReflectionClass;
 
 class GenerateDiagramCommand extends Command
 {
@@ -49,6 +49,9 @@ public function __construct(ModelFinder $modelFinder, RelationFinder $relationFi
         $this->graphBuilder = $graphBuilder;
     }
 
+    /**
+     * @throws \phpDocumentor\GraphViz\Exception
+     */
     public function handle()
     {
         $models = $this->getModelsThatShouldBeInspected();

src/Model.php

@@ -30,7 +30,7 @@ public function getModel()
      */
     public function getNodeName()
     {
-        return str_slug($this->model);
+        return str_replace('-', '', str_slug($this->model));
     }
 
     /**
@@ -42,7 +42,7 @@ public function getLabel()
     }
 
     /**
-     * @return ModelRelation[]
+     * @return Collection
      */
     public function getRelations()
     {

tests/FindModelsFromConfigTest.php

@@ -19,7 +19,11 @@ public function it_can_find_class_names_from_directory()
         $classNames = $finder->getModelsInDirectory(__DIR__ . "/Models");
 
         $this->assertCount(4, $classNames);
-        $this->assertSame(Avatar::class, $classNames->first());
+
+        $this->assertSame(
+            [Avatar::class, Comment::class, Post::class, User::class],
+            $classNames->values()->all()
+        );
     }
 
     /** @test */
@@ -42,5 +46,4 @@ public function it_will_ignore_a_model_if_it_is_excluded_on_config()
             $classNames->values()->all()
         );
     }
-
 }

tests/ModelTest.php

@@ -0,0 +1,16 @@
+<?php
+
+namespace BeyondCode\ErdGenerator\Tests;
+
+use BeyondCode\ErdGenerator\Model;
+
+class ModelTest extends TestCase
+{
+    /** @test */
+    public function it_generates_a_node_name_without_hyphens()
+    {
+        $model = new Model('Test_Class', 'Test_Class', collect());
+
+        $this->assertEquals('testclass', $model->getNodeName());
+    }
+}