summaryrefslogtreecommitdiffstats
path: root/testing/mozbase/mozproxy/tests/support.py
blob: 6471c080ef596497eb7a7880325621f3c96896ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from __future__ import absolute_import, print_function
import shutil
import contextlib
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)