summaryrefslogtreecommitdiff
path: root/spec/ruby/core/enumerator/plus_spec.rb
blob: c9bae08b042a2e1b9d9b90c737254c0d30d51bab (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
require_relative '../../spec_helper'

ruby_version_is "2.6" do
  describe "Enumerator#+" do
    before :each do
      ScratchPad.record []
    end

    it "returns a chain of self and provided enumerators" do
      one   = Enumerator.new { |y| y << 1 }
      two   = Enumerator.new { |y| y << 2 }
      three = Enumerator.new { |y| y << 3 }

      chain = one + two + three

      chain.should be_an_instance_of(Enumerator::Chain)
      chain.each { |item| ScratchPad << item }
      ScratchPad.recorded.should == [1, 2, 3]
    end

    it "calls #each on each argument" do
      enum = Enumerator.new { |y| y << "one" }

      obj1 = mock("obj1")
      obj1.should_receive(:each).once.and_yield("two")

      obj2 = mock("obj2")
      obj2.should_receive(:each).once.and_yield("three")

      chain = enum + obj1 + obj2
      chain.each { |item| ScratchPad << item }
      ScratchPad.recorded.should == ["one", "two", "three"]
    end
  end
end