31 lines
778 B
Ruby
31 lines
778 B
Ruby
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
|
|
|
|
failure_message do
|
|
actual_ids = map_ids(actual)
|
|
expected_ids = map_ids(expected)
|
|
|
|
"expected IDs #{actual_ids} in:\n\n #{actual.inspect}\n\nto match IDs #{expected_ids} in:\n\n #{expected.inspect}"
|
|
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
|