def test_04_reTransfer(self): kwargs = getKwargsReTransfer() removalTask = RemovalTask( **kwargs ) removalTask.dataLoggingClient = Mock( return_value = Mock(spec = DataLoggingClient ) ) removalTask.dataLoggingClient().addFileRecord = Mock() removalTask.dataLoggingClient().addFileRecord.return_value = { "OK" : True, "Value" : "" } removalTask.requestClient = Mock( return_value = Mock(spec=RequestClient) ) removalTask.requestClient().updateRequest = Mock() removalTask.requestClient().updateRequest.return_value = { "OK" : True, "Value" : None } removalTask.requestClient().getRequestStatus = Mock() removalTask.requestClient().getRequestStatus.return_value = { "OK" : True, "Value" : { "RequestStatus" : "Done", "SubRequestStatus" : "Done" }} removalTask.requestClient().finalizeRequest = Mock() removalTask.requestClient().finalizeRequest.return_value = { "OK" : True, "Value" : None } removalTask.replicaManager = Mock( return_value = Mock( spec=ReplicaManager) ) removalTask.replicaManager().onlineRetransfer = Mock() removalTask.replicaManager().onlineRetransfer.return_value = { "OK" : True, "Value" : { "Failed" : {}, "Successful" : { "srm://srm-lhcb.gridpp.rl.ac.uk/castor/ads.rl.ac.uk/prod/lhcb/user/c/cibak/11889/11889410/test.zzz" : True } } } self.assertEqual( removalTask.__call__(), {'OK': True, 'Value': {'monitor': {'Execute': 1, 'Done': 1}}} ) del removalTask
def test_03_physicalRemoval( self ): kwargs = getKwargsPhysicalRemoval() removalTask = RemovalTask( **kwargs ) removalTask.dataLoggingClient = Mock( return_value = Mock(spec = DataLoggingClient ) ) removalTask.dataLoggingClient().addFileRecord = Mock() removalTask.dataLoggingClient().addFileRecord.return_value = { "OK" : True, "Value" : "" } removalTask.getProxyForLFN = Mock( return_value = { "OK" : True, "Value" : None} ) removalTask.requestClient = Mock( return_value = Mock(spec=RequestClient) ) removalTask.requestClient().updateRequest = Mock() removalTask.requestClient().updateRequest.return_value = { "OK" : True, "Value" : None } removalTask.requestClient().getRequestStatus = Mock() removalTask.requestClient().getRequestStatus.return_value = { "OK" : True, "Value" : { "RequestStatus" : "Done", "SubRequestStatus" : "Done" }} removalTask.requestClient().finalizeRequest = Mock() removalTask.requestClient().finalizeRequest.return_value = { "OK" : True, "Value" : None } removalTask.replicaManager = Mock( return_value = Mock( spec=ReplicaManager) ) removalTask.replicaManager().removeStorageFile = Mock() removalTask.replicaManager().removeStorageFile.return_value = { "OK" : True, "Value" : { "Failed" : {}, "Successful" : { "/lhcb/user/c/cibak/11889/11889410/test.zzz" : True } } } self.assertEqual( removalTask.__call__(), {'OK': True, 'Value': {'monitor': {'Done': 1, 'Execute': 1, 'PhysicalRemovalAtt': 1, 'PhysicalRemovalDone': 1}}} ) del removalTask