def test_aggregate_normal(self):
record_work_time.record_work_times(
record_work_time.convert_work_times(
u'projectB', u'20101027',
[u'900', u'1300', u'1400', u'1900']),
self.conn)
record_work_time.record_work_times(
record_work_time.convert_work_times(
u'projectA', u'20101128',
[u'900', u'1200', u'1300', u'1800']),
self.conn)
record_work_time.record_work_times(
record_work_time.convert_work_times(
u'projectB', u'20101129',
[u'900', u'1300', u'1400', u'1900']),
self.conn)
record_work_time.record_work_times(
record_work_time.convert_work_times(
u'projectA', u'20101130',
[u'900', u'1200', u'1300', u'1800']),
self.conn)
times = aggregate_work_time.aggregate_work_time(
u'2010-10-01', u'2010-11-30', self.conn)
self.assertEquals(3, len(times))
self.assert_time(u'2010-10', u'projectB', 9, times[0])
self.assert_time(u'2010-11', u'projectA', 16, times[1])
self.assert_time(u'2010-11', u'projectB', 9, times[2])
def test_record_a_work_time(self):
project = u'project'
work_times = record_work_time.convert_work_times(
project, u'20101128', [u'910', u'1250'])
record_work_time.record_work_times(work_times, self.conn)
cursor = self.conn.execute(
u'select {project}, {day}, {start}, {end} from {table}'.format(
project = work_recorder.COLUMN_PROJECT,
day = work_recorder.COLUMN_DAY,
start = work_recorder.COLUMN_START,
end = work_recorder.COLUMN_END,
table = work_recorder.TABLE_WORK_TIME))
result = cursor.fetchall()
self.assertEquals(1, len(result))
a_row = result[0]
self.assertEquals(
(project, u'2010-11-28', u'09:10:00', u'12:50:00'),
a_row)
