test/test_ext.rb in invoice_printer-0.0.2 vs test/test_ext.rb in invoice_printer-0.0.3

@@ -1,93 +1,125 @@
+# Requiring this file extends the original PDFDocument class with +to_a+ method
+# returning the strings representation of the class.
+
 module InvoicePrinter
   class PDFDocument
     # Expose the document as an array of attributes in order as their
     # appear on PDF
     def to_a
       strings = []
-
       strings << @labels[:name]
       strings << @document.number
+      strings << provider_box
+      strings << purchaser_box
+      strings << account_box
+      strings << dates_box
+      strings << items_table
+      strings << totals_table
+      strings.flatten.reject(&:empty?)
+    end
 
+    private
+
+    # Strings representaion of provider's box
+    def provider_box
+      strings = []
       strings << @labels[:provider]
       strings << @document.provider_name
       strings << "#{@document.provider_street}    #{@document.provider_street_number}".strip
       strings << @document.provider_postcode
       strings << @document.provider_city
       strings << @document.provider_city_part
       strings << @document.provider_extra_address_line
+      strings << "#{@labels[:ic]}:    #{@document.provider_ic}" \
+        unless @document.provider_ic.empty?
+      strings << "#{@labels[:dic]}:    #{@document.provider_dic}" \
+        unless @document.provider_dic.empty?
+      strings
+    end
 
-      strings << "#{@labels[:ic]}:    #{@document.provider_ic}" unless @document.provider_ic.empty?
-      strings << "#{@labels[:dic]}:    #{@document.provider_dic}" unless @document.provider_dic.empty?
-
+    # Strings representaion of purchaser's box
+    def purchaser_box
+      strings = []
       strings << @labels[:purchaser]
       strings << @document.purchaser_name
       strings << "#{@document.purchaser_street}    #{@document.purchaser_street_number}".strip
       strings << @document.purchaser_postcode
       strings << @document.purchaser_city
       strings << @document.purchaser_city_part
       strings << @document.purchaser_extra_address_line
+      strings << "#{@labels[:ic]}:    #{@document.purchaser_ic}" \
+        unless @document.purchaser_ic.empty?
+      strings << "#{@labels[:dic]}:    #{@document.purchaser_dic}" \
+        unless @document.purchaser_dic.empty?
+      strings
+    end
 
-      strings << "#{@labels[:ic]}:    #{@document.purchaser_ic}" unless @document.purchaser_ic.empty?
-      strings << "#{@labels[:dic]}:    #{@document.purchaser_dic}" unless @document.purchaser_dic.empty?
-
-      # Account
+    # Strings representaion of account's box
+    def account_box
+      strings = []
       if @document.bank_account_number.nil?
         strings << @labels[:payment_in_cash]
       else
         strings << @labels[:payment_by_transfer]
       end
-
       strings << "#{@labels[:account_number]}:"
       strings << @document.bank_account_number
       strings << "#{@labels[:swift]}:"
       strings << @document.account_swift
       strings << "#{@labels[:iban]}:"
       strings << @document.account_iban
+      strings
+    end
 
-      # Dates
+    # Strings representaion of dates box
+    def dates_box
+      strings = []
       strings << "#{@labels[:issue_date]}:"
       strings << @document.issue_date
       strings << "#{@labels[:due_date]}:"
       strings << @document.due_date
+      strings
+    end
 
-      # Items table
+    # Strings representaion of items table
+    def items_table
+      strings = []
       strings << @labels[:item] if determine_items_structure[:names]
       strings << @labels[:quantity] if determine_items_structure[:quantities]
       strings << @labels[:unit] if determine_items_structure[:units]
       strings << @labels[:price_per_item] if determine_items_structure[:prices]
       strings << @labels[:tax] if determine_items_structure[:taxes]
       strings << @labels[:amount] if determine_items_structure[:amounts]
       strings << items_to_a(@document.items)
+      strings
+    end
 
-      # Total table
+    # Strings representaion of totals table
+    def totals_table
+      strings = []
       strings << "#{@labels[:subtotal]}:"
       strings << @document.subtotal
       strings << "#{@labels[:tax]}:"
       strings << @document.tax
       strings << "#{@labels[:tax2]}:"
       strings << @document.tax2
       strings << "#{@labels[:tax3]}:"
       strings << @document.tax3
-
       strings << "#{@labels[:total]}:   #{@document.total}"
-
-      # TODO: dynamically test page numbers
-      strings << '1 / 1'
-
-      strings.flatten.reject(&:empty?)
+      strings
     end
 
-    private
-
+    # Convert document +items+ to a single string array
     def items_to_a(items)
       ary = []
       items.each do |item|
         ary << item_to_a(item)
       end
       ary.flatten
     end
 
+    # Convert document +item+ to a single string array
     def item_to_a(item)
       ary = []
       ary << item.name
       ary << item.quantity
       ary << item.unit