Skip to content
Prev Previous commit
Next Next commit
fixup! bpo-1154351: add get_current_dir_name() to os module
  • Loading branch information
bradengroom committed Oct 26, 2018
commit 89e1984c064251740f5871b34ed1b9b3a819fe5a
29 changes: 12 additions & 17 deletions Lib/test/test_os.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
import uuid
import warnings
from test import support
from unittest import mock

try:
import resource
Expand Down Expand Up @@ -1255,7 +1256,6 @@ def tearDownClass(cls):
class CurrentDirTests(unittest.TestCase):

def setUp(self):
self.pwd = os.environ.get('PWD')
base = os.path.abspath(support.TESTFN)
self.tmp_dir = base + '_dir'
self.tmp_lnk = base + '_lnk'
Expand All @@ -1268,32 +1268,27 @@ def test_getcwd(self):
os.symlink(self.tmp_dir, self.tmp_lnk, True)
os.chdir(self.tmp_lnk)
self.assertEqual(self.tmp_dir, os.getcwd())
os.environ['PWD'] = self.tmp_dir
self.assertEqual(self.tmp_dir, os.getcwd())
os.environ['PWD'] = self.tmp_lnk
self.assertEqual(self.tmp_dir, os.getcwd())
with mock.patch.dict('os.environ', {'PWD': self.tmp_dir}):
self.assertEqual(self.tmp_dir, os.getcwd())
with mock.patch.dict('os.environ', {'PWD': self.tmp_lnk}):
self.assertEqual(self.tmp_dir, os.getcwd())
os.unlink(self.tmp_lnk)
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The link will be leaked in the case of failure. It is better to use addCleanup(). Add the following line just before creating a link.

self.addCleanup(support.unlink, self.tmp_lnk)


def test_get_current_dir_name(self):
# os.get_current_dir_name() returns the direct path--mirroring
# the PWD environment variable if it exists regardless of
# whether the path contains symlinks.
with support.temp_cwd(self.tmp_dir):
os.environ['PWD'] = self.tmp_dir
self.assertEqual(self.tmp_dir, os.get_current_dir_name())
os.symlink(self.tmp_dir, self.tmp_lnk, True)
if os.name == 'posix':
os.environ['PWD'] = self.tmp_lnk
self.assertEqual(self.tmp_lnk, os.get_current_dir_name())
else:
os.environ['PWD'] = self.tmp_lnk
with mock.patch.dict('os.environ', {'PWD': self.tmp_dir}):
self.assertEqual(self.tmp_dir, os.get_current_dir_name())
os.symlink(self.tmp_dir, self.tmp_lnk, True)
with mock.patch.dict('os.environ', {'PWD': self.tmp_lnk}):
if os.name == 'posix':
self.assertEqual(self.tmp_lnk, os.get_current_dir_name())
else:
self.assertEqual(self.tmp_dir, os.get_current_dir_name())
os.unlink(self.tmp_lnk)

def tearDown(self):
if self.pwd is not None:
os.environ['PWD'] = self.pwd


class RemoveDirsTests(unittest.TestCase):
def setUp(self):
Expand Down