Merge pull request #60 from nathane/bugfix/issues-59-fix-enums-and-set

ENUM and SET datatypes for FactoryGenerator

Author: jasonmccreary

src/Generators/FactoryGenerator.php

@@ -79,6 +79,10 @@ protected function buildDefinition(Model $model)
                 $faker = $this->fakerData($column->name()) ?? $this->fakerDataType($column->dataType());
                 $definition .= '$faker->' . $faker;
                 $definition .= ',' . PHP_EOL;
+
+                if (in_array($column->dataType(), ['enum', 'set']) and !empty($column->attributes())) {
+                    $definition = str_replace("/** {$column->dataType()}_attributes **/", json_encode($column->attributes()), $definition);
+                }
             }
         }
 
@@ -143,7 +147,9 @@ protected function fakerDataType(string $type)
             'decimal' => 'randomFloat()',
             'float' => 'randomFloat()',
             'longtext' => 'text',
-            'boolean' => 'boolean'
+            'boolean' => 'boolean',
+            'set' => 'randomElement(/** set_attributes **/)',
+            'enum' => 'randomElement(/** enum_attributes **/)',
         ];
 
         return $fakeableTypes[$type] ?? null;

tests/Feature/Generator/FactoryGeneratorTest.php

@@ -87,6 +87,7 @@ public function output_respects_configuration()
     public function modelTreeDataProvider()
     {
         return [
+            ['definitions/phone.bp', 'database/factories/PhoneFactory.php', 'factories/phone.php'],
             ['definitions/post.bp', 'database/factories/PostFactory.php', 'factories/post.php'],
             ['definitions/team.bp', 'database/factories/TeamFactory.php', 'factories/team.php'],
             ['definitions/unconventional.bp', 'database/factories/TeamFactory.php', 'factories/unconventional.php'],

tests/fixtures/definitions/phone.bp

@@ -0,0 +1,7 @@
+models:
+  Phone:
+    label: string
+    user_id: id
+    phone_number: text
+    type: enum:home,cell
+    status: set:archived,deleted

tests/fixtures/factories/phone.php

@@ -0,0 +1,16 @@
+<?php
+
+/** @var \Illuminate\Database\Eloquent\Factory $factory */
+
+use App\Phone;
+use Faker\Generator as Faker;
+
+$factory->define(Phone::class, function (Faker $faker) {
+    return [
+        'label' => $faker->word,
+        'user_id' => factory(\App\User::class),
+        'phone_number' => $faker->phoneNumber,
+        'type' => $faker->randomElement(["home","cell"]),
+        'status' => $faker->randomElement(["archived","deleted"]),
+    ];
+});