import unittest
import sys, os
sys.path.append('../code')
import directory
TESTDIR = os.path.realpath(os.path.join(os.path.dirname(sys.argv[0]), 'testdir'))
NONEXISTANT_DIR = '/this/directory/will/most/certainly/not/exist'
class Test1(unittest.TestCase):
def testInitialCondition(self):
# Check for the expected initial condition
dir = directory.Directory(TESTDIR)
self.assertEqual(dir.path, TESTDIR)
self.assertFalse(dir.files_loaded)
self.assertEqual(dir.files, None)
self.assertRaises(directory.NotLoadedYet, len, dir)
self.assertRaises(directory.NotLoadedYet, dir.__getitem__, 0)
def testAfterFilesLoaded(self):
# Check whether the directory has the correct list of files.
dir = directory.Directory(TESTDIR)
dir.load_files()
self.assertTrue(dir.exists)
self.assertEqual(type(dir.files), list)
# Get the files you expect it to have and sort both before
# comparing. I don't expect any order after only loading the files.
assumed_files = os.listdir(TESTDIR)
assumed_files.sort()
dir.files.sort()
self.assertTrue(len(dir) > 0)
self.assertEqual(dir.files, assumed_files)
def testNonexistantDir(self):
dir = directory.Directory(NONEXISTANT_DIR)
dir.load_files()
self.assertTrue(dir.files_loaded)
self.assertFalse(dir.exists)
self.assertFalse(dir.accessible)
self.assertEqual(dir.files, None)
self.assertRaises(directory.NotLoadedYet, len, dir)
self.assertRaises(directory.NotLoadedYet, dir.__getitem__, 0)
def testModifyFrozenClone(self):
dir = directory.Directory(TESTDIR)
clone = dir.frozenClone()
# assert that their attributes are equal, except for frozen, which
# should be true for the clone.
self.assertTrue(clone.frozen)
clone.frozen = False
self.assertEqual(dir.__dict__, clone.__dict__)
clone.frozen = True
# check for inequality after loading files with one object
self.assertEqual(dir.files, clone.files)
dir.load_files()
self.assertNotEqual(dir.files, clone.files)
self.assertRaises(directory.FrozenException, clone.load_files)
unittest.main()