fix: ENUM and SET datatype FactoryGenerator

Author: Nathan E

src/Generators/FactoryGenerator.php

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

tests/fixtures/definitions/phone.bp

@@ -4,3 +4,4 @@ models:
     user_id: id
     phone_number: text
     type: enum:home,cell
+    status: set:archived,deleted

tests/fixtures/factories/phone.php

@@ -10,6 +10,7 @@
         'label' => $faker->word,
         'user_id' => factory(\App\User::class),
         'phone_number' => $faker->phoneNumber,
-        'type' => $faker->randomElement(["home", "cell"]),
+        'type' => $faker->randomElement(["home","cell"]),
+        'status' => $faker->randomElement(["archived","deleted"]),
     ];
 });