updated the that operator's contains? method

FrozenCanuck · Dec 6, 2010 · 192cdab · 192cdab
1 parent 44fe475
commit 192cdab
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/lebowski/rspec/operators/that.rb b/lib/lebowski/rspec/operators/that.rb
@@ -96,9 +96,15 @@ def less_than?(value)
 
         def contains?(value)
           return false if value.nil?
-          @args.all? do |x|
-            value.member? x
+          if value.kind_of? String
+            return false if @args.length > 1
+            return value == @args[0]
+          elsif value.kind_of? Array
+            return @args.all? do |x|
+              value.member? x
+            end
           end
+          return false
         end
 
         def matches?(value)