can now pass an array of values to the has method

pelargir · Apr 17, 2009 · 8b08aa6 · 8b08aa6
1 parent 7d04477
commit 8b08aa6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/test/spec/rails/convenience.rb b/lib/test/spec/rails/convenience.rb
@@ -3,8 +3,12 @@ module Convenience
 
   # shortcut for should.not.be.nil
   # has :page_title => assigns(:page_title).should.not.be.nil
-  def has(var)
-    assigns(var).should.not.be.nil
+  # OR you can also pass an array of values 
+  # has :page_title, :user # => assigns(:page_title).should.not.be.nil
+                           # => assigns(:user).should.not.be.nil 
+  def has(*var)
+    return assigns(var).should.not.be.nil if var.length == 1
+    var.each { |v| assigns(v).should.not.be.nil }
   end
 
   # shortcut for flash[:type].should.not.be.nil