diff options
Diffstat (limited to 'tests/testutil.py')
-rw-r--r-- | tests/testutil.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/testutil.py b/tests/testutil.py index 092a4d2..bde84f7 100644 --- a/tests/testutil.py +++ b/tests/testutil.py @@ -11,6 +11,7 @@ import os import os.path import random import shutil +import stat import tempfile import threading import unittest @@ -21,6 +22,17 @@ from http.server import HTTPServer, BaseHTTPRequestHandler from urllib.parse import urlparse, parse_qs +def disallow_writes(path): + """Mark path non-writable, recursively""" + for dirpath, dirs, files in os.walk(path, topdown=False): + for f in files + dirs: + st = os.lstat(os.path.join(dirpath, f)) + if not stat.S_ISLNK(st.st_mode): + os.chmod(os.path.join(dirpath, f), + st.st_mode & ~0o222) + os.chmod(path, 0o555) + + class LoggingTestCase(unittest.TestCase): def setUp(self): self.log = io.StringIO() |