Skip to content

Commit 2e2823e

Browse files
authored
[Test Proxy] Use string sanitizers for env vars (Azure#25824)
1 parent 35c5935 commit 2e2823e

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tools/azure-sdk-tools/devtools_testutils/envvariable_loader.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from . import AzureMgmtPreparer
1010
from azure_devtools.scenario_tests.exceptions import AzureTestError
1111
from dotenv import load_dotenv, find_dotenv
12-
from .sanitizers import add_general_regex_sanitizer
12+
from .sanitizers import add_general_string_sanitizer
1313

1414

1515
class EnvironmentVariableLoader(AzureMgmtPreparer):
@@ -76,9 +76,9 @@ def create_resource(self, name, **kwargs):
7676
# test proxy tests have no scrubber, and instead register sanitizers using fake values
7777
else:
7878
try:
79-
add_general_regex_sanitizer(
79+
add_general_string_sanitizer(
8080
value=scrubbed_value,
81-
regex=self.real_values[key.lower()],
81+
target=self.real_values[key.lower()],
8282
)
8383
except:
8484
logger = logging.getLogger()

tools/azure-sdk-tools/devtools_testutils/proxy_fixtures.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from .helpers import get_test_id, is_live, is_live_and_not_recording
2121
from .proxy_testcase import start_record_or_playback, stop_record_or_playback, transform_request
2222
from .proxy_startup import test_proxy
23-
from .sanitizers import add_general_regex_sanitizer
23+
from .sanitizers import add_general_string_sanitizer
2424

2525
if TYPE_CHECKING:
2626
from typing import Any, Callable, Dict, Optional, Tuple
@@ -45,7 +45,7 @@ def sanitize(self, variable: str, value: str) -> str:
4545
self._fake_values[variable] = value
4646
real_value = os.getenv(variable)
4747
if real_value:
48-
add_general_regex_sanitizer(regex=real_value, value=value)
48+
add_general_string_sanitizer(target=real_value, value=value)
4949
else:
5050
_LOGGER.info(f"No value for {variable} was found, so a sanitizer could not be registered for the variable.")
5151

0 commit comments

Comments
 (0)