The Tor Browser / file comparison
comparison: testing/mozbase/mozfile/tests/stubs.py
testing/mozbase/mozfile/tests/stubs.py
- changeset 0
- 6474c204b198
equal
deleted
inserted
replaced
| -1:000000000000 | 0:f862849b426a |
|---|---|
| 1 import os | |
| 2 import shutil | |
| 3 import tempfile | |
| 4 | |
| 5 | |
| 6 # stub file paths | |
| 7 files = [('foo.txt',), | |
| 8 ('foo', 'bar.txt',), | |
| 9 ('foo', 'bar', 'fleem.txt',), | |
| 10 ('foobar', 'fleem.txt',), | |
| 11 ('bar.txt',), | |
| 12 ('nested_tree', 'bar', 'fleem.txt',), | |
| 13 ('readonly.txt',), | |
| 14 ] | |
| 15 | |
| 16 | |
| 17 def create_stub(): | |
| 18 """create a stub directory""" | |
| 19 | |
| 20 tempdir = tempfile.mkdtemp() | |
| 21 try: | |
| 22 for path in files: | |
| 23 fullpath = os.path.join(tempdir, *path) | |
| 24 dirname = os.path.dirname(fullpath) | |
| 25 if not os.path.exists(dirname): | |
| 26 os.makedirs(dirname) | |
| 27 contents = path[-1] | |
| 28 f = file(fullpath, 'w') | |
| 29 f.write(contents) | |
| 30 f.close() | |
| 31 return tempdir | |
| 32 except Exception, e: | |
| 33 try: | |
| 34 shutil.rmtree(tempdir) | |
| 35 except: | |
| 36 pass | |
| 37 raise e |
