Add matcher to match elements by ids

Initially added in gitlab-org/gitlab-ee!4864 and
gitlab-org/gitlab-ee!4689.
This commit is contained in:
Toon Claes 2018-03-06 20:42:14 +01:00
parent 35f6efaee0
commit b0e5906d19
1 changed files with 24 additions and 0 deletions

View File

@ -0,0 +1,24 @@
RSpec::Matchers.define :match_ids do |*expected|
match do |actual|
actual_ids = map_ids(actual)
expected_ids = map_ids(expected)
expect(actual_ids).to match_array(expected_ids)
end
description do
'matches elements by ids'
end
def map_ids(elements)
elements = elements.flatten if elements.respond_to?(:flatten)
if elements.respond_to?(:map)
elements.map(&:id)
elsif elements.respond_to?(:id)
[elements.id]
else
raise ArgumentError, "could not map elements to ids: #{elements}"
end
end
end