def test_05_replicateAndRegister( self ):
""" 'replicateAndRegister' operation """
kwargs = getKwargs( "replicateAndRegister" )
tTask = TransferTask( **kwargs )
tTask.dataLoggingClient = Mock( return_value = Mock(spec = DataLoggingClient ) )
tTask.dataLoggingClient().addFileRecord = Mock()
tTask.dataLoggingClient().addFileRecord.return_value = SOK
tTask.requestClient = Mock( return_value = Mock(spec=RequestClient) )
tTask.requestClient().updateRequest = Mock()
tTask.requestClient().updateRequest.return_value = SOK
tTask.requestClient().getRequestStatus = Mock()
tTask.requestClient().getRequestStatus.return_value = { "OK" : True,
"Value" : { "RequestStatus" : "Done",
"SubRequestStatus" : "Done" }}
tTask.requestClient().finalizeRequest = Mock()
tTask.requestClient().finalizeRequest.return_value = SOK
tTask.replicaManager = Mock( return_value = Mock( spec=ReplicaManager) )
tTask.replicaManager().replicateAndRegister = Mock()
tTask.replicaManager().replicateAndRegister.return_value = { "OK": True,
"Value": {
"Failed": {},
"Successful": { "/lhcb/user/c/cibak/11889/11889410/test.zzz" : { "replicate": 1, "register": 1 } } } }
self.assertEqual( tTask(),
{ 'OK': True,
'Value': { 'monitor':
{ 'Replica registration successful': 1,
'Execute': 1,
'Done': 1,
'Replication successful': 1,
'Replicate and register': 1}}} )
del tTask
def test_05_replicateAndRegister(self):
""" 'replicateAndRegister' operation """
kwargs = getKwargs("replicateAndRegister")
tTask = TransferTask(**kwargs)
tTask.dataLoggingClient = Mock(return_value=Mock(
spec=DataLoggingClient))
tTask.dataLoggingClient().addFileRecord = Mock()
tTask.dataLoggingClient().addFileRecord.return_value = SOK
tTask.requestClient = Mock(return_value=Mock(spec=RequestClient))
tTask.requestClient().updateRequest = Mock()
tTask.requestClient().updateRequest.return_value = SOK
tTask.requestClient().getRequestStatus = Mock()
tTask.requestClient().getRequestStatus.return_value = {
"OK": True,
"Value": {
"RequestStatus": "Done",
"SubRequestStatus": "Done"
}
}
tTask.requestClient().finalizeRequest = Mock()
tTask.requestClient().finalizeRequest.return_value = SOK
tTask.replicaManager = Mock(return_value=Mock(spec=ReplicaManager))
tTask.replicaManager().replicateAndRegister = Mock()
tTask.replicaManager().replicateAndRegister.return_value = {
"OK": True,
"Value": {
"Failed": {},
"Successful": {
"/lhcb/user/c/cibak/11889/11889410/test.zzz": {
"replicate": 1,
"register": 1
}
}
}
}
self.assertEqual(
tTask(), {
'OK': True,
'Value': {
'monitor': {
'Replica registration successful': 1,
'Execute': 1,
'Done': 1,
'Replication successful': 1,
'Replicate and register': 1
}
}
})
del tTask
