def test_remote_environment_not_called_if_final_zip_not_downloaded(self):
self.remote_env_mock.clean_remote_server = mock.Mock()
study = StudyDTO(path="hello")
study.local_final_zipfile_path = ""
new_study = self.remote_server_cleaner.clean(study)
self.remote_env_mock.clean_remote_server.assert_not_called()
assert new_study == study
study.local_final_zipfile_path = None
new_study = self.remote_server_cleaner.clean(study)
self.remote_env_mock.clean_remote_server.assert_not_called()
assert new_study == study
def test_remote_env_not_called_if_final_zip_already_downloaded(self):
self.remote_env.download_final_zip = mock.Mock()
downloaded_study = StudyDTO("hello")
downloaded_study.local_final_zipfile_path = "results.zip"
new_study = self.final_zip_downloader.download(downloaded_study)
self.remote_env.download_final_zip.assert_not_called()
assert new_study == downloaded_study
def test_remote_environment_not_called_if_remote_server_is_already_clean(self):
self.remote_env_mock.clean_remote_server = mock.Mock()
study = StudyDTO(path="hello")
study.local_final_zipfile_path = "hello.zip"
study.remote_server_is_clean = True
new_study = self.remote_server_cleaner.clean(study)
self.remote_env_mock.clean_remote_server.assert_not_called()
assert new_study == study
def test_when_study_finished_with_error_check_if_study_is_done_returns_true(
result,
with_error,
logs_downloaded,
local_final_zipfile_path,
remote_server_is_clean,
final_zip_extracted,
):
my_study = StudyDTO(path="hello",
job_id=42,
started=True,
finished=True)
my_study.with_error = with_error
my_study.logs_downloaded = logs_downloaded
my_study.local_final_zipfile_path = local_final_zipfile_path
my_study.remote_server_is_clean = remote_server_is_clean
my_study.final_zip_extracted = final_zip_extracted
assert StudyRetriever.check_if_study_is_done(my_study) is result