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)