def testGetDirlistNoFile( self ):
''' Dirlist : testing the getDirlist function for directory with no dirlist file '''
dl = Dirlist( Data1Dir )
res = dl.getDirlist()
#~ print dl.generateOutput( res )
#~ print dl.generateOutput( Data1Result )
self.assertEqual( res, Data1Result, "Bad data read from dirlist file" )
os.remove( os.path.join( Data1Dir, '.dirlist.md5' ) )
def main():
d = diskwalk('.')
dirs = d.enumerateDir()
print "processing %d directories" % len( dirs )
print dirs
for Dir in dirs:
print "Processing dir:", Dir
dl = Dirlist( Dir )
dl.getDirlist()
return 0
def testGetDirlistBadLink( self ):
''' Dirlist : testing the getDirlist function for directory with a bad link '''
if os.path.isfile( os.path.join( Data3Dir, '.dirlist.md5' ) ):
os.remove( os.path.join( Data3Dir, '.dirlist.md5' ) )
dl = Dirlist( Data3Dir )
res = dl.getDirlist()
#~ print res
#~ print Data3Result
self.assertEqual( res, Data3Result )
os.remove( os.path.join( Data3Dir, '.dirlist.md5' ) )
def testWriteDirlistToFile( self ):
''' Dirlist : test writes the dirlist to file '''
# cleanup before we start
if os.path.isfile( os.path.join( Data1Dir, '.dirlist.md5' ) ):
os.remove( os.path.join( Data1Dir, '.dirlist.md5' ) )
dl = Dirlist( Data1Dir )
self.assertFalse( dl.isDirlistPresent() ) # prerequisite for the test
dl.writeDirlistToFile( Data1Result )
self.assertEqual( dl.readDirlistFromFile(), Data1Result, "Bad data read from dirlist file" )
# cleanup
os.remove( os.path.join( Data1Dir, '.dirlist.md5' ) )
def testDirlistOutput( self ):
''' Dirlist : Testing file content output function '''
dl = Dirlist( Data1Dir )
DirlistData = dl.createDirlist()
output = dl.generateOutput( DirlistData )
self.assertEqual( output, Data1Output, "Output does not match expectations" )
def testGenerateDirlist( self ):
''' Dirlist : create the dirlistfile '''
dl = Dirlist( Data1Dir )
res = dl.createDirlist()
#self.assertEqual( res, Data1Result, "Processed data does not match expectations" )
self.assertEqual( res, Data1Result )
def testInstantiate( self ):
""" Dirlist : Simple instantiation and basedir check """
dl = Dirlist( Data1Dir )
self.assertEqual( dl.getBasedir(), Data1Dir, "Basedir and supplied dir does not match");
def testGetDirlistUsingFile( self ):
''' Dirlist : testing the getDirlist function for directory with a dirlist file '''
dl = Dirlist( Data2Dir )
self.assertEqual( dl.getDirlist(), Data2Result, "Bad data read from dirlist file" )
def testGetDirlistFromFile( self ):
''' Dirlist : test reads the dirlist from file '''
dl = Dirlist( Data2Dir )
self.assertEqual( dl.readDirlistFromFile(), Data2Result )
def testCheckNonexistingDirlistFile( self ):
''' Dirlist : check for non-existing dirlist file '''
dl = Dirlist( Data1Dir )
self.assertFalse( dl.isDirlistPresent() )
def testCheckExistingDirlistFile( self ):
''' Dirlist : check for existing dirlist file '''
dl = Dirlist( Data2Dir )
self.assertTrue( dl.isDirlistPresent() )
def testGetDirlistFromOutput( self ):
''' Dirlist : test use the output to return to the processed data '''
dl = Dirlist( Data1Dir )
DirlistData = dl.createDirlist()
Output = dl.generateOutput( DirlistData )
self.assertEqual( dl.extractDirlist( Output ), DirlistData, "Extracted data does not match input" )
