raise arguement error if uninitialized constant

ruby-shoryuken · Feb 9, 2024 · 679eada · 679eada
1 parent e4ad9cb
commit 679eada
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/shoryuken/options.rb b/lib/shoryuken/options.rb
@@ -68,11 +68,13 @@ def polling_strategy(group)
       when 'StrictPriority'
         Polling::StrictPriority
       when String
+        begin
         Object.const_get(strategy)
+        rescue NameError
+          raise ArgumentError, "#{strategy} is not a valid polling_strategy"
+        end
       when Class
         strategy
-      else
-        raise ArgumentError, "#{strategy} is not a valid polling_strategy"
       end
     end