Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 1 | # Copyright (C) 2003 Python Software Foundation |
| 2 | |
| 3 | import unittest |
| 4 | import shutil |
| 5 | import tempfile |
Brett Cannon | 1c3fa18 | 2004-06-19 21:11:35 +0000 | [diff] [blame] | 6 | import os |
| 7 | import os.path |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 8 | from test import test_support |
| 9 | |
| 10 | class TestShutil(unittest.TestCase): |
| 11 | def test_rmtree_errors(self): |
| 12 | # filename is guaranteed not to exist |
| 13 | filename = tempfile.mktemp() |
| 14 | self.assertRaises(OSError, shutil.rmtree, filename) |
| 15 | self.assertEqual(shutil.rmtree(filename, True), None) |
Guido van Rossum | 8cec3ab | 2004-07-14 00:48:58 +0000 | [diff] [blame^] | 16 | shutil.rmtree(filename, False, lambda func, arg, exc: None) |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 17 | |
Brett Cannon | 1c3fa18 | 2004-06-19 21:11:35 +0000 | [diff] [blame] | 18 | def test_dont_move_dir_in_itself(self): |
| 19 | src_dir = tempfile.mkdtemp() |
| 20 | try: |
| 21 | dst = os.path.join(src_dir, 'foo') |
| 22 | self.assertRaises(shutil.Error, shutil.move, src_dir, dst) |
| 23 | finally: |
| 24 | try: |
| 25 | os.rmdir(src_dir) |
| 26 | except: |
| 27 | pass |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 28 | |
Brett Cannon | 1c3fa18 | 2004-06-19 21:11:35 +0000 | [diff] [blame] | 29 | |
| 30 | |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 31 | def test_main(): |
Walter Dörwald | 21d3a32 | 2003-05-01 17:45:56 +0000 | [diff] [blame] | 32 | test_support.run_unittest(TestShutil) |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 33 | |
Barry Warsaw | 7fc2cca | 2003-01-24 17:34:13 +0000 | [diff] [blame] | 34 | if __name__ == '__main__': |
Walter Dörwald | 21d3a32 | 2003-05-01 17:45:56 +0000 | [diff] [blame] | 35 | test_main() |