41 lines
956 B
Ruby
41 lines
956 B
Ruby
RSpec::Matchers.define :contain_tag do |klass|
|
|
match do |collection|
|
|
if @num.blank?
|
|
collection.any? {|tag| tag.is_a? klass}
|
|
else
|
|
(@count = collection.count {|tag| tag.is_a? klass}) == @num
|
|
end
|
|
end
|
|
|
|
def count(num)
|
|
@num = num
|
|
self
|
|
end
|
|
|
|
description do
|
|
"contain #{@num || 'any'} instance(s) of #{klass.name}"
|
|
end
|
|
failure_message_for_should do |collection|
|
|
"expected #{@num || 'any'} instance(s) of #{klass.name} but was #{@count}"
|
|
end
|
|
end
|
|
|
|
RSpec::Matchers.define :contain_tag_old do |count|
|
|
match do |collection|
|
|
(@count = collection.count {|tag| tag.is_a? @klass}) == count
|
|
end
|
|
|
|
def instance_of(klass)
|
|
@klass = klass
|
|
self
|
|
end
|
|
alias :instances_of :instance_of
|
|
|
|
description do
|
|
"contain #{count || 'any'} instance(s) of #{@klass.name}"
|
|
end
|
|
failure_message_for_should do |collection|
|
|
"expected #{count || 'any'} instance(s) of #{@klass.name} but was #{@count}"
|
|
end
|
|
end
|