aboutsummaryrefslogtreecommitdiffstats
path: root/spec/rubyspec/core/enumerator/feed_spec.rb
blob: 32ea77a30d4f4b5f5b6c111fc5f3db34b481266d (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
require File.expand_path('../../../spec_helper', __FILE__)
require File.expand_path('../fixtures/common', __FILE__)

describe "Enumerator#feed" do
  before :each do
    ScratchPad.record []
    @enum = EnumeratorSpecs::Feed.new.to_enum(:each)
  end

  it "sets the future return value of yield if called before advancing the iterator" do
    @enum.feed :a
    @enum.next
    @enum.next
    @enum.next
    ScratchPad.recorded.should == [:a, nil]
  end

  it "causes yield to return the value if called during iteration" do
    @enum.next
    @enum.feed :a
    @enum.next
    @enum.next
    ScratchPad.recorded.should == [:a, nil]
  end

  it "can be called for each iteration" do
    @enum.next
    @enum.feed :a
    @enum.next
    @enum.feed :b
    @enum.next
    ScratchPad.recorded.should == [:a, :b]
  end

  it "returns nil" do
    @enum.feed(:a).should be_nil
  end

  it "raises a TypeError if called more than once without advancing the enumerator" do
    @enum.feed :a
    @enum.next
    lambda { @enum.feed :b }.should raise_error(TypeError)
  end

  it "sets the return value of Yielder#yield" do
    enum = Enumerator.new { |y| ScratchPad << y.yield }
    enum.next
    enum.feed :a
    lambda { enum.next }.should raise_error(StopIteration)
    ScratchPad.recorded.should == [:a]
  end
end