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

describe "Enumerator::Chain#rewind" do
  before(:each) do
    @obj = mock('obj')
    @obj.should_receive(:each).any_number_of_times.and_yield
    @second = mock('obj')
    @second.should_receive(:each).any_number_of_times.and_yield
    @enum = Enumerator::Chain.new(@obj, @second)
  end

  it "returns self" do
    @enum.rewind.should equal @enum
  end

  it "does nothing if receiver has not been iterated" do
    @obj.should_not_receive(:rewind)
    @obj.respond_to?(:rewind).should == true # sanity check
    @enum.rewind
  end

  it "does nothing on objects that don't respond_to rewind" do
    @obj.respond_to?(:rewind).should == false # sanity check
    @enum.each {}
    @enum.rewind
  end

  it "calls_rewind its objects" do
    @obj.should_receive(:rewind)
    @enum.each {}
    @enum.rewind
  end

  it "calls_rewind in reverse order" do
    @obj.should_not_receive(:rewind)
    @second.should_receive(:rewind).and_raise(RuntimeError)
    @enum.each {}
    -> { @enum.rewind }.should raise_error(RuntimeError)
  end

  it "calls rewind only for objects that have actually been iterated on" do
    @obj = mock('obj')
    @obj.should_receive(:each).any_number_of_times.and_raise(RuntimeError)
    @enum = Enumerator::Chain.new(@obj, @second)

    @obj.should_receive(:rewind)
    @second.should_not_receive(:rewind)
    -> { @enum.each {} }.should raise_error(RuntimeError)
    @enum.rewind
  end
end