Reworked root and selector conditional assignment in css_select.

kaspth · Jun 16, 2014 · bffa646 · bffa646
1 parent 5a14dbf
commit bffa646
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/actionview/lib/action_view/testing/assertions/selector.rb b/actionview/lib/action_view/testing/assertions/selector.rb
@@ -49,11 +49,8 @@ module SelectorAssertions
       def css_select(*args)
         raise ArgumentError, "you at least need a selector" if args.empty?
 
-        if args.first.is_a?(String)
-          root, selector = response_from_page, args.first
-        else
-          root, selector = args.shift, args.first
-        end
+        root = args.size == 1 ? response_from_page : args.shift
+        selector = args.first
 
         root.css(selector).tap do |matches|
           if matches.empty? && root.matches?(selector)