From d896dc28b09baa9ab832a8273caffc0551927b30 Mon Sep 17 00:00:00 2001
From: John Bintz <john@coswellproductions.com>
Date: Mon, 12 Aug 2013 10:57:05 -0400
Subject: [PATCH] better finder

---
 lib/semantic_rails_view_helpers/capybara.rb | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/lib/semantic_rails_view_helpers/capybara.rb b/lib/semantic_rails_view_helpers/capybara.rb
index 54fb82f..a1ac6f0 100644
--- 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?