Add orderline to order and calculate order total

app.rb

@@ -1,13 +1,14 @@
#!/usr/bin/ruby

require_relative './lib/order'
require_relative './lib/handle_input'

command = HandleInput.new

$stdout.print "Would you like to LIST available products, SHOP, VIEW basket, EXIT without placing an order? \n"
command = HandleInput.new(Order.new)

# Read user input
loop do
  $stdout.print "Would you like to LIST available products, SHOP, VIEW basket, EXIT without placing an order? \n"

  input = gets.chomp

  next if input.empty?

lib/handle_input.rb

@@ -1,11 +1,18 @@
require_relative 'item'
# require_relative 'order'
require_relative 'order_line'

# HandleInput class
class HandleInput
  attr_reader :order

  # Actions
  ACTIONS = %w[LIST SHOP VIEW EXIT].freeze

  def initialize(order)
    @order = order
  end

  # Interpret method
  # @param command [String]
  # @return
@@ -18,7 +25,26 @@ def interpret(command)
      # list all availble items and package sizes
    end

    return shop if command == 'SHOP'
    # return shop if command == 'SHOP'
    if command == 'SHOP'
      $stdout.print "Select qty and items, example: 3 watermelon \n"

      loop do
        input = $stdin.gets.chomp

        next if input.empty?

        unless 'BACK'.match?(input)
          shop(input)
          next
        end

        $stdout.print "Returning to main menu \n"
        break
      end
    end

    # loop { shop if command == 'SHOP' }

    if command == 'VIEW'
      # show invoice
@@ -29,13 +55,8 @@ def interpret(command)
    end
  end

  def shop
    # retrieve another input
    $stdout.print "Select qty and items, example: 3 watermelon \n"

    input = $stdin.gets.chomp

    return if input.empty? # TODO: OR if it doesn't match a specific order pattern
  def shop(input)
    # TODO: OR if it doesn't match a specific order pattern

    # break it down into qty, item
    line = input.split(/\W+/)
@@ -51,6 +72,14 @@ def shop
    # Present order_line for invoice
    order_line.present_line(product)

    # order = Order.new

    add_to_order = @order.add_item(order_line)

    # puts add_to_order

    puts order.find_order_total

    order_line
  end
end

lib/order.rb

@@ -1,9 +1,21 @@
# Order class
class Order
  attr_reader :items, :basket
  attr_reader :items

  def initialize(basket)
  def initialize
    @items = []
    @basket = basket
  end

  def add_item(order_item)
    @items.push(order_item)
  end

  def find_order_total
    total = 0
    @items.each do |item|
      total += item.total_quantity_of_items
    end

    total
  end
end

lib/order_line.rb

@@ -71,12 +71,17 @@ def present_line(whole_packs)
      sub_total << k * i
    end

    total = sub_total.inject(0) { |sum, x| sum + x }
    @line_total = sub_total.inject(0) { |sum, x| sum + x }

    puts "#{@order_qty} #{@order_item.name} #{total}"
    puts "#{@order_qty} #{@order_item.name} #{@line_total}"

    sub_item.each do |s|
      puts s
    end
  end

  def total_quantity_of_items
    # @order_qty * @line_total
    @line_total
  end
end

spec/handle_input_spec.rb

@@ -5,7 +5,7 @@
describe HandleInput do
  describe '#interpret' do
    it 'should return nil unless command matches a value in ACTIONS array' do
      instance = HandleInput.new
      instance = HandleInput.new(Order.new)

      expect(instance.interpret('hello')).to be_nil
    end
@@ -17,9 +17,9 @@
    end

    it 'should return an instance of OrderLine' do
      instance = HandleInput.new
      instance = HandleInput.new(Order.new)

      expect(instance.shop).to be_a OrderLine
      expect(instance.shop('10 watermelon')).to be_a OrderLine
    end
  end
end

spec/item_spec.rb

@@ -19,4 +19,12 @@
      expect(instance.packs('watermelon')).to be_a Array
    end
  end

  describe '#packs' do
    it 'should nil unless fruit is valid' do
      instance = Item.new('Watermelons')

      expect(instance.packs('hello')).to be_nil
    end
  end
end

spec/order_spec.rb

@@ -4,17 +4,28 @@

describe Order do
  describe '#initialize' do
    it 'should have a items attribute which is an empty Array' do
      instance = Order.new(Basket.new)
    it 'should have an items attribute which is an empty Array' do
      instance = Order.new

      expect(instance.items).to be_a Array
      expect(instance.items).to be_empty
    end
  end

  describe '#add_item' do
    it 'should return an Array of items' do
      instance = Order.new
      orderline = OrderLine.new(10, Item.new('watermelon'))

      expect(instance.add_item(orderline)).to be_a Array
    end
  end

    it 'should have a basket attribute which is an instance of Basket' do
      instance = Order.new(Basket.new)
  describe '#find_order_total' do
    it 'should return 0 if there are no items in the items array' do
      instance = Order.new

      expect(instance.basket).to be_a Basket
      expect(instance.find_order_total).to eq 0
    end
  end
end