def test_checkinputsb_goodRepl(self): job_mock = Mock() job_mock.workflow.findParameter.return_value.getValue.return_value = 'LFN:/some/file.txt' ret_dict = {'Failed': [], 'Successful': {'/some/file.txt': {'Awesome-Disk-SE': 'surl'}}} def setOptions(*args): if 'SingleReplicaSEs' in args[0]: return ['Awesome-Disk-SE'] if 'Minimum' in args[0]: return 2 if args[0].endswith('PreferredSEs'): return ['Awesome-Tape-SE'] ops_mock = Mock() ops_mock.getValue = setOptions self.dilc.ops = ops_mock with patch('%s.DiracILC.getReplicas' % MODULE_NAME, new=Mock(return_value=S_OK(ret_dict))) as replica_mock: assertDiracSucceeds(self.dilc.checkInputSandboxLFNs(job_mock), self) replica_mock.assert_called_once_with(['/some/file.txt'])
def opsMock(): """Return fixture for Operations.""" def mockOpsConfig(*args, **kwargs): # pylint: disable=unused-argument """Mock the operations getValue calls.""" opsDict = {'DefaultDetectorModel': 'detModel', 'DefaultConfigVersion': 'Config', 'DefaultConfigPackage': 'Click', 'DefaultSoftwareVersion': 'Software', 'FailOverSE': 'FAIL=SRM', 'DefaultWhizard2Version': '1.9.5', 'BasePath': '', } for opName, value in opsDict.items(): if args[0].endswith(opName): return value assert args[0] == opsDict return None theOps = Mock(name='OpsMock') theOps.getValue = mockOpsConfig return theOps
def setUp( self ): """set up the objects""" ops_mock = Mock() mocked_modules = { 'DIRAC.ConfigurationSystem.Client.Helpers.Operations' : ops_mock } self.module_patcher = patch.dict( sys.modules, mocked_modules ) self.module_patcher.start() from ILCDIRAC.Interfaces.API.DiracILC import DiracILC self.dilc = DiracILC() def setOptions(*args): if 'SingleReplicaSEs' in args[0]: return ['SE'] if 'Minimum' in args[0]: return 1 if args[0].endswith('PreferredSEs'): return ['Awesome-Tape-SE'] ops_mock = Mock() ops_mock.getValue = Mock() ops_mock.getValue.side_effect = setOptions self.dilc.ops = ops_mock