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