def __init__(self, plugin, transClient=None, dataManager=None, fc=None): """Constructor of the TransformationPlugin. Instantiate clients, if not given, and set up the PluginUtilities. """ super(TransformationPlugin, self).__init__(plugin) self.data = {} self.files = False self.startTime = time.time() self.valid = False if transClient is None: transClient = TransformationClient() if dataManager is None: dataManager = DataManager() if fc is None: fc = FileCatalog() self.util = PluginUtilities(plugin, transClient=transClient, dataManager=dataManager, fc=fc)
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 __init__(self, plugin, transClient=None, dataManager=None): """ plugin name has to be passed in: it will then be executed as one of the functions below, e.g. plugin = 'BySize' will execute TransformationPlugin('BySize')._BySize() """ super(TransformationPlugin, self).__init__(plugin) self.data = {} self.files = False self.startTime = time.time() if transClient is None: transClient = TransformationClient() if dataManager is None: dataManager = DataManager() self.util = PluginUtilities(plugin, transClient, dataManager)
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')