better finder

2013-08-12 10:57:05 -04:00 · 2013-08-12 10:57:05 -04:00 · d896dc28b0
parent 3b6863a124
commit d896dc28b0
1 changed files with 15 additions and 2 deletions
--- a/lib/semantic_rails_view_helpers/capybara.rb
+++ b/lib/semantic_rails_view_helpers/capybara.rb
@ -1,5 +1,18 @@
-def find_attribute(name, value = nil)
-  attribute = find("[data-field='#{name}']")
+def find_attribute(name, options_or_value = nil, options = nil)
+  if options
+    value = options_or_value
+  else
+    case options_or_value
+    when Hash
+      options = options_or_value
+      value = nil
+    else
+      value = options_or_value
+      options = {}
+    end
+  end
+
+  attribute = find("[data-field='#{name}']", options)

  yield attribute, @inputs[name] if block_given?