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