1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
from unittest import mock
from devtools_testutils import (
test_proxy,
add_body_key_sanitizer,
add_oauth_response_sanitizer,
remove_batch_sanitizers,
add_uri_regex_sanitizer,
is_live,
)
import pytest
# autouse=True will trigger this fixture on each pytest run, even if it's not explicitly used by a test method
@pytest.fixture(scope="session", autouse=True)
def start_proxy(test_proxy):
add_body_key_sanitizer(json_path="$..id_token", value="Sanitized")
add_body_key_sanitizer(json_path="$..client_info", value="Sanitized")
add_oauth_response_sanitizer()
add_uri_regex_sanitizer(regex="\\.(?<location>.*)\\.devcenter\\.azure\\.com", group_for_replace="location", value="location")
# Remove the following sanitizers since certain fields are needed in tests and are non-sensitive:
# - AZSDK2003: Location
# - AZSDK3493: $..name
remove_batch_sanitizers(["AZSDK2003", "AZSDK3493"])
return
@pytest.fixture(scope="session", autouse=True)
def patch_async_sleep():
async def immediate_return(_):
return
if not is_live():
with mock.patch("asyncio.sleep", immediate_return):
yield
else:
yield
@pytest.fixture(scope="session", autouse=True)
def patch_sleep():
def immediate_return(_):
return
if not is_live():
with mock.patch("time.sleep", immediate_return):
yield
else:
yield
|
