Check that enum name is a symbol

Fixes #1627
presidentbeef · Sep 18, 2021 · f379c48 · f379c48
1 parent 03d1c7a
commit f379c48
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/lib/brakeman/processors/model_processor.rb b/lib/brakeman/processors/model_processor.rb
@@ -93,6 +93,7 @@ def process_call exp
   def add_enum_method call
     arg = call.first_arg
     return unless hash? arg
+    return unless symbol? arg[1]
 
     enum_name = arg[1].value # first key
     enums = arg[2] # first value

diff --git a/test/apps/rails6/app/models/user.rb b/test/apps/rails6/app/models/user.rb
@@ -30,4 +30,6 @@ def recent_stuff
   def check_enum
     where("state = #{User.states["pending"]}")
   end
+
+  enum "stuff_#{stuff}": [:things]
 end