Allow non-string enum list #79

nobu · nobu · commit c5ec052efc7e · 2025-03-09T22:39:12.000+09:00
Command line arguments are strings, convert enum list elements to
strings to match.
diff --git a/lib/optparse.rb b/lib/optparse.rb
@@ -1503,7 +1503,7 @@ def make_switch(opts, block = nil)
         else
           raise ArgumentError, "argument pattern given twice"
         end
-        o.each {|pat, *v| pattern[pat] = v.fetch(0) {pat}}
+        o.each {|pat, *v| pattern[pat.to_s] = v.fetch(0) {pat}}
       when Module
         raise ArgumentError, "unsupported argument type: #{o}", ParseError.filter_backtrace(caller(4))
       when *ArgumentStyle.keys
diff --git a/test/optparse/test_placearg.rb b/test/optparse/test_placearg.rb
@@ -8,6 +8,7 @@ def setup
     @opt.def_option("--option [VAL]") {|x| @flag = x}
     @opt.def_option("-T [level]", /^[0-4]$/, Integer) {|x| @topt = x}
     @opt.def_option("--enum [VAL]", [:Alpha, :Bravo, :Charlie]) {|x| @enum = x}
+    @opt.def_option("--integer [VAL]", [1, 2, 3]) {|x| @integer = x}
     @topt = nil
     @opt.def_option("-n") {}
     @opt.def_option("--regexp [REGEXP]", Regexp) {|x| @reopt = x}
@@ -99,4 +100,9 @@ def test_enum
     assert_equal([], no_error {@opt.parse!(%w"--enum=A")})
     assert_equal(:Alpha, @enum)
   end
+
+  def test_enum_conversion
+    assert_equal([], no_error {@opt.parse!(%w"--integer=1")})
+    assert_equal(1, @integer)
+  end
 end
