Pytest: fix local params overriden (#30312)

* fix local params being overriden

* just reset prefix after completion
This commit is contained in:
Justin Newberry
2023-10-23 20:41:19 -04:00
committed by GitHub
parent 1802de3f01
commit 412f4cbc1a
2 changed files with 8 additions and 0 deletions
+3
View File
@@ -14,6 +14,7 @@ class OpenpilotPrefix:
self.clean_dirs_on_exit = clean_dirs_on_exit
def __enter__(self):
self.original_prefix = os.environ.get('OPENPILOT_PREFIX', None)
os.environ['OPENPILOT_PREFIX'] = self.prefix
try:
os.mkdir(self.msgq_path)
@@ -28,6 +29,8 @@ class OpenpilotPrefix:
self.clean_dirs()
try:
del os.environ['OPENPILOT_PREFIX']
if self.original_prefix is not None:
os.environ['OPENPILOT_PREFIX'] = self.original_prefix
except KeyError:
pass
return False