def test_upload_challenge_file_uploader_is_none(self): temp_dir = tempfile.mkdtemp() try: params = D3RParameters() chall = ChallengeDataTask(temp_dir, params) chall._upload_challenge_file('/foo') self.assertEqual(chall.get_email_log(), 'No uploader available ' 'to upload challenge ' 'data\n') finally: shutil.rmtree(temp_dir)
def test_upload_challenge_file_no_remote_challenge_dir(self): temp_dir = tempfile.mkdtemp() try: params = D3RParameters() chall = ChallengeDataTask(temp_dir, params) ftp = FtpFileTransfer(None) chall.set_file_transfer(ftp) chall._upload_challenge_file('/foo') self.assertEqual(chall.get_email_log(), 'No remote challenge' ' directory set for ' 'ftp upload\n') finally: shutil.rmtree(temp_dir)
def test_upload_challenge_file_challenge_file_is_none(self): temp_dir = tempfile.mkdtemp() try: params = D3RParameters() chall = ChallengeDataTask(temp_dir, params) chall._upload_challenge_file(None) chall.set_file_transfer(params) self.assertEqual(chall.get_email_log(), 'challenge_file is None in' ' _upload_challenge_file' '\n') finally: shutil.rmtree(temp_dir)