def __init__(self, commandCallerIn=None):
if commandCallerIn is not None:
self.cc = commandCallerIn
else:
from DIRAC.ResourceStatusSystem.Command.CommandCaller import CommandCaller
self.cc = CommandCaller()
self.policyInvoker = PolicyInvoker()
def setUp(self):
sys.modules["DIRAC"] = DIRAC.ResourceStatusSystem.test.fake_Logger
sys.modules[
"DIRAC.ResourceStatusSystem.Utilities.CS"] = DIRAC.ResourceStatusSystem.test.fake_Logger
sys.modules[
"DIRAC.Core.Utilities.SiteCEMapping"] = DIRAC.ResourceStatusSystem.test.fake_Logger
sys.modules[
"DIRAC.Core.Utilities.SiteSEMapping"] = DIRAC.ResourceStatusSystem.test.fake_Logger
sys.modules[
"DIRAC.Core.Utilities.SitesDIRACGOCDBmapping"] = DIRAC.ResourceStatusSystem.test.fake_Logger
sys.modules[
"DIRAC.Interfaces.API.DiracAdmin"] = DIRAC.ResourceStatusSystem.test.fake_Admin
sys.modules[
"DIRAC.FrameworkSystem.Client.NotificationClient"] = DIRAC.ResourceStatusSystem.test.fake_NotificationClient
from DIRAC.ResourceStatusSystem.Utilities.InfoGetter import InfoGetter
from DIRAC.ResourceStatusSystem.PolicySystem.PolicyBase import PolicyBase
from DIRAC.ResourceStatusSystem.PolicySystem.PolicyInvoker import PolicyInvoker
from DIRAC import gConfig
self.VO = gConfig.getValue("DIRAC/Extensions")
if 'LHCb' in self.VO:
self.VO = 'LHCb'
self.mock_command = Mock()
self.mock_policy = Mock()
self.mock_p = Mock()
self.mock_args = Mock()
self.pb = PolicyBase()
self.pi = PolicyInvoker()
self.mock_pdp = Mock()
self.mock_rsDB = Mock()
self.mock_rmDB = Mock()
self.mock_nc = Mock()
self.mock_da = Mock()
self.mock_da.getBannedSites.return_value = {
'OK': True,
'Value': ['LCG.APC.fr', 'LCG.Bari.it', 'LCG.Catania.it']
}
self.mock_da.addSiteInMask.return_value = {'OK': True, 'Value': ''}
self.mock_da.banSiteFromMask.return_value = {'OK': True, 'Value': ''}
self.mock_da.sendMail.return_value = {'OK': True, 'Value': ''}
self.mock_csAPI = Mock()
self.mock_csAPI.setOption.return_value = {'OK': True, 'Value': ''}
self.mock_csAPI.commit.return_value = {'OK': True, 'Value': ''}
self.ig = InfoGetter(self.VO)
