def test_check_move_status(self):
to_test = DataMoverServices(None, None)
to_test._executor = mock.MagicMock()
to_test._move_jobs = {}
dest = 'scp://visualiser/some/path'
source = 'ala://ala/?lsid=lsid'
out = to_test.move(source, dest)
self.assertIsNotNone(out)
self.assertEqual(MoveJob.STATUS_PENDING, out['status'])
id = to_test._move_jobs.keys()[0]
out = to_test.check_move_status(id)
self.assertIsNotNone(out)
self.assertEqual(MoveJob.STATUS_PENDING, out['status'])
self.assertEqual(id, out['id'])
# job should still exists
out = to_test.check_move_status(id)
self.assertIsNotNone(out)
self.assertEqual(MoveJob.STATUS_PENDING, out['status'])
self.assertEqual(id, out['id'])
# Now update the move job to complete.
to_test._move_jobs[id].update(status=MoveJob.STATUS_COMPLETE)
# Check that the status is updated to complete
out = to_test.check_move_status(id)
self.assertIsNotNone(out)
self.assertEqual(MoveJob.STATUS_COMPLETE, out['status'])
self.assertEqual(id, out['id'])
# Do a 2nd status query, job shall not exist anymore.
out = to_test.check_move_status(id)
self.assertIsNotNone(out)
self.assertEqual(STATUS_REJECTED, out['status'])
self.assertEqual(REASON_JOB_DOES_NOT_EXIST, out['reason'])
def test_xml_move(self):
to_test = DataMoverServices(None, None)
to_test._executor = mock.MagicMock()
to_test._move_jobs = {}
dest = 'scp://visualiser/some/path'
source = 'ala://ala/?lsid=lsid'
move_job = MoveJob(source, dest, None, False)
out = to_test.move(source, dest)
self.assertIsNotNone(out)
self.assertEqual(MoveJob.STATUS_PENDING, out['status'])
act_move_job = to_test._move_jobs[to_test._move_jobs.keys()[0]]
self.assertEqual(act_move_job.source, move_job.source)
self.assertEqual(act_move_job.destination, move_job.destination)
self.assertEqual(act_move_job.zip, move_job.zip)
self.assertEqual(act_move_job.userid, move_job.userid)
to_test._executor.submit.assert_called_with(fn=to_test._move_service.worker, move_job=act_move_job)
