def test_findByDate(self):
activityRepo = ActivitiesRepository()
activityRepo.add(Activity(1, datetime(2014, 11, 21, 12, 30), "test1"))
activityRepo.add(Activity(2, datetime(1999, 9, 15, 19, 25), "test2"))
activityRepo.add(Activity(3, datetime(2015, 1, 31, 2, 10), "test3"))
activityRepo.add(Activity(4, datetime(2015, 1, 31, 12, 36), "test4"))
self.assertEqual(activityRepo.findByDate(date(2015, 1, 31)), [Activity(3, datetime(2015, 1, 31, 2, 10), "test3"),
Activity(4, datetime(2015, 1, 31, 12, 36), "test4")])
def test_remove(self):
activityRepo = ActivitiesRepository()
activityRepo.add(Activity(1, datetime(2014, 11, 21, 12, 30), "test1"))
activityRepo.add(Activity(2, datetime(1999, 9, 15, 19, 25), "test2"))
activityRepo.add(Activity(3, datetime(2015, 1, 31, 2, 10), "test3"))
self.assertEqual(activityRepo.remove(3), Activity(3, datetime(2015, 1, 31, 2, 10), "test3"))
def test_add(self):
activityRepo = ActivitiesRepository()
activityRepo.add(Activity(1, datetime(2014, 11, 21, 12, 30), "test1"))
self.assertIn(Activity(1, datetime(2014, 11, 21, 12, 30), "test1"), activityRepo.getAll())
def test_findById(self):
activityRepo = ActivitiesRepository()
activityRepo.add(Activity(1, datetime(2014, 11, 21, 12, 30), "test1"))
self.assertEqual(activityRepo.findById(1), Activity(1, datetime(2014, 11, 21, 12, 30), "test1"))
self.assertIsNone(activityRepo.findById(3))
