Merge pull request #38 from laravel-shift/axit-joost-master

Generate models for relationships in factory definitions

Author: jasonmccreary

src/Generators/FactoryGenerator.php

@@ -4,6 +4,7 @@
 
 use Blueprint\Contracts\Generator;
 use Blueprint\Models\Model;
+use Illuminate\Support\Str;
 
 class FactoryGenerator implements Generator
 {
@@ -61,10 +62,19 @@ protected function buildDefinition(Model $model)
                 continue;
             }
 
-            $definition .= self::INDENT . "'{$column->name()}' => ";
-            $faker = $this->fakerData($column->name()) ?? $this->fakerDataType($column->dataType());
-            $definition .= '$faker->' . $faker;
-            $definition .= ',' . PHP_EOL;
+            if ($column->dataType() === 'id') {
+                $name = Str::substr($column->name(), 0, -3);
+                $class = Str::studly($column->attributes()[0] ?? $name);
+
+                $definition .= self::INDENT . "'{$column->name()}' => ";
+                $definition .= sprintf("factory(\App\%s::class)", $class);
+                $definition .= ',' . PHP_EOL;
+            } else {
+                $definition .= self::INDENT . "'{$column->name()}' => ";
+                $faker = $this->fakerData($column->name()) ?? $this->fakerDataType($column->dataType());
+                $definition .= '$faker->' . $faker;
+                $definition .= ',' . PHP_EOL;
+            }
         }
 
         return trim($definition);

tests/Feature/Generator/FactoryGeneratorTest.php

@@ -64,7 +64,8 @@ public function output_writes_migration_for_model_tree($definition, $path, $migr
     public function modelTreeDataProvider()
     {
         return [
-            ['definitions/post.bp', 'database/factories/PostFactory.php', 'factories/post.php']
+            ['definitions/post.bp', 'database/factories/PostFactory.php', 'factories/post.php'],
+            ['definitions/team.bp', 'database/factories/TeamFactory.php', 'factories/team.php']
         ];
     }
 }

tests/fixtures/definitions/team.bp

@@ -0,0 +1,4 @@
+models:
+  Team:
+    name: string
+    owner_id: id:user

tests/fixtures/factories/post.php

@@ -8,7 +8,7 @@
 $factory->define(Post::class, function (Faker $faker) {
     return [
         'title' => $faker->sentence(4),
-        'author_id' => $faker->randomDigitNotNull,
+        'author_id' => factory(\App\Author::class),
         'content' => $faker->paragraphs(3, true),
         'published_at' => $faker->dateTime(),
         'word_count' => $faker->randomNumber(),

tests/fixtures/factories/team.php

@@ -0,0 +1,13 @@
+<?php
+
+/** @var \Illuminate\Database\Eloquent\Factory $factory */
+
+use App\Team;
+use Faker\Generator as Faker;
+
+$factory->define(Team::class, function (Faker $faker) {
+    return [
+        'name' => $faker->name,
+        'owner_id' => factory(\App\User::class),
+    ];
+});