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

describe "Enumerator#next" do
  before :each do
    @enum = 1.upto(3)
  end

  it "returns the next element of the enumeration" do
    @enum.next.should == 1
    @enum.next.should == 2
    @enum.next.should == 3
  end

  it "raises a StopIteration exception at the end of the stream" do
    3.times { @enum.next }
    -> { @enum.next }.should raise_error(StopIteration)
  end

  it "cannot be called again until the enumerator is rewound" do
    3.times { @enum.next }
    -> { @enum.next }.should raise_error(StopIteration)
    -> { @enum.next }.should raise_error(StopIteration)
    -> { @enum.next }.should raise_error(StopIteration)
    @enum.rewind
    @enum.next.should == 1
  end

  it "restarts the enumerator if an exception terminated a previous iteration" do
    exception = StandardError.new
    enum = Enumerator.new do
      raise exception
    end

    result = 2.times.map { enum.next rescue $! }

    result.should == [exception, exception]
  end
end