require_relative '../../spec_helper'

ruby_version_is "2.6" do
  describe "Enumerator#+" do
    before :each do
      ScratchPad.record []
    end

    it "returns a chain of self and provided enumerators" do
      one   = Enumerator.new { |y| y << 1 }
      two   = Enumerator.new { |y| y << 2 }
      three = Enumerator.new { |y| y << 3 }

      chain = one + two + three

      chain.should be_an_instance_of(Enumerator::Chain)
      chain.each { |item| ScratchPad << item }
      ScratchPad.recorded.should == [1, 2, 3]
    end

    it "calls #each on each argument" do
      enum = Enumerator.new { |y| y << "one" }

      obj1 = mock("obj1")
      obj1.should_receive(:each).once.and_yield("two")

      obj2 = mock("obj2")
      obj2.should_receive(:each).once.and_yield("three")

      chain = enum + obj1 + obj2
      chain.each { |item| ScratchPad << item }
      ScratchPad.recorded.should == ["one", "two", "three"]
    end
  end
end