[Feature #20702]

Works the same way than `Hash#fetch_values` for for array.
This commit is contained in:
Jean Boussier 2024-09-05 20:00:32 +02:00
Родитель 214668fccb
Коммит bc85c8d852
2 изменённых файлов: 70 добавлений и 0 удалений

Просмотреть файл

@ -210,4 +210,26 @@ class Array
end
end
end
# call-seq:
# array.fetch_values(*indexes) -> new_array
# array.fetch_values(*indexes) {|key| ... } -> new_array
#
# Returns a new Array containing the values associated with the given indexes *indexes:
# a = [:foo, :bar, :baz]
# a.fetch_values(3, 1) # => [:baz, :foo]
#
# Returns a new empty Array if no arguments given.
#
# When a block is given, calls the block with each missing index,
# treating the block's return value as the value for that index:
# a = [:foo, :bar, :baz]
# values = a.fetch_values(1, 0, 42, 777) {|index| index.to_s}
# values # => [:bar, :foo, "42", "777"]
#
# When no block is given, raises an exception if any given key is not found.
def fetch_values(*indexes, &block)
indexes.map! { |i| fetch(i, &block) }
indexes
end
end

Просмотреть файл

@ -0,0 +1,48 @@
require_relative '../../spec_helper'
require_relative 'fixtures/classes'
describe "Array#fetch_values" do
before :each do
@array = [:a, :b, :c]
end
ruby_version_is "3.4" do
describe "with matched indexes" do
it "returns the values for indexes" do
@array.fetch_values(0).should == [:a]
@array.fetch_values(0, 2).should == [:a, :c]
end
it "returns the values for indexes ordered in the order of the requested indexes" do
@array.fetch_values(2, 0).should == [:c, :a]
end
end
describe "with unmatched indexes" do
it "raises a index error if no block is provided" do
-> { @array.fetch_values(0, 1, 44) }.should raise_error(IndexError)
end
it "returns the default value from block" do
@array.fetch_values(44) { |index| "`#{index}' is not found" }.should == ["`44' is not found"]
@array.fetch_values(0, 44) { |index| "`#{index}' is not found" }.should == [:a, "`44' is not found"]
end
end
describe "without keys" do
it "returns an empty Array" do
@array.fetch_values.should == []
end
end
it "tries to convert the passed argument to an Integer using #to_int" do
obj = mock('to_int')
obj.should_receive(:to_int).and_return(2)
@array.fetch_values(obj).should == [:c]
end
it "raises a TypeError when the passed argument can't be coerced to Integer" do
-> { [].fetch_values("cat") }.should raise_error(TypeError)
end
end
end