def setUp(self):
""" set up
:param self: self reference
"""
self.log = gLogger.getSubLogger(self.__class__.__name__)
self.requestDB = RequestDBFile()
setRequest = self.requestDB.setRequest("testRequest", reqStr)
class RequestDBFileTests(unittest.TestCase):
"""
.. class:: RequestDBFileTests
"""
def setUp( self ):
""" set up
:param self: self reference
"""
self.log = gLogger.getSubLogger( self.__class__.__name__ )
self.requestDB = RequestDBFile()
setRequest = self.requestDB.setRequest( "testRequest", reqStr )
def tearDown( self ):
#deleteRequest = self.requestDB.deleteRequest( "testRequest" )
pass
def test_01_getRequestStatus( self ):
self.assertEqual( self.requestDB.getRequestStatus( "testRequest" ),
{'OK': True, 'Value': {'SubRequestStatus': 'Waiting', 'RequestStatus': 'Waiting'}})
## get request
getRemoval = self.requestDB.getRequest( "removal" )
oRequest = RequestContainer( getRemoval["Value"]["RequestString"] )
self.assertEqual( self.requestDB.getRequestStatus( "testRequest" ),
{'OK': True, 'Value': {'SubRequestStatus': 'Assigned', 'RequestStatus': 'Waiting'}} )
## make removal Done
oRequest.subRequests["removal"][0]["Attributes"]["Status"] = "Done"
oRequest.subRequests["removal"][0]["Files"][0]["Status"] = "Done"
update = self.requestDB.updateRequest( getRemoval["Value"]["RequestName"],
oRequest.toXML()["Value"] )
## get status
self.assertEqual( self.requestDB.getRequestStatus( "testRequest" ),
{'OK': True, 'Value': {'SubRequestStatus': 'Waiting', 'RequestStatus': u'Waiting'}})
## make transfer Done
oRequest.subRequests["transfer"][0]["Attributes"]["Status"] = "Done"
oRequest.subRequests["transfer"][0]["Files"][0]["Status"] = "Done"
update = self.requestDB.updateRequest( getRemoval["Value"]["RequestName"],
oRequest.toXML()["Value"] )
## get status
self.assertEqual( self.requestDB.getRequestStatus( "testRequest" ),
{'OK': True, 'Value': {'SubRequestStatus': 'Done', 'RequestStatus': 'Done'}} )
def test_02_getRequest( self ):
""" getRequest and JobID """
getRequest = self.requestDB.getRequest("transfer")
self.assertEqual( getRequest["OK"], True )
self.assertEqual( getRequest["Value"]["JobID"], 0 )
def setUp( self ):
""" set up
:param self: self reference
"""
self.log = gLogger.getSubLogger( self.__class__.__name__ )
self.requestDB = RequestDBFile()
setRequest = self.requestDB.setRequest( "testRequest", reqStr )
def initializeRequestManagerHandler(serviceInfo):
global requestDB
csSection = PathFinder.getServiceSection( "RequestManagement/RequestManager" )
backend = gConfig.getValue('%s/Backend' % csSection)
if not backend:
fatStr = "RequestManager.initializeRequestManagerHandler: Failed to get backed for RequestDB from CS."
gLogger.fatal(fatStr)
return S_ERROR(fatStr)
gLogger.info("RequestManager.initializeRequestManagerHandler: Initialising with backend",backend)
if backend == 'file':
from DIRAC.RequestManagementSystem.DB.RequestDBFile import RequestDBFile
requestDB = RequestDBFile()
elif backend == 'mysql':
from DIRAC.RequestManagementSystem.DB.RequestDBMySQL import RequestDBMySQL
requestDB = RequestDBMySQL()
else:
fatStr = "RequestManager.initializeRequestManagerHandler: Supplied backend is not supported."
gLogger.fatal(fatStr,backend)
return S_ERROR(fatStr)
return S_OK()
class RequestDBFileTests(unittest.TestCase):
"""
.. class:: RequestDBFileTests
"""
def setUp(self):
""" set up
:param self: self reference
"""
self.log = gLogger.getSubLogger(self.__class__.__name__)
self.requestDB = RequestDBFile()
setRequest = self.requestDB.setRequest("testRequest", reqStr)
def tearDown(self):
#deleteRequest = self.requestDB.deleteRequest( "testRequest" )
pass
def test_01_getRequestStatus(self):
self.assertEqual(
self.requestDB.getRequestStatus("testRequest"), {
'OK': True,
'Value': {
'SubRequestStatus': 'Waiting',
'RequestStatus': 'Waiting'
}
})
## get request
getRemoval = self.requestDB.getRequest("removal")
oRequest = RequestContainer(getRemoval["Value"]["RequestString"])
self.assertEqual(
self.requestDB.getRequestStatus("testRequest"), {
'OK': True,
'Value': {
'SubRequestStatus': 'Assigned',
'RequestStatus': 'Waiting'
}
})
## make removal Done
oRequest.subRequests["removal"][0]["Attributes"]["Status"] = "Done"
oRequest.subRequests["removal"][0]["Files"][0]["Status"] = "Done"
update = self.requestDB.updateRequest(
getRemoval["Value"]["RequestName"],
oRequest.toXML()["Value"])
## get status
self.assertEqual(
self.requestDB.getRequestStatus("testRequest"), {
'OK': True,
'Value': {
'SubRequestStatus': 'Waiting',
'RequestStatus': u'Waiting'
}
})
## make transfer Done
oRequest.subRequests["transfer"][0]["Attributes"]["Status"] = "Done"
oRequest.subRequests["transfer"][0]["Files"][0]["Status"] = "Done"
update = self.requestDB.updateRequest(
getRemoval["Value"]["RequestName"],
oRequest.toXML()["Value"])
## get status
self.assertEqual(
self.requestDB.getRequestStatus("testRequest"), {
'OK': True,
'Value': {
'SubRequestStatus': 'Done',
'RequestStatus': 'Done'
}
})
def test_02_getRequest(self):
""" getRequest and JobID """
getRequest = self.requestDB.getRequest("transfer")
self.assertEqual(getRequest["OK"], True)
self.assertEqual(getRequest["Value"]["JobID"], 0)
