Add in rule for enums and sets

Author: Mark Salmon

src/Translators/Rules.php

@@ -52,6 +52,9 @@ public static function fromColumn(Column $column)
             }
         }
 
+        if (in_array($column->dataType(), ['enum', 'set'])) {
+            $rules = array_merge($rules, ['in:' . implode(',', $column->attributes())]);
+        }
 
         if ($column->attributes()) {
             if (in_array($column->dataType(), ['string', 'char'])) {

tests/Feature/Translators/RulesTest.php

@@ -99,6 +99,19 @@ public function forColumn_returns_gt0_rule_for_unsigned_numeric_types()
         $this->assertContains('numeric', Rules::fromColumn($column));
     }
 
+    /**
+     * @test
+     */
+    public function forColumn_returns_in_rule_for_enums_and_sets()
+    {
+        $column = new Column('test', 'enum', [], ['alpha', 'bravo', 'charlie']);
+        $this->assertContains('in:alpha,bravo,charlie', Rules::fromColumn($column));
+
+        $column = new Column('test', 'set', [], [2,4,6]);
+
+        $this->assertContains('in:2,4,6', Rules::fromColumn($column));
+    }
+
     public function stringDataTypesProvider()
     {
         return [