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))
