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([])
