require 'spec_helper'

describe Ru::Iterator do
  context "Array" do
    describe "#to_a" do
      it "returns the array" do
        iterator = described_class.new(%w{john paul george ringo})
        expect(iterator.to_a).to eq(%w{john paul george ringo})
      end
    end

    describe "#to_stdout" do
      it "returns the string" do
        iterator = described_class.new(%w{john paul george ringo})
        expect(iterator.to_stdout).to eq("john\npaul\ngeorge\nringo")
      end

      context "with a method called on it" do
        it "returns the string" do
          iterator = described_class.new(%w{john paul george ringo})
          expect(iterator.to_s.to_stdout).to eq("john\npaul\ngeorge\nringo")
        end
      end
    end
  end

  context "Ru::Array" do
    describe "#to_a" do
      it "returns the array" do
        iterator = described_class.new(Ru::Array.new(%w{john paul george ringo}))
        expect(iterator.to_a).to eq(%w{john paul george ringo})
      end
    end

    describe "#to_stdout" do
      it "returns the string" do
        iterator = described_class.new(Ru::Array.new(%w{john paul george ringo}))
        expect(iterator.to_stdout).to eq("john\npaul\ngeorge\nringo")
      end

      context "with a method called on it" do
        it "returns the string" do
          iterator = described_class.new(Ru::Array.new(%w{john paul george ringo}))
          expect(iterator.to_s.to_stdout).to eq("john\npaul\ngeorge\nringo")
        end
      end
    end
  end

  context "Ru::Stream" do
    describe "#to_a" do
      it "returns the array" do
        iterator = described_class.new(Ru::Stream.new(%w{john paul george ringo}.each.lazy))
        expect(iterator.to_a).to eq(%w{john paul george ringo})
      end
    end

    describe "#to_stdout" do
      it "returns the string" do
        iterator = described_class.new(Ru::Stream.new(%w{john paul george ringo}.each.lazy))
        expect(iterator.to_stdout).to eq("john\npaul\ngeorge\nringo")
      end

      context "with a method called on it" do
        it "returns the string" do
          iterator = described_class.new(Ru::Stream.new(%w{john paul george ringo}.each.lazy))
          expect(iterator.to_s.to_stdout).to eq("john\npaul\ngeorge\nringo")
        end
      end
    end
  end
end