class TimesheetTestCase(unittest.TestCase):
def setUp(self):
self.timesheet = TimesheetManager(os.path.join(root, "tests/example.yaml"))
def test_meta(self):
self.assertEqual(self.timesheet.get_project_name(), "Example Project")
def test_entries(self):
self.assertEqual(self.timesheet.get_total_entries(), 3)
def test_total_hours(self):
self.assertEqual(self.timesheet.get_total_hours(), 6.5)
def test_hours_within(self):
f = datetime.datetime(2012, 7, 1)
t = datetime.datetime(2012, 8, 1)
self.assertEqual(self.timesheet.get_hours_within(f, t), 6.5)
f = datetime.datetime(2012, 7, 1)
t = datetime.datetime(2012, 7, 31)
self.assertEqual(self.timesheet.get_hours_within(f, t), 3.0)
def test_add_hours(self):
f = datetime.datetime(2012, 8, 1, 10, 0)
t = datetime.datetime(2012, 8, 1, 12, 0)
self.timesheet.add_hours(f, t, desc="Testing")
self.assertEqual(self.timesheet.get_total_entries(), 4)
f = datetime.datetime(2012, 8, 1)
t = datetime.datetime(2012, 8, 31)
self.assertEqual(self.timesheet.get_hours_within(f, t), 2.0)
def setUp(self):
self.timesheet = TimesheetManager(os.path.join(root, "tests/example.yaml"))