summaryrefslogtreecommitdiff
path: root/spec/ruby/core/enumerator/lazy/force_spec.rb
blob: 03ff9a0fb6e4055ed51f0f5d1ee182208495e80d (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
# -*- encoding: us-ascii -*-

require File.expand_path('../../../../spec_helper', __FILE__)
require File.expand_path('../fixtures/classes', __FILE__)

describe "Enumerator::Lazy#force" do
  before :each do
    @yieldsmixed = EnumeratorLazySpecs::YieldsMixed.new.to_enum.lazy
    @eventsmixed = EnumeratorLazySpecs::EventsMixed.new.to_enum.lazy
    ScratchPad.record []
  end

  after :each do
    ScratchPad.clear
  end

  it "passes given arguments to receiver.each" do
    @yieldsmixed.force(:arg1, :arg2, :arg3).should ==
      EnumeratorLazySpecs::YieldsMixed.gathered_yields_with_args(:arg1, :arg2, :arg3)
  end

  describe "on a nested Lazy" do
    it "calls all block and returns an Array" do
      (0..Float::INFINITY).lazy.map(&:succ).take(2).force.should == [1, 2]

      @eventsmixed.take(1).map(&:succ).force.should == [1]
      ScratchPad.recorded == [:after_yields]
    end
  end
end