Commit 6aea437b authored by guyroz's avatar guyroz

fixing tests

--HG--
branch : distribute
extra : rebase_source : 5d3983c1447f6389c00b416e5942f1764eaf9ece
parents c0032c0d 8b25dd9f
...@@ -67,7 +67,7 @@ class TestEasyInstallTest(unittest.TestCase): ...@@ -67,7 +67,7 @@ class TestEasyInstallTest(unittest.TestCase):
old_platform = sys.platform old_platform = sys.platform
try: try:
name, script = get_script_args(dist).next() name, script = get_script_args(dist).next()[0:2]
finally: finally:
sys.platform = old_platform sys.platform = old_platform
...@@ -141,9 +141,13 @@ class TestPTHFileWriter(unittest.TestCase): ...@@ -141,9 +141,13 @@ class TestPTHFileWriter(unittest.TestCase):
self.assert_(pth.dirty) self.assert_(pth.dirty)
def test_add_from_site_is_ignored(self): def test_add_from_site_is_ignored(self):
pth = PthDistributions('does-not_exist', ['/test/location/does-not-have-to-exist']) if os.name != 'nt':
location = '/test/location/does-not-have-to-exist'
else:
location = 'c:\\does_not_exist'
pth = PthDistributions('does-not_exist', [location, ])
self.assert_(not pth.dirty) self.assert_(not pth.dirty)
pth.add(PRDistribution('/test/location/does-not-have-to-exist')) pth.add(PRDistribution(location))
self.assert_(not pth.dirty) self.assert_(not pth.dirty)
...@@ -221,7 +225,7 @@ class TestUserInstallTest(unittest.TestCase): ...@@ -221,7 +225,7 @@ class TestUserInstallTest(unittest.TestCase):
sys.path.append(target) sys.path.append(target)
old_ppath = os.environ.get('PYTHONPATH') old_ppath = os.environ.get('PYTHONPATH')
os.environ['PYTHONPATH'] = ':'.join(sys.path) os.environ['PYTHONPATH'] = os.path.pathsep.join(sys.path)
try: try:
dist = Distribution() dist = Distribution()
dist.script_name = 'setup.py' dist.script_name = 'setup.py'
...@@ -234,8 +238,13 @@ class TestUserInstallTest(unittest.TestCase): ...@@ -234,8 +238,13 @@ class TestUserInstallTest(unittest.TestCase):
self.assertEquals(res.location, new_location) self.assertEquals(res.location, new_location)
finally: finally:
sys.path.remove(target) sys.path.remove(target)
shutil.rmtree(new_location) for basedir in [new_location, target, ]:
shutil.rmtree(target) if not os.path.exists(basedir) or not os.path.isdir(basedir):
continue
try:
shutil.rmtree(basedir)
except:
pass
if old_ppath is not None: if old_ppath is not None:
os.environ['PYTHONPATH'] = old_ppath os.environ['PYTHONPATH'] = old_ppath
else: else:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment