blob: a3367852addbe2d2a82f63efd2387882f23a671d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
import contextlib
import shutil
import tempfile
# This helper can be replaced by pytest tmpdir fixture
# once Bug 1536029 lands (@mock.patch disturbs pytest fixtures)
@contextlib.contextmanager
def tempdir():
dest_dir = tempfile.mkdtemp()
try:
yield dest_dir
finally:
shutil.rmtree(dest_dir, ignore_errors=True)
|