Fix to proxy fixture/recorder

Azure · Dec 8, 2021 · 1afeb3b · 1afeb3b
1 parent 815c327
commit 1afeb3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tools/azure-sdk-tools/devtools_testutils/proxy_docker_startup.py b/tools/azure-sdk-tools/devtools_testutils/proxy_docker_startup.py
@@ -173,3 +173,5 @@ def test_proxy():
         # Everything after it will be run after invoking fixtures are done executing
         yield
         stop_test_proxy()
+    else:
+        yield
diff --git a/tools/azure-sdk-tools/devtools_testutils/proxy_testcase.py b/tools/azure-sdk-tools/devtools_testutils/proxy_testcase.py
@@ -25,7 +25,7 @@
 
 # the trimming function to clean up incoming arguments to the test function we are wrapping
 from azure_devtools.scenario_tests.utilities import trim_kwargs_from_test_function
-from .helpers import is_live_and_not_recording
+from .helpers import is_live, is_live_and_not_recording
 from .config import PROXY_URL
 
 if TYPE_CHECKING:
-Original file line number
+Diff line change
@@ Expand Up / @@ -173,3 +173,5 @@ def test_proxy(): @@
             # Everything after it will be run after invoking fixtures are done executing
             yield
             stop_test_proxy()
+        else:
+            yield