def import_csv(session, filepath, bankaccount):
tx_imported = 0
if bankaccount:
with open(filepath, "r") as csvfile:
batch = ImportBatch.from_csv(session, csvfile)
if batch is not None:
batch.bank_account = bankaccount
session.add(batch)
tx_imported = len(batch.transactions)
session.commit()
return tx_imported
def test_from_csv_invalid(self):
self.assertIsNone(ImportBatch.from_csv(self.session, StringIO.StringIO("")))
self.assertIsNone(ImportBatch.from_csv(self.session, StringIO.StringIO("sdfj, asdfakfvankvdsj")))
def test_from_csv(self):
self.session.query.return_value.filter_by.return_value.count.return_value = 0
csvdata = StringIO.StringIO("""06/09/2013,"-400.00",description""")
batch = ImportBatch.from_csv(self.session, csvdata)
self.assertEqual(1, len(batch.transactions))
