def checkSettings(self, script):
"""check if all required parameters are set, print error message and return S_ERROR if not"""
args = script.getPositionalArgs()
if len(args) == 4:
self.setProdIDs( args[0] )
self.setTargetSE( args[1] )
self.setSourceSE( args[2] )
self.setDatatype( args[3] )
elif len(args) == 2 and self.allFor:
self.setProdIDs( self.allFor )
## place the indiviual entries as well.
prodTemp = list( self.prodIDs )
for prodID in prodTemp:
self.prodIDs.append( prodID+1 )
self.prodIDs.append( prodID+2 )
self.prodIDs = sorted( self.prodIDs )
self.setTargetSE( args[0] )
self.setSourceSE( args[1] )
else:
self.errorMessages.append("ERROR: Not enough arguments")
from ILCDIRAC.Core.Utilities.CheckAndGetProdProxy import checkAndGetProdProxy
ret = checkAndGetProdProxy()
if not ret['OK']:
self.errorMessages.append( ret['Message'] )
if not self.errorMessages:
return S_OK()
gLogger.error("\n".join(self.errorMessages))
script.showHelp()
return S_ERROR()
def test_failure_7( self ): """test for CheckandGetProdProxy: semi failure 6................................................""" res = checkAndGetProdProxy() self.assertTrue( not res['OK'] )
def test_failure_2( self ): """test for CheckandGetProdProxy: semi failure 1................................................""" res = checkAndGetProdProxy() self.assertFalse( res['OK'] )
def test_success_2( self ): """test for CheckandGetProdProxy: sucess 2......................................................""" res = checkAndGetProdProxy() self.assertTrue( res['OK'] )
def test_failure_7( self ):
"""test for CheckandGetProdProxy: semi failure 6................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_ERROR("no proxy info"),S_ERROR("Still no proxy")])):
res = checkAndGetProdProxy()
self.assertTrue( not res['OK'] )
def test_failure_4( self ):
"""test for CheckandGetProdProxy: semi failure 3................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_ERROR("no proxy info"),S_OK({"group":"ilc_prod"})])):
res = checkAndGetProdProxy()
self.assertTrue( res['OK'] )
def test_failure_2( self ):
"""test for CheckandGetProdProxy: semi failure 1................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(return_value=S_ERROR('message'))):
res = checkAndGetProdProxy()
self.assertFalse( res['OK'] )
def test_failure( self ):
"""test for CheckandGetProdProxy: failure......................................................."""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=1)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(return_value=S_ERROR('No proxy info'))):
res = checkAndGetProdProxy()
self.assertFalse( res['OK'] )
def test_success_2( self ):
"""test for CheckandGetProdProxy: sucess 2......................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_OK({"group":"ilc_user"}),S_OK({"group":"ilc_prod"})])):
res = checkAndGetProdProxy()
self.assertTrue( res['OK'] )
def test_failure_7(self):
"""test for CheckandGetProdProxy: semi failure 6................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_ERROR("no proxy info"),S_ERROR("Still no proxy")])):
res = checkAndGetProdProxy()
self.assertTrue(not res['OK'])
def test_failure_4(self):
"""test for CheckandGetProdProxy: semi failure 3................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_ERROR("no proxy info"),S_OK({"group":"ilc_prod"})])):
res = checkAndGetProdProxy()
self.assertTrue(res['OK'])
def test_failure_2(self):
"""test for CheckandGetProdProxy: semi failure 1................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(return_value=S_ERROR('message'))):
res = checkAndGetProdProxy()
self.assertFalse(res['OK'])
def test_failure(self):
"""test for CheckandGetProdProxy: failure......................................................."""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=1)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(return_value=S_ERROR('No proxy info'))):
res = checkAndGetProdProxy()
self.assertFalse(res['OK'])
def test_success_2(self):
"""test for CheckandGetProdProxy: sucess 2......................................................"""
with patch("%s.call" % MODULE_NAME, new=Mock(return_value=0)), \
patch("%s.getProxyInfo" % MODULE_NAME, new=Mock(side_effect=[S_OK({"group":"ilc_user"}),S_OK({"group":"ilc_prod"})])):
res = checkAndGetProdProxy()
self.assertTrue(res['OK'])
