def test_save_and_setUp_Projects(self): self.expense.setUp({'test':True}) proj1 = Projects.ConsumingProject(9999, date(2012, 1, 1), 3333) proj2 = Projects.ConsumingProject( 999, date(2012, 1, 1), 3333, 'type', u'Food & Drinks' ) self.expense.addProject('proj1', proj1) self.expense.addProject('proj2', proj2) addSampleRecords(self.expense) self.expense.saveProjects() expense2 = Expense() expense2.setUp({'test':True}) self.assertEqual(len(expense2.projects), 2) self.assertDictEqual( expense2.projects['proj1'].export_to_dict(), self.expense.projects['proj1'].export_to_dict() ) self.assertDictEqual( expense2.projects['proj2'].export_to_dict(), self.expense.projects['proj2'].export_to_dict() ) Config.setProjectBuffer(dict())
def test_getBufferObj_and_setBufferObj(self): projbuf = { 'python': 6, 'tcl': 3, 'ruby': 4, 'lua': 3, 'perl': 4, } Config.setProjectBuffer(projbuf) bufobj = Config.getProjectBuffer() self.assertDictEqual(bufobj, projbuf) Config.setProjectBuffer({}) rec = BaseRecord(1, (u'Food & Drinks', u'Snacks'), u'Cash', u'CHY') recbuf = [ [ '2012-01-03', rec.amount, rec.type[0], rec.type[1], rec.payment, rec.currency, rec.tag, rec.comment ] ] Config.setRecordBuffer(recbuf) bufobj = Config.getRecordBuffer() self.assertListEqual(bufobj, recbuf) Config.setRecordBuffer([])