Merge pull request #3 from jonbartlett/master

support for writing multiple Account entries

lib/qif/writer.rb

@@ -60,7 +60,7 @@ def initialize(io, type = 'Bank', format = 'dd/mm/yyyy')
       end
     end
 
-    # Add a transaction for writing
+    # Add a QIF entry for writing
     def <<(transaction)
       case transaction.class.to_s
       when "Qif::Transaction"
@@ -89,7 +89,9 @@ def write_header
     end
 
     def write_account
-      write_record(@accounts.first.to_s) # only one account per file
+      @accounts.each do |a|
+        write_record(a)
+      end
     end
 
     def write_transactions

spec/lib/writer_spec.rb

@@ -60,6 +60,17 @@
       @buffer.should include('NExpenses:Eating and Drinking:Coffee')
     end
 
+    it 'should write multiple accounts' do
+
+      Qif::Writer.new(@io) do |writer|
+        writer << Qif::Account.new(:name => 'Expenses:Eating and Drinking:Coffee')
+        writer << Qif::Account.new(:name => 'Expenses:Eating and Drinking:Tea')
+      end
+
+      @buffer.should include('NExpenses:Eating and Drinking:Coffee')
+      @buffer.should include('NExpenses:Eating and Drinking:Tea')
+    end
+
     it 'should write the transactions' do
       date = Time.now