Fixing view helper so that it works without blocks. Handy for <% if(a…

…b_test("show_some_element") %> ... <% end %>
ryanb · Aug 16, 2009 · a328f34 · a328f34
1 parent a120f95
commit a328f34
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/abingo_view_helper.rb b/lib/abingo_view_helper.rb
@@ -22,8 +22,13 @@ def ab_test(test_name, alternatives = nil, options = {}, &block)
     else
       choice = Abingo.test(test_name, alternatives, options)
     end
-    content_tag = capture(choice, &block)
-    block_called_from_erb?(block) ? concat(content_tag) : content_tag
+
+    if block
+      content_tag = capture(choice, &block)
+      block_called_from_erb?(block) ? concat(content_tag) : content_tag
+    else
+      choice
+    end
   end
 
   def bingo!(test_name, options = {})