From 192cdab87fdb682a7d6854e78a5ea7cb18837067 Mon Sep 17 00:00:00 2001 From: Michael Cohen Date: Mon, 6 Dec 2010 00:23:14 -0500 Subject: [PATCH] updated the that operator's contains? method --- lib/lebowski/rspec/operators/that.rb | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/lib/lebowski/rspec/operators/that.rb b/lib/lebowski/rspec/operators/that.rb index 99a8003..3ad5a9d 100644 --- 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)