-
Notifications
You must be signed in to change notification settings - Fork 2.8k
/
conftest.py
50 lines (45 loc) · 1.99 KB
/
conftest.py
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
# ------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# -------------------------------------------------------------------------
import sys
import pytest
from devtools_testutils import test_proxy
from devtools_testutils.sanitizers import (
add_remove_header_sanitizer,
add_general_regex_sanitizer,
add_oauth_response_sanitizer,
set_custom_default_matcher
)
collect_ignore = []
@pytest.fixture(scope="session", autouse=True)
def add_sanitizers(test_proxy):
add_remove_header_sanitizer(headers="aeg-sas-key")
add_remove_header_sanitizer(headers="aeg-sas-token")
add_remove_header_sanitizer(headers="ServiceBusSupplementaryAuthorization")
add_remove_header_sanitizer(headers="ServiceBusDlqSupplementaryAuthorization")
add_general_regex_sanitizer(
value="fakeresource",
regex="(?<=\\/\\/)[a-z-]+(?=\\.servicebus\\.windows\\.net)"
)
add_oauth_response_sanitizer()
# Only run stress tests on request.
if not any([arg.startswith('test_stress') or arg.endswith('StressTest') for arg in sys.argv]):
collect_ignore.append("stress/scripts")
# Allow us to pass stress_test_duration from the command line.
def pytest_addoption(parser):
parser.addoption('--stress_test_duration_seconds', action="store", default=None)
# Note: This is duplicated between here and the basic conftest, so that it does not throw warnings if you're
# running locally to this SDK. (Everything works properly, pytest just makes a bit of noise.)
def pytest_configure(config):
# register an additional marker
config.addinivalue_line(
"markers", "liveTest: mark test to be a live test only"
)
config.addinivalue_line(
"markers", "live_test_only: mark test to be a live test only"
)
config.addinivalue_line(
"markers", "playback_test_only: mark test to be a playback test only"
)