unittests: add utility context manager for changing directories

Using this context manager ensures that you change back to the original
directory when leaving the context manager

Signed-off-by: Dylan Baker <dylanx.c.baker@intel.com>

1 files changed, 16 insertions, 0 deletions

diff --git a/unittests/utils/nose.py b/unittests/utils/nose.py
index 42982960a..701b567c9 100644
--- a/unittests/utils/nose.py
+++ b/unittests/utils/nose.py
@@ -404,3 +404,19 @@ def capture_stderr(func):
             sys.stderr = restore
 
     return _inner
+
+
+@contextmanager
+def chdir(goto):
+    """chdir to a directory and back, guaranteed.
+
+    This contextmanager ensures that after changing directory you cahgne back,
+    using a try/finally block.
+
+    """
+    returnto = getcwd()
+    os.chdir(goto)
+    try:
+        yield
+    finally:
+        os.chdir(returnto)