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')])