def create_task(self): self.task = self.task_class( output_root='/fake/output', auth_user_path='/fake/input1', auth_userprofile_path='/fake/input2', explicit_event_whitelist='explicit_events.tsv', ) explicit_event_list = """ admin browser edx.instructor.report.downloaded admin server add-forum-admin admin server add-forum-community-TA admin server add-forum-mod admin server add-instructor admin server list-staff enrollment server edx.course.enrollment.activated # problem server problem_rescore """ results = { 'explicit_events': FakeTarget(value=self.reformat(explicit_event_list)), } self.task.input_local = MagicMock(return_value=results) self.task.init_local() self.task.geoip = FakeGeoLocation() self.task.user_info_requirements = get_mock_user_info_requirements()
def create_task(self): self.task = self.task_class( output_root='/fake/output', auth_user_path='/fake/input1', auth_userprofile_path='/fake/input2', explicit_event_whitelist='explicit_events.tsv', ) explicit_event_list = """ admin browser edx.instructor.report.downloaded admin server add-forum-admin admin server add-forum-community-TA admin server add-forum-mod admin server add-instructor admin server list-staff enrollment server edx.course.enrollment.activated # problem server problem_rescore """ results = { 'explicit_events': FakeTarget(value=self.reformat(explicit_event_list)), } self.task.input_local = MagicMock(return_value=results) self.task.init_local() self.task.geoip = FakeGeoLocation() self.task.user_info_requirements = get_mock_user_info_requirements()
def run_task(self): """Runs the task with fake targets.""" output_archive_root = tempfile.mkdtemp() self.addCleanup(shutil.rmtree, output_archive_root) with tempfile.NamedTemporaryFile() as tmp_input_archive: with tarfile.open(mode='w:gz', fileobj=tmp_input_archive) as input_archive_file: input_archive_file.add(self.archive_root, arcname='') tmp_input_archive.seek(0) task = obfuscate.CourseContentTask( course=sentinel.ignored, output_directory=sentinel.ignored, data_directory=sentinel.ignored, auth_user_path=sentinel.ignored, auth_userprofile_path=sentinel.ignored, ) fake_input = {'data': [LocalTarget(path=tmp_input_archive.name)]} task.input = MagicMock(return_value=fake_input) output_target = FakeTarget() task.output = MagicMock(return_value=output_target) task.user_info_requirements = get_mock_user_info_requirements() reset_user_info_for_testing() task.run() with tarfile.open(mode='r:gz', fileobj=output_target.buffer) as output_archive_file: output_archive_file.extractall(output_archive_root) self.output_course_root = os.path.join(output_archive_root, get_filename_safe_course_id(self.COURSE_ID))
def run_task(self, task_cls, source): """Runs the task with fake targets.""" task = task_cls( course=sentinel.ignored, output_directory=sentinel.ignored, data_directory=sentinel.ignored, auth_user_path=sentinel.ignored, auth_userprofile_path=sentinel.ignored, ) fake_input = {'data': [FakeTarget(value=source)]} task.input = MagicMock(return_value=fake_input) output_target = FakeTarget() task.output = MagicMock(return_value=output_target) task.user_info_requirements = get_mock_user_info_requirements() reset_user_info_for_testing() task.run() return output_target.buffer.read()
def run_task(self, task_cls, source): """Runs the task with fake targets.""" task = task_cls( course=sentinel.ignored, output_directory=sentinel.ignored, data_directory=sentinel.ignored, auth_user_path=sentinel.ignored, auth_userprofile_path=sentinel.ignored, ) fake_input = {'data': [FakeTarget(value=source)]} task.input = MagicMock(return_value=fake_input) output_target = FakeTarget() task.output = MagicMock(return_value=output_target) task.user_info_requirements = get_mock_user_info_requirements() reset_user_info_for_testing() task.run() return output_target.buffer.read()
def run_task(self): """Runs the task with fake targets.""" output_archive_root = tempfile.mkdtemp() self.addCleanup(shutil.rmtree, output_archive_root) with tempfile.NamedTemporaryFile() as tmp_input_archive: with tarfile.open(mode='w:gz', fileobj=tmp_input_archive) as input_archive_file: input_archive_file.add(self.archive_root, arcname='') tmp_input_archive.seek(0) task = obfuscate.CourseContentTask( course=sentinel.ignored, output_directory=sentinel.ignored, data_directory=sentinel.ignored, auth_user_path=sentinel.ignored, auth_userprofile_path=sentinel.ignored, ) fake_input = {'data': [LocalTarget(path=tmp_input_archive.name)]} task.input = MagicMock(return_value=fake_input) output_target = FakeTarget() task.output = MagicMock(return_value=output_target) task.user_info_requirements = get_mock_user_info_requirements() reset_user_info_for_testing() task.run() with tarfile.open( mode='r:gz', fileobj=output_target.buffer) as output_archive_file: output_archive_file.extractall(output_archive_root) self.output_course_root = os.path.join( output_archive_root, get_filename_safe_course_id(self.COURSE_ID))