Citrus

Squisher

class Squisher

  def initialize(items)
    @inputs = items
    @outputs = []
  end

  attr_reader :outputs, :inputs

  def build
    squish(inputs)
  end

  private

  def squish(items)
    items.each do |item|
      is_array?(item) ? squish(item) : add_to_output(item)
    end
  end

  def add_to_output(input)
    outputs.push(input)
  end

  def is_array?(input)
    input.class == Array
  end

end

context Squisher do

  describe "#squish" do

    it "a three level array should produce a flattened array" do
      input = [[1,2,[3]],4]
      squished = Squisher.new(input)
      squished.build
      expect(squished.outputs).to eq([1,2,3,4])
    end

  end

end