Merge pull request #26 from jhoff/add_tests

Adds tests for traits and default filenames

Author: mpociot

.travis.yml

@@ -9,6 +9,10 @@ env:
     - COMPOSER_FLAGS="--prefer-lowest"
     - COMPOSER_FLAGS=""
 
+before_install:
+  - sudo apt-get update
+  - sudo apt-get install -y graphviz
+
 before_script:
   - travis_retry composer self-update
   - travis_retry composer update ${COMPOSER_FLAGS} --no-interaction --prefer-source

tests/GenerationTest.php

@@ -46,4 +46,16 @@ public function it_generated_graphviz_for_test_models_with_db_columns()
 
         $this->assertMatchesSnapshot(Artisan::output());
     }
+
+    /** @test */
+    public function it_generated_graphviz_in_jpeg_format()
+    {
+        $this->app['config']->set('erd-generator.directories', [__DIR__.'/Models']);
+
+        Artisan::call('generate:erd', [
+            '--format' => 'jpeg'
+        ]);
+
+        $this->assertContains('Wrote diagram to graph.jpeg', Artisan::output());
+    }
 }

tests/Models/Post.php

@@ -3,18 +3,15 @@
 namespace BeyondCode\ErdGenerator\Tests\Models;
 
 use Illuminate\Database\Eloquent\Model;
+use BeyondCode\ErdGenerator\Tests\Traits\HasComments;
 
 class Post extends Model
 {
+    use HasComments;
 
     public function user()
     {
         return $this->belongsTo(User::class);
     }
 
-    public function comments()
-    {
-        return $this->hasMany(Comment::class);
-    }
-
-}
+}

tests/Traits/HasComments.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace BeyondCode\ErdGenerator\Tests\Traits;
+
+use BeyondCode\ErdGenerator\Tests\Models\Comment;
+
+trait HasComments
+{
+
+    public function comments()
+    {
+        return $this->hasMany(Comment::class);
+    }
+
+}