def initializeHandler(cls, serviceInfoDict):
""" initialize handler """
try:
from DIRAC.DataManagementSystem.DB.FTSDB import FTSDB
cls.__ftsDB = FTSDB()
except RuntimeError, error:
gLogger.exception(error)
return S_ERROR(error)
def initializeHandler(cls, serviceInfoDict):
""" initialize handler """
try:
cls.ftsDB = FTSDB()
except RuntimeError, error:
gLogger.exception(error)
return S_ERROR(error)
def test03FTSHistory(self):
""" history view """
return
db = FTSDB()
ret = db.getFTSHistory()
self.assertEqual(ret["OK"], True, "getFTSHistory failed")
for ftsHistory in ret["Value"]:
self.assertEqual(isinstance(ftsHistory, FTSHistoryView), True,
"getFTSHistory wrong instance")
def test01Create(self):
""" test create tables and views """
db = FTSDB()
createTables = db.createTables(["FTSSite", "FTSFile", "FTSJob"], True)
self.assertEqual(
createTables["OK"], True,
"tables creation error: %s" % createTables.get("Message", ""))
createViews = db.createViews(True)
self.assertEqual(
createViews["OK"], True,
"views creation error: %s" % createViews.get("Message", ""))
def test04GetFTSIDs(self):
""" get ids """
return
db = FTSDB()
ftsJobIDs = db.getFTSJobIDs(["Submitted"])
self.assertEqual(ftsJobIDs["OK"], True, "getFTSJobIDs error")
self.assertEqual(len(ftsJobIDs["Value"]), self.submitted,
"getFTSJobIDs wrong value returned")
ftsFileIDs = db.getFTSFileIDs(["Waiting"])
self.assertEqual(ftsFileIDs["OK"], True, "getFTSFileIDs error")
self.assertEqual(type(ftsFileIDs["Value"]), list,
"getFTSFileIDs wrong value returned")
ftsFileList = db.getFTSFileList(["Waiting.*"])
self.assertEqual(ftsFileList["OK"], True, "getFTSFileList error")
self.assertEqual(type(ftsFileList["Value"]), list,
"fgetFTSFileList wrong value returned")
print "read ", len(ftsFileList["Value"]), " records"
def test05Delete(self):
""" delete files and jobs """
return
db = FTSDB()
for i in range(1, 301):
delete = db.deleteFTSFile(i)
self.assertEqual(delete["OK"], True, "deleleFTSFile failed")
# for i in range( 1, 201 ):
# delete = db.deleteFTSJob( i )
# self.assertEqual( delete["OK"], True, "deleleFTSJob failed" )
summary = db.getDBSummary()
self.assertEqual(summary["OK"], True, "getDBSummary failed")
self.assertEqual("FTSJob" in summary["Value"], True,
"getDBSummary FTSJob missing")
self.assertEqual("FTSFile" in summary["Value"], True,
"getDBSummary FTSFile missing")
self.assertEqual("FTSHistory" in summary["Value"], True,
"getDBSummary FTSHistory missing")
self.assertEqual("FTSSite" in summary["Value"], True,
"getDBSummary FTSSite missing")
def test02PutGetDelete(self):
""" put, get, peek methods """
db = FTSDB()
print "put site"
for ftsSite in self.ftsSites:
put = db.putFTSSite(ftsSite)
self.assertEqual(put["OK"], True, "putFTSSite failed")
put = db.putFTSSite(ftsSite)
self.assertEqual(put["OK"], False, "re-putFTSSite failed")
return
print "getFTSSite"
for i, ftsSite in enumerate(self.ftsSites):
get = db.getFTSSite(i + 1)
self.assertEqual(get["OK"], True, "getFTSSite failed")
self.assertEqual(isinstance(get["Value"], FTSSite), True,
"getFTSSite wrong value returned")
print "getFTSSiteList"
getFTSSitesList = db.getFTSSitesList()
self.assertEqual(getFTSSitesList["OK"], True, "getFTSSitesList failed")
for item in getFTSSitesList["Value"]:
self.assertEqual(isinstance(item, FTSSite), True,
"getFTSSitesList wrong value returned")
print "putFile"
for ftsFile in self.ftsFiles:
put = db.putFTSFile(ftsFile)
self.assertEqual(put["OK"], True, "putFTSFile failed")
print "peekFile"
for i in range(1, 101):
peek = db.peekFTSFile(i)
self.assertEqual(peek["OK"], True, "peekFTSFile failed")
self.assertEqual(isinstance(peek["Value"], FTSFile), True,
"peekFTSFile wrong value")
print "getFile"
for i in range(1, 101):
get = db.getFTSFile(i)
self.assertEqual(get["OK"], True, "getFTSFile failed")
self.assertEqual(isinstance(get["Value"], FTSFile), True,
"getFTSFile wrong value")
print "putJob"
for ftsJob in self.ftsJobs:
put = db.putFTSJob(ftsJob)
self.assertEqual(put["OK"], True, "putFTSJob failed")
print "peekJob"
for i in range(1, 101):
peek = db.peekFTSJob(i)
self.assertEqual(peek["OK"], True, "peekFTSJob failed")
self.assertEqual(isinstance(peek["Value"], FTSJob), True,
"peekFTSJob wrong value returned")
self.assertEqual(len(peek["Value"]), 1,
"peekFTSJob wrong number of files ")
print "getJob"
for i in range(1, 101):
get = db.getFTSJob(i)
self.assertEqual(get["OK"], True, "getFTSJob failed")
self.assertEqual(isinstance(get["Value"], FTSJob), True,
"getFTSJob wrong value returned")
self.assertEqual(len(get["Value"]), 1,
"getFTSJob wrong number of files")
print "duSummary"
summary = db.getDBSummary()
self.assertEqual(summary["OK"], True, "getDBSummary failed")
self.assertEqual("FTSJob" in summary["Value"], True,
"getDBSummary FTSJob missing")
self.assertEqual("FTSFile" in summary["Value"], True,
"getDBSummary FTSFile missing")
self.assertEqual("FTSHistory" in summary["Value"], True,
"getDBSummary FTSHistory missing")
self.assertEqual("FTSSite" in summary["Value"], True,
"getDBSummary FTSSite missing")
