cpython/Lib/test/test_shutil.py
Guido van Rossum 8cec3ab0e4 - Bug #981530: Fix UnboundLocalError in shutil.rmtree(). This affects
the documented behavior: the function passed to the onerror()
  handler can now also be os.listdir.

[I could've sworn I checked this in, but apparently I didn't, or it
got lost???]
2004-07-14 00:48:58 +00:00

35 lines
923 B
Python

# Copyright (C) 2003 Python Software Foundation
import unittest
import shutil
import tempfile
import os
import os.path
from test import test_support
class TestShutil(unittest.TestCase):
def test_rmtree_errors(self):
# filename is guaranteed not to exist
filename = tempfile.mktemp()
self.assertRaises(OSError, shutil.rmtree, filename)
self.assertEqual(shutil.rmtree(filename, True), None)
shutil.rmtree(filename, False, lambda func, arg, exc: None)
def test_dont_move_dir_in_itself(self):
src_dir = tempfile.mkdtemp()
try:
dst = os.path.join(src_dir, 'foo')
self.assertRaises(shutil.Error, shutil.move, src_dir, dst)
finally:
try:
os.rmdir(src_dir)
except:
pass
def test_main():
test_support.run_unittest(TestShutil)
if __name__ == '__main__':
test_main()