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