summaryrefslogtreecommitdiff
path: root/spec/ruby/core/enumerable/slice_after_spec.rb
blob: 0e46688db1493caa98aa55be161d509d01d7a1d5 (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
require_relative '../../spec_helper'
require_relative 'fixtures/classes'

describe "Enumerable#slice_after" do
  before :each do
    @enum = EnumerableSpecs::Numerous.new(7, 6, 5, 4, 3, 2, 1)
  end

  describe "when given an argument and no block" do
    it "calls === on the argument to determine when to yield" do
      arg = mock("filter")
      arg.should_receive(:===).and_return(false, true, false, false, false, true, false)
      e = @enum.slice_after(arg)
      e.should be_an_instance_of(Enumerator)
      e.to_a.should == [[7, 6], [5, 4, 3, 2], [1]]
    end

    it "doesn't yield an empty array if the filter matches the first entry or the last entry" do
      arg = mock("filter")
      arg.should_receive(:===).and_return(true).exactly(7)
      e = @enum.slice_after(arg)
      e.to_a.should == [[7], [6], [5], [4], [3], [2], [1]]
    end

    it "uses standard boolean as a test" do
      arg = mock("filter")
      arg.should_receive(:===).and_return(false, :foo, nil, false, false, 42, false)
      e = @enum.slice_after(arg)
      e.to_a.should == [[7, 6], [5, 4, 3, 2], [1]]
    end
  end

  describe "when given a block" do
    describe "and no argument" do
      it "calls the block to determine when to yield" do
        e = @enum.slice_after{ |i| i == 6 || i == 2 }
        e.should be_an_instance_of(Enumerator)
        e.to_a.should == [[7, 6], [5, 4, 3, 2], [1]]
      end
    end

    describe "and an argument" do
      it "raises an ArgumentError" do
        -> { @enum.slice_after(42) { |i| i == 6 } }.should raise_error(ArgumentError)
      end
    end
  end

  it "raises an ArgumentError when given an incorrect number of arguments" do
    -> { @enum.slice_after("one", "two") }.should raise_error(ArgumentError)
    -> { @enum.slice_after }.should raise_error(ArgumentError)
  end
end

describe "when an iterator method yields more than one value" do
  it "processes all yielded values" do
    enum = EnumerableSpecs::YieldsMulti.new
    result = enum.slice_after { |i| i == [3, 4, 5] }.to_a
    result.should == [[[1, 2], [3, 4, 5]], [[6, 7, 8, 9]]]
  end
end