another object finder

2012-12-06 08:11:14 -05:00 · 2012-12-06 08:11:14 -05:00 · d0c101f032
parent 06e59c26fd
commit d0c101f032
1 changed files with 6 additions and 1 deletions
--- a/lib/semantic_rails_view_helpers/capybara.rb
+++ b/lib/semantic_rails_view_helpers/capybara.rb
@ -85,7 +85,12 @@ def dont_find(search)
 end

 def dont_find_object(object)
-  dont_find("[data-id='#{object.id}']")
+  case object
+  when Class
+    dont_find("[data-type='#{object}']")
+  else
+    dont_find("[data-id='#{object.id}']")
+  end
 end

 def find_object(object)