def test__export_as_csv(self):
t = DbObject(DBH.test)
t.tbl_.truncate()
t.db_.commit()
t.company_id = COMPANY_ID
t.name = NAME
t.number = NUMBER
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add()) # Add succeeds
output = cStringIO.StringIO()
t.export_as_csv(out=output)
# String fields are double quoted.
self.assertEqual(
"""%s,%s,%s,"%s",%s,"%s","%s","%s","%s"\r\n"""
% (t.id, COMPANY_ID, NUMBER, NAME, AMOUNT, START_DATE, STATUS, t.created_on, t.updated_on),
output.getvalue(),
)
t.tbl_.truncate()
t.db_.commit()
return
