106
106
if not os.path.exists(directory):
107
107
os.makedirs(directory)
109
def setUp(self, parentDir='.'):
109
def setUp(self, parentDir=None):
110
if parentDir is None:
111
parentDir = self.mktemp()
112
self.parent = parentDir
110
113
self.createFiles(self.files, parentDir)
112
def removeFiles(self, files, parentDir):
114
for filename, _ in self.files:
115
dirName = os.path.dirname(os.path.join(parentDir, filename))
116
directories[dirName] = True
117
dirs = directories.keys()
120
for directory in dirs:
121
shutil.rmtree(directory)
123
def tearDown(self, parentDir='.'):
124
116
self.cleanUpModules()
125
self.removeFiles(self.files, parentDir)
118
class SysPathManglingTest(PackageTest):
119
def setUp(self, parent=None):
120
self.oldPath = sys.path[:]
121
self.newPath = sys.path[:]
123
parent = self.mktemp()
124
PackageTest.setUp(self, parent)
125
self.newPath.append(self.parent)
126
self.mangleSysPath(self.newPath)
129
PackageTest.tearDown(self)
130
self.mangleSysPath(self.oldPath)
132
def mangleSysPath(self, pathVar):
133
sys.path[:] = pathVar