def test__exit(self):
"Test TemporaryDirectory __exit__() method"
tdir = TemporaryDirectory()
trace = """ Called tempfile.mkdtemp('', 'tmp', None)
Called shutil.rmtree('/fake/tmp/dir')"""
tdir.__exit__(None, None, None)
assert_same_trace(self.tt, trace)
def test__exit(self):
"Test TemporaryDirectory __exit__() method"
tdir = TemporaryDirectory()
trace = """ Called tempfile.mkdtemp('', 'tmp', None)
Called shutil.rmtree('/fake/tmp/dir')"""
tdir.__exit__(None, None, None)
assert_same_trace(self.tt, trace)
def test_change_cwd(self):
"Test TemporaryDirectory changing the cwd"
expected = "/fake/tmp/dir"
trace = """ Called tempfile.mkdtemp('', 'tmp', None)
Called os.getcwd()
Called os.chdir('/fake/tmp/dir')
Called os.chdir('/old/cur/dir')
Called shutil.rmtree('/fake/tmp/dir')"""
tdir = TemporaryDirectory(change=True)
tdir.__enter__()
self.assertEqual(self.cwd, expected)
self.assertEqual(tdir.old_wd, '/old/cur/dir')
tdir.__exit__(None, None, None)
self.assertEqual(self.cwd, '/old/cur/dir')
assert_same_trace(self.tt, trace)
def test_change_cwd(self):
"Test TemporaryDirectory changing the cwd"
expected = "/fake/tmp/dir"
trace = """ Called tempfile.mkdtemp('', 'tmp', None)
Called os.getcwd()
Called os.chdir('/fake/tmp/dir')
Called os.chdir('/old/cur/dir')
Called shutil.rmtree('/fake/tmp/dir')"""
tdir = TemporaryDirectory(change=True)
tdir.__enter__()
self.assertEqual(self.cwd, expected)
self.assertEqual(tdir.old_wd, '/old/cur/dir')
tdir.__exit__(None, None, None)
self.assertEqual(self.cwd, '/old/cur/dir')
assert_same_trace(self.tt, trace)
