def initialize(self): """ Sets defaults """ TaskManagerAgentBase.initialize(self) RequestTasks.__init__(self) self.transType = ['Replication', 'Removal'] # This sets the Default Proxy to used as that defined under # /Operations/Shifter/ProductionManager # the shifterProxy option in the Configuration can be used to change this default. self.am_setOption('shifterProxy', 'ProductionManager') return S_OK()
def initialize( self ): """ Sets defaults """ TaskManagerAgentBase.initialize( self ) RequestTasks.__init__( self ) self.transType = ['Replication', 'Removal'] # This sets the Default Proxy to used as that defined under # /Operations/Shifter/ProductionManager # the shifterProxy option in the Configuration can be used to change this default. self.am_setOption( 'shifterProxy', 'ProductionManager' ) return S_OK()
def setUp( self ): self.mockTransClient = Mock() self.mockTransClient.setTaskStatusAndWmsID.return_value = {'OK':True} self.WMSClientMock = Mock() self.jobMonitoringClient = Mock() self.mockReqClient = Mock() self.jobMock = Mock() self.jobMock2 = Mock() mockWF = Mock() mockPar = Mock() mockWF.findParameter.return_value = mockPar mockPar.getValue.return_value = 'MySite' self.jobMock2.workflow = mockWF self.jobMock2.setDestination.return_value = {'OK':True} self.jobMock.workflow.return_value = '' self.jobMock.return_value = self.jobMock2 self.taskBase = TaskBase( transClient = self.mockTransClient ) self.wfTasks = WorkflowTasks( transClient = self.mockTransClient, submissionClient = self.WMSClientMock, jobMonitoringClient = self.jobMonitoringClient, outputDataModule = "mock", jobClass = self.jobMock ) self.requestTasks = RequestTasks( transClient = self.mockTransClient, requestClient = self.mockReqClient ) self.tc = TransformationClient() self.transformation = Transformation() self.maxDiff = None
def _getClients(self): """ Here the taskManager becomes a RequestTasks object """ res = TaskManagerAgentBase._getClients(self) threadTaskManager = RequestTasks() res.update({'TaskManager': threadTaskManager}) return res
def _getClients(self, ownerDN=None, ownerGroup=None): """Set the clients for task submission. Here the taskManager becomes a RequestTasks object. See :func:`DIRAC.TransformationSystem.TaskManagerAgentBase._getClients`. """ res = super(RequestTaskAgent, self)._getClients(ownerDN=ownerDN, ownerGroup=ownerGroup) threadTaskManager = RequestTasks(ownerDN=ownerDN, ownerGroup=ownerGroup) res.update({'TaskManager': threadTaskManager}) return res
def __init__(self, *args, **kwargs): ''' c'tor ''' TaskManagerAgentBase.__init__(self, *args, **kwargs) self.taskManager = RequestTasks(transClient=self.transClient) self.shifterProxy = 'ProductionManager' agentTSTypes = self.am_getOption('TransType', []) if agentTSTypes: self.transType = agentTSTypes else: self.transType = Operations().getValue( 'Transformations/DataManipulation', ['Replication', 'Removal'])
def setUp(self): self.mockTransClient = mock.MagicMock() self.mockTransClient.setTaskStatusAndWmsID.return_value = {'OK': True} self.mockReqClient = mock.MagicMock() self.taskBase = TaskBase(transClient=self.mockTransClient) self.pu = PluginUtilities(transClient=self.mockTransClient) self.requestTasks = RequestTasks(transClient=self.mockTransClient, requestClient=self.mockReqClient, requestValidator=reqValFake) self.transformation = Transformation() self.maxDiff = None
def initialize(self): """ Standard initialize method """ res = TaskManagerAgentBase.initialize(self) if not res['OK']: return res # clients self.taskManager = RequestTasks(transClient=self.transClient) agentTSTypes = self.am_getOption('TransType', []) if agentTSTypes: self.transType = agentTSTypes else: self.transType = Operations().getValue( 'Transformations/DataManipulation', ['Replication', 'Removal']) return S_OK()
def setUp(self): self.mockTransClient = MagicMock() self.mockTransClient.setTaskStatusAndWmsID.return_value = {'OK': True} self.WMSClientMock = MagicMock() self.jobMonitoringClient = MagicMock() self.mockReqClient = MagicMock() self.jobMock = MagicMock() self.jobMock2 = MagicMock() mockWF = MagicMock() mockPar = MagicMock() mockWF.findParameter.return_value = mockPar mockPar.getValue.return_value = 'MySite' self.jobMock2.workflow = mockWF self.jobMock2.setDestination.return_value = {'OK': True} self.jobMock.workflow.return_value = '' self.jobMock.return_value = self.jobMock2 self.reqValidatorMock = MagicMock() self.reqValidatorMock.validate.return_value = {'OK': True} self.taskBase = TaskBase(transClient=self.mockTransClient) self.pu = PluginUtilities(transClient=self.mockTransClient) self.wfTasks = WorkflowTasks( transClient=self.mockTransClient, submissionClient=self.WMSClientMock, jobMonitoringClient=self.jobMonitoringClient, outputDataModule="mock") self.requestTasks = RequestTasks(transClient=self.mockTransClient, requestClient=self.mockReqClient, requestValidator=reqValFake) self.tc = TransformationClient() self.transformation = Transformation() self.maxDiff = None gLogger.setLevel('DEBUG')
def setUp(self): self.mockTransClient = Mock() self.mockTransClient.setTaskStatusAndWmsID.return_value = {'OK': True} self.WMSClientMock = Mock() self.jobMonitoringClient = Mock() self.mockRequestClient = Mock() self.jobMock = Mock() self.jobMock.setDestination.return_value = {'OK': True} self.taskBase = TaskBase(transClient=self.mockTransClient) self.wfTasks = WorkflowTasks( transClient=self.mockTransClient, submissionClient=self.WMSClientMock, jobMonitoringClient=self.jobMonitoringClient, outputDataModule="mock", jobClass=self.jobMock) self.requestTasks = RequestTasks(transClient=self.mockTransClient, requestClient=self.mockRequestClient)