aboutsummaryrefslogtreecommitdiffstats
path: root/spec/ruby/core/enumerable/shared/collect_concat.rb
blob: 54e10692eb600e6bd0f9d4292d1efe44ac959b4c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
require File.expand_path('../enumerable_enumeratorized', __FILE__)

describe :enumerable_collect_concat, shared: true do
  it "yields elements to the block and flattens one level" do
    numerous = EnumerableSpecs::Numerous.new(1, [2, 3], [4, [5, 6]], {foo: :bar})
    numerous.send(@method) { |i| i }.should == [1, 2, 3, 4, [5, 6], {foo: :bar}]
  end

  it "appends non-Array elements that do not define #to_ary" do
    obj = mock("to_ary undefined")

    numerous = EnumerableSpecs::Numerous.new(1, obj, 2)
    numerous.send(@method) { |i| i }.should == [1, obj, 2]
  end

  it "concatenates the result of calling #to_ary if it returns an Array" do
    obj = mock("to_ary defined")
    obj.should_receive(:to_ary).and_return([:a, :b])

    numerous = EnumerableSpecs::Numerous.new(1, obj, 2)
    numerous.send(@method) { |i| i }.should == [1, :a, :b, 2]
  end

  it "does not call #to_a" do
    obj = mock("to_ary undefined")
    obj.should_not_receive(:to_a)

    numerous = EnumerableSpecs::Numerous.new(1, obj, 2)
    numerous.send(@method) { |i| i }.should == [1, obj, 2]
  end

  it "appends an element that defines #to_ary that returns nil" do
    obj = mock("to_ary defined")
    obj.should_receive(:to_ary).and_return(nil)

    numerous = EnumerableSpecs::Numerous.new(1, obj, 2)
    numerous.send(@method) { |i| i }.should == [1, obj, 2]
  end

  it "raises a TypeError if an element defining #to_ary does not return an Array or nil"  do
    obj = mock("to_ary defined")
    obj.should_receive(:to_ary).and_return("array")

    lambda { [1, obj, 3].send(@method) { |i| i } }.should raise_error(TypeError)
  end

  it "returns an enumerator when no block given" do
    enum = EnumerableSpecs::Numerous.new(1, 2).send(@method)
    enum.should be_an_instance_of(Enumerator)
    enum.each{ |i| [i] * i }.should == [1, 2, 2]
  end

  it_should_behave_like :enumerable_enumeratorized_with_origin_size
end