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

describe "Range#each" do
  it "passes each element to the given block by using #succ" do
    a = []
    (-5..5).each { |i| a << i }
    a.should == [-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5]

    a = []
    ('A'..'D').each { |i| a << i }
    a.should == ['A','B','C','D']

    a = []
    ('A'...'D').each { |i| a << i }
    a.should == ['A','B','C']

    a = []
    (0xfffd...0xffff).each { |i| a << i }
    a.should == [0xfffd, 0xfffe]

    y = mock('y')
    x = mock('x')
    x.should_receive(:<=>).with(y).any_number_of_times.and_return(-1)
    x.should_receive(:<=>).with(x).any_number_of_times.and_return(0)
    x.should_receive(:succ).any_number_of_times.and_return(y)
    y.should_receive(:<=>).with(x).any_number_of_times.and_return(1)
    y.should_receive(:<=>).with(y).any_number_of_times.and_return(0)

    a = []
    (x..y).each { |i| a << i }
    a.should == [x, y]
  end

  it "raises a TypeError if the first element does not respond to #succ" do
    lambda { (0.5..2.4).each { |i| i } }.should raise_error(TypeError)

    b = mock('x')
    (a = mock('1')).should_receive(:<=>).with(b).and_return(1)

    lambda { (a..b).each { |i| i } }.should raise_error(TypeError)
  end

  it "returns self" do
    range = 1..10
    range.each{}.should equal(range)
  end

  ruby_version_is "1.8.7" do
    it "returns an enumerator when no block given" do
      enum = (1..3).each
      enum.should be_an_instance_of(enumerator_class)
      enum.to_a.should == [1, 2, 3]
    end
  end

  ruby_version_is '1.9' do
    it "raises a TypeError if the first element is a Time object" do
      t = Time.now
      lambda { (t..t+1).each { |i| i } }.should raise_error(TypeError)
    end

    it "passes each Symbol element by using #succ" do
      (:aa..:ac).each.to_a.should == [:aa, :ab, :ac]
      (:aa...:ac).each.to_a.should == [:aa, :ab]
    end
  end

end