Пример #1
0
 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" , "" ) )
Пример #2
0
 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" )
Пример #3
0
 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")
Пример #4
0
 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", ""))
Пример #5
0
    def initializeHandler(cls, serviceInfoDict):
        """ initialize handler """

        try:
            cls.ftsDB = FTSDB()
        except RuntimeError, error:
            gLogger.exception(error)
            return S_ERROR(error)
Пример #6
0
  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"
Пример #7
0
 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)
Пример #8
0
  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" )
Пример #9
0
    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"
Пример #10
0
    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")
Пример #11
0
  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" )
Пример #12
0
    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")