def test_remote_env_not_called_if_upload_was_done(self):
self.remote_env.upload_file = mock.Mock()
study = StudyDTO(path="hello")
study.zip_is_sent = True
new_study = self.study_uploader.upload(study)
self.remote_env.upload_file.assert_not_called()
assert new_study == study
def test_remote_env_is_called_if_upload_was_not_done(self):
self.remote_env.upload_file = mock.Mock()
study = StudyDTO(path="hello")
study.zip_is_sent = False
expected_study = copy(study)
expected_study.zip_is_sent = True
new_study = self.study_uploader.upload(study)
self.remote_env.upload_file.assert_called_once_with(study.zipfile_path)
assert new_study == expected_study
