def test_should_get_jobs_older_than_seven_days(notify_db, notify_db_session): one_millisecond_before_midnight = datetime(2016, 10, 9, 23, 59, 59, 999) midnight = datetime(2016, 10, 10, 0, 0, 0, 0) one_millisecond_past_midnight = datetime(2016, 10, 10, 0, 0, 0, 1) job_1 = create_job(notify_db, notify_db_session, created_at=one_millisecond_before_midnight) create_job(notify_db, notify_db_session, created_at=midnight) create_job(notify_db, notify_db_session, created_at=one_millisecond_past_midnight) with freeze_time('2016-10-17T00:00:00'): jobs = dao_get_jobs_older_than(7) assert len(jobs) == 1 assert jobs[0].id == job_1.id
def remove_csv_files(): jobs = dao_get_jobs_older_than(7) for job in jobs: s3.remove_job_from_s3(job.service_id, job.id) current_app.logger.info("Job ID {} has been removed from s3.".format(job.id))