def test_move_challenge_data_package_into_task_dir_valid(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
task = ExternalDataSubmissionTask(temp_dir, 'yo',
'hi', params)
task.create_dir()
basedir = os.path.join(task.get_dir(), 'foo')
os.makedirs(basedir)
can1 = os.path.join(basedir, '5juw')
os.makedirs(can1)
pdb1 = os.path.join(can1, 'apo.pdb')
open(pdb1, 'a').close()
mol1 = os.path.join(can1, 'apo.mol')
open(mol1, 'a').close()
can2 = os.path.join(basedir, 'abcd')
os.makedirs(can2)
pdb2 = os.path.join(can2, 'apo.pdb')
open(pdb2, 'a').close()
mol2 = os.path.join(can2, 'apo.mol')
open(mol2, 'a').close()
task._move_challenge_data_package_into_task_dir('foo')
can3 = os.path.join(task.get_dir(), '5juw')
can4 = os.path.join(task.get_dir(), 'abcd')
self.assertTrue(os.path.isdir(can3))
self.assertTrue(os.path.isdir(can4))
self.assertTrue(os.path.isfile(os.path.join(can3, 'apo.pdb')))
self.assertTrue(os.path.isfile(os.path.join(can3, 'apo.mol')))
self.assertTrue(os.path.isfile(os.path.join(can4, 'apo.pdb')))
self.assertTrue(os.path.isfile(os.path.join(can4, 'apo.mol')))
finally:
shutil.rmtree(temp_dir)
def test_untar_challenge_data_package_dotdot_tarfile(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
task = ExternalDataSubmissionTask(temp_dir, 'hello',
'hi', params)
task.create_dir()
self.create_bad_tarfile_dotdot_path(task.get_dir(),
'hello')
c = task._untar_challenge_data_package('hello.tar.gz')
self.assertEqual(c, 'hello')
self.assertEqual(task.get_email_log(), 'Skipping, found .. in '
'path: hello/5juw/apo-'
'5juw_2eb2_..docked.mol\n'
'Ignoring non dir/file '
'entry in tar 5juw/hello\n')
candidate = os.path.join(task.get_dir(), 'hello', '5juw')
self.assertTrue(os.path.isdir(candidate))
mypdb = os.path.join(candidate,
'apo-5juw_2eb2_docked.pdb')
mymol = os.path.join(candidate,
'apo-5juw_2eb2.._docked.mol')
self.assertTrue(os.path.isfile(mypdb))
self.assertFalse(os.path.isfile(mymol))
finally:
shutil.rmtree(temp_dir)
def test_untar_challenge_data_package_good_tarfile(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
task = ExternalDataSubmissionTask(temp_dir, 'hello', 'hi', params)
task.create_dir()
self.create_good_tarfile(task.get_dir(), 'hello')
c = task._untar_challenge_data_package('hello.tar.gz')
self.assertEqual(c, 'hello')
self.assertEqual(task.get_email_log(), None)
candidate = os.path.join(task.get_dir(), 'hello', '5juw')
self.assertTrue(os.path.isdir(candidate))
mypdb = os.path.join(candidate, 'apo-5juw_2eb2_docked.pdb')
mymol = os.path.join(candidate, 'apo-5juw_2eb2_docked.mol')
self.assertTrue(os.path.isfile(mypdb))
self.assertTrue(os.path.isfile(mymol))
finally:
shutil.rmtree(temp_dir)
def test_untar_challenge_data_package_wrong_prefix(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
task = ExternalDataSubmissionTask(temp_dir, 'yo', 'hi', params)
task.create_dir()
tfile = self.create_good_tarfile(task.get_dir(), 'yo')
shutil.move(tfile, os.path.join(task.get_dir(), 'byte.tar.gz'))
c = task._untar_challenge_data_package('byte.tar.gz')
self.assertEqual(c, 'byte')
candidate = os.path.join(task.get_dir(), 'hello', '5juw')
self.assertFalse(os.path.isdir(candidate))
mypdb = os.path.join(candidate, 'apo-5juw_2eb2_docked.pdb')
mymol = os.path.join(candidate, 'apo-5juw_2eb2.._docked.mol')
self.assertFalse(os.path.isfile(mypdb))
self.assertFalse(os.path.isfile(mymol))
finally:
shutil.rmtree(temp_dir)
def test_runtask_success(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
pkg = '/remote/celpp1_2017_dockedresults_yo.tar.gz'
task = ExternalDataSubmissionTask(temp_dir, 'yo', pkg, params)
mockft = D3RParameters()
mockft.connect = Mock(return_value=None)
mockft.download_file = Mock(return_value=True)
mockft.disconnect = Mock(return_value=None)
task.set_file_transfer(mockft)
task.create_dir()
self.create_good_tarfile(task.get_dir(),
'celpp1_2017_dockedresults_yo')
task._runtask()
self.assertEqual(task.get_error(), None)
self.assertEqual(task.get_email_log(), '')
finally:
shutil.rmtree(temp_dir)
def test_externaltask_can_run(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
task = ExternalDataSubmissionTask(temp_dir, 'name', '/remote',
params)
self.assertTrue(task.can_run())
task.create_dir()
self.assertFalse(task.can_run())
self.assertEqual(task.get_error(), 'stage.6.name.extsubmission '
'already exists and status is '
'unknown')
open(os.path.join(task.get_dir(),
D3RTask.COMPLETE_FILE), 'a').close()
self.assertFalse(task.can_run())
self.assertEqual(task.get_error(), None)
finally:
shutil.rmtree(temp_dir)
def test__download_remote_challenge_data_package_success(self):
temp_dir = tempfile.mkdtemp()
try:
params = D3RParameters()
pkg = '/remote/celpp2_3.tar.gz'
task = ExternalDataSubmissionTask(temp_dir, 'yo',
pkg, params)
task.create_dir()
mockft = D3RParameters()
mockft.connect = Mock(return_value=None)
mockft.download_file = Mock(return_value=True)
mockft.disconnect = Mock(return_value=None)
task.set_file_transfer(mockft)
val = task._download_remote_challenge_data_package()
self.assertEqual(val, 'celpp2_3.tar.gz')
localfile = os.path.join(task.get_dir(), 'celpp2_3.tar.gz')
mockft.download_file.assert_called_with(pkg, localfile)
mockft.connect.assert_called_with()
mockft.disconnect.assert_called_with()
finally:
shutil.rmtree(temp_dir)