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

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