def create_validation_task(self, generate_before=True, tuple_output=True, include_nonstate_changes=True, earliest_timestamp=None, expected_validation=None): """Create a task for testing purposes.""" interval = '2013-01-01-2014-10-10' interval_value = luigi.DateIntervalParameter().parse(interval) earliest_timestamp_value = luigi.DateHourParameter().parse( earliest_timestamp) if earliest_timestamp else None expected_validation_value = ( luigi.DateHourParameter().parse(expected_validation) if expected_validation else None) self.task = CourseEnrollmentValidationTask( interval=interval_value, output_root="/fake/output", generate_before=generate_before, tuple_output=tuple_output, include_nonstate_changes=include_nonstate_changes, earliest_timestamp=earliest_timestamp_value, expected_validation=expected_validation_value, ) self.task.init_local()
def setUp(self): self.initialize_ids() fake_param = luigi.DateIntervalParameter() self.task = CourseEnrollmentValidationTask( interval=fake_param.parse('2013-12-17'), output_root='/fake/output') self.task.init_local() self.user_id = 21 self.timestamp = "2013-12-17T15:38:32.805444" self.mode = 'honor' self.factory = SyntheticEventFactory( timestamp=self.timestamp, event_source='server', event_type=ACTIVATED, synthesizer='enrollment_from_db', reason='db entry', user_id=self.user_id, course_id=self.course_id, org_id=self.org_id, )