Check if type: array values are in enum

Author: movermeyer

lib/thor/parser/arguments.rb

@@ -122,7 +122,15 @@ def parse_hash(name)
     def parse_array(name)
       return shift if peek.is_a?(Array)
       array = []
-      array << shift while current_is_value?
+      while current_is_value?
+        value = shift
+        if !value.empty? && @switches.is_a?(Hash) && switch = @switches[name]
+          if switch.enum && !switch.enum.include?(value)
+            raise MalformattedArgumentError, "Expected all values of '#{name}' to be one of #{switch.enum.join(', ')}; got #{value}"
+          end
+        end
+        array << value
+      end
       array
     end
 

spec/parser/options_spec.rb

@@ -443,6 +443,13 @@ def remaining
         create :attributes => Thor::Option.new("attributes", :type => :array, :repeatable => true)
         expect(parse("--attributes", "1", "2", "--attributes", "3", "4")["attributes"]).to eq([["1", "2"], ["3", "4"]])
       end
+
+      it "raises error when value isn't in enum" do
+        enum = %w(apple banana)
+        create :fruit => Thor::Option.new("fruits", :type => :array, :enum => enum)
+        expect { parse("--fruits=", "apple", "banana", "strawberry") }.to raise_error(Thor::MalformattedArgumentError,
+            "Expected all values of '--fruits' to be one of #{enum.join(', ')}; got strawberry")
+      end
     end
 
     describe "with :numeric type" do