summaryrefslogtreecommitdiff
path: root/spec/ruby/core/enumerable/cycle_spec.rb
blob: 2db1ae56fe4026c13313fbcb2b604859f939596b (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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
require_relative '../../spec_helper'
require_relative 'fixtures/classes'
require_relative 'shared/enumeratorized'

describe "Enumerable#cycle" do
  describe "passed no argument or nil" do
    it "loops indefinitely" do
      [[],[nil]].each do |args|
        bomb = 10
        EnumerableSpecs::Numerous.new.cycle(*args) do
          bomb -= 1
          break 42 if bomb <= 0
        end.should == 42
        bomb.should == 0
      end
    end

    it "returns nil if there are no elements" do
      out = EnumerableSpecs::Empty.new.cycle { break :nope }
      out.should be_nil
    end

    it "yields successive elements of the array repeatedly" do
      b = []
      EnumerableSpecs::Numerous.new(1,2,3).cycle do |elem|
        b << elem
        break if b.size == 7
      end
      b.should == [1,2,3,1,2,3,1]
    end

    it "calls each at most once" do
      enum = EnumerableSpecs::EachCounter.new(1, 2)
      enum.cycle.first(6).should == [1,2,1,2,1,2]
      enum.times_called.should == 1
    end

    it "yields only when necessary" do
      enum = EnumerableSpecs::EachCounter.new(10, 20, 30)
      enum.cycle { |x| break if x == 20}
      enum.times_yielded.should == 2
    end
  end

  describe "passed a number n as an argument" do
    it "returns nil and does nothing for non positive n" do
      EnumerableSpecs::ThrowingEach.new.cycle(0) {}.should be_nil
      EnumerableSpecs::NoEach.new.cycle(-22) {}.should be_nil
    end

    it "calls each at most once" do
      enum = EnumerableSpecs::EachCounter.new(1, 2)
      enum.cycle(3).to_a.should == [1,2,1,2,1,2]
      enum.times_called.should == 1
    end

    it "yields only when necessary" do
      enum = EnumerableSpecs::EachCounter.new(10, 20, 30)
      enum.cycle(3) { |x| break if x == 20}
      enum.times_yielded.should == 2
    end

    it "tries to convert n to an Integer using #to_int" do
      enum = EnumerableSpecs::Numerous.new(3, 2, 1)
      enum.cycle(2.3).to_a.should == [3, 2, 1, 3, 2, 1]

      obj = mock('to_int')
      obj.should_receive(:to_int).and_return(2)
      enum.cycle(obj).to_a.should == [3, 2, 1, 3, 2, 1]
    end

    it "raises a TypeError when the passed n cannot be coerced to Integer" do
      enum = EnumerableSpecs::Numerous.new
      lambda{ enum.cycle("cat"){} }.should raise_error(TypeError)
    end

    it "raises an ArgumentError if more arguments are passed" do
      enum = EnumerableSpecs::Numerous.new
      lambda{ enum.cycle(1, 2) {} }.should raise_error(ArgumentError)
    end

    it "gathers whole arrays as elements when each yields multiple" do
      multi = EnumerableSpecs::YieldsMulti.new
      multi.cycle(2).to_a.should == [[1, 2], [3, 4, 5], [6, 7, 8, 9], [1, 2], [3, 4, 5], [6, 7, 8, 9]]
    end
  end

  describe "Enumerable with size" do
    before :all do
      @object = EnumerableSpecs::NumerousWithSize.new(1, 2, 3, 4)
      @empty_object = EnumerableSpecs::EmptyWithSize.new
    end
    it_should_behave_like :enumeratorized_with_cycle_size
  end

  describe "Enumerable with no size" do
    before :all do
      @object = EnumerableSpecs::Numerous.new(1, 2, 3, 4)
      @method = :cycle
    end
    it_should_behave_like :enumeratorized_with_unknown_size
  end

end