def test_basic(self):
with mock.patch('os.getcwd', return_value='cwd'), \
mock.patch('os.chdir') as os_chdir: # noqa
with path.pushd('foo'):
self.assertEqual(os_chdir.mock_calls, [mock.call('foo')])
self.assertEqual(
os_chdir.mock_calls,
[mock.call('foo'), mock.call('cwd')])
def test_exception(self):
with mock.patch('os.getcwd', return_value='cwd'), \
mock.patch('os.chdir') as os_chdir: # noqa
with self.assertRaises(ValueError):
with path.pushd('foo'):
self.assertEqual(os_chdir.mock_calls, [mock.call('foo')])
raise ValueError('uh oh!')
self.assertEqual(
os_chdir.mock_calls,
[mock.call('foo'), mock.call('cwd')])
def test_makedirs(self):
with mock.patch('os.makedirs') as os_makedirs, \
mock.patch('os.getcwd', return_value='cwd'), \
mock.patch('os.chdir') as os_chdir: # noqa
with path.pushd('foo', makedirs=True):
self.assertEqual(os_makedirs.mock_calls,
[mock.call('foo', 0o777)])
self.assertEqual(os_chdir.mock_calls, [mock.call('foo')])
self.assertEqual(
os_chdir.mock_calls,
[mock.call('foo'), mock.call('cwd')])
