1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
import unittest
import sys, os, time
sys.path.append('../code')
os.stat_float_times(True)
import directory, fsobject, file, debug
TESTDIR = os.path.realpath(os.path.join(os.path.dirname(sys.argv[0]), 'testdir'))
TESTFILE = os.path.join(TESTDIR, 'testfile5234148')
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.content_loaded)
self.assertEqual(dir.filenames, None)
self.assertEqual(dir.files, None)
self.assertRaises(fsobject.NotLoadedYet, len, dir)
self.assertRaises(fsobject.NotLoadedYet, dir.__getitem__, 0)
def testAfterContentLoaded(self):
# Check whether the directory has the correct list of filenames.
dir = directory.Directory(TESTDIR)
dir.load_content()
self.assertTrue(dir.exists)
self.assertEqual(type(dir.filenames), list)
# Get the filenames you expect it to have and sort both before
# comparing. I don't expect any order after only loading the filenames.
assumed_filenames = os.listdir(TESTDIR)
assumed_filenames = list(map(lambda str: os.path.join(TESTDIR, str), assumed_filenames))
assumed_filenames.sort()
dir.filenames.sort()
self.assertTrue(len(dir) > 0)
self.assertEqual(dir.filenames, assumed_filenames)
# build a file object for each file in the list assumed_filenames
# and find exactly one equivalent in dir.files
for name in assumed_filenames:
f = file.File(name)
f.load()
equal = 0
for dirfile in dir.files:
if (f.__dict__ == dirfile.__dict__):
equal += 1
self.assertEqual(equal, 1)
def testNonexistantDir(self):
dir = directory.Directory(NONEXISTANT_DIR)
dir.load_content()
self.assertTrue(dir.content_loaded)
self.assertFalse(dir.exists)
self.assertFalse(dir.accessible)
self.assertEqual(dir.filenames, None)
self.assertRaises(fsobject.NotLoadedYet, len, dir)
self.assertRaises(fsobject.NotLoadedYet, dir.__getitem__, 0)
def testModifyFrozenClone(self):
dir = directory.Directory(TESTDIR)
clone = dir.frozen_clone()
# 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 filenames with one object
self.assertEqual(dir.filenames, clone.filenames)
dir.load_content()
self.assertNotEqual(dir.filenames, clone.filenames)
self.assertRaises(fsobject.FrozenException, clone.load_content)
def test_load_if_outdated(self):
# modify the directory. If the time between the last modification
# was within the filesystems resolution of mtime, we should have a re-load.
def modify_dir():
open(TESTFILE, 'w').close()
os.unlink(TESTFILE)
def mtime():
return os.stat(TESTDIR).st_mtime
dir = directory.Directory(TESTDIR)
dir.load()
# If the modification happens to be in the same second as the
# last modification, it will result in mtime having the same
# integer value. So we wait until the resolution is exceeded
# and mtime differs.
old_mtime = mtime()
for i in range(50):
modify_dir()
if old_mtime != mtime(): break
time.sleep(0.1)
else:
# fail after 5 seconds of trying
self.fail("Cannot perform test: mtime of TESTDIR is not being updated.")
self.assertTrue(dir.load_if_outdated())
unittest.main()
|