Пример #1
0
 def test_download_remote_challenge_data_package_with_retry_retry_neg(self):
     params = D3RParameters()
     task = ExternalDataSubmissionTask('/foo', 'yo', 'hi', params)
     task.set_download_max_retry_count(-1)
     task.set_download_retry_sleep(0)
     try:
         task._download_remote_challenge_data_package_with_retry()
         self.fail('Expected ChallengePackageDownloadError')
     except ChallengePackageDownloadError as e:
         self.assertEqual(str(e), 'Unable to download hi')
Пример #2
0
 def test_download_remote_challenge_data_package_with_1st_retry_false(self):
     temp_dir = tempfile.mkdtemp()
     try:
         params = D3RParameters()
         pkg = '/remote/celpp1_2017_dockedresults_yo.tar.gz'
         task = ExternalDataSubmissionTask(temp_dir, 'yo', pkg, params)
         task.set_download_retry_sleep(0)
         task.set_download_max_retry_count(3)
         mockft = D3RParameters()
         mockft.connect = Mock()
         mockft.download_file = Mock()
         mockft.download_file.side_effect = [False, True]
         mockft.disconnect = Mock(return_value=None)
         task.set_file_transfer(mockft)
         c = task._download_remote_challenge_data_package_with_retry()
         self.assertEqual(c, 'celpp1_2017_dockedresults_yo.tar.gz')
     finally:
         shutil.rmtree(temp_dir)
Пример #3
0
 def test_download_remote_challenge_data_package_with_retry_false(self):
     temp_dir = tempfile.mkdtemp()
     try:
         params = D3RParameters()
         pkg = '/remote/celpp1_2017_dockedresults_yo.tar.gz'
         task = ExternalDataSubmissionTask(temp_dir, 'yo', pkg, params)
         task.set_download_retry_sleep(0)
         task.set_download_max_retry_count(1)
         mockft = D3RParameters()
         mockft.connect = Mock()
         mockft.download_file = Mock(return_value=False)
         mockft.disconnect = Mock(return_value=None)
         task.set_file_transfer(mockft)
         try:
             task._download_remote_challenge_data_package_with_retry()
         except ChallengePackageDownloadError as e:
             self.assertEqual(
                 str(e), 'Unable to download ' +
                 'celpp1_2017_dockedresults_yo.tar.gz')
     finally:
         shutil.rmtree(temp_dir)