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
55
56
57
58
59
60
61
62
63
64
|
# 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.
# --------------------------------------------------------------------------
import logging
import os
import pytest
from devtools_testutils import (
add_general_regex_sanitizer,
add_body_key_sanitizer,
add_uri_regex_sanitizer,
test_proxy,
is_live,
remove_batch_sanitizers,
)
from testcase import import_image, is_public_endpoint
from constants import HELLO_WORLD
logger = logging.getLogger()
@pytest.fixture(scope="session", autouse=True)
def load_registry():
if not is_live():
return
logger.info("loading registry")
endpoint = os.environ.get("CONTAINERREGISTRY_ENDPOINT")
endpoint_anon = os.environ.get("CONTAINERREGISTRY_ANONREGISTRY_ENDPOINT")
repo = HELLO_WORLD
tags = ["latest", "v1"]
try:
import_image(endpoint, repo, tags)
if is_public_endpoint(endpoint_anon):
import_image(endpoint_anon, repo, tags)
except Exception as e:
logger.exception(e)
raise
@pytest.fixture(scope="session", autouse=True)
def add_sanitizers(test_proxy):
# sanitizes access and refresh tokens that are present in single-string request or response bodies
# we expect tokens to either be at the end of this body string, or followed by "\u0026" and more content
add_general_regex_sanitizer(value="access_token", regex="(?<=access_token=).*?(?=(?:\\u0026|$))")
add_general_regex_sanitizer(value="refresh_token", regex="(?<=refresh_token=).*?(?=(?:\\u0026|$))")
# sanitizes access and refresh tokens in JSON-formatted request or response bodies
add_body_key_sanitizer(json_path="$..access_token", value="access_token")
add_body_key_sanitizer(json_path="$..refresh_token", value="refresh_token")
add_uri_regex_sanitizer(regex="sha256%3A(.*)", value="sha256_encoded_stream")
client_id = os.environ.get("CONTAINERREGISTRY_CLIENT_ID", "client-id")
add_general_regex_sanitizer(regex=client_id, value="client-id")
client_secret = os.environ.get("CONTAINERREGISTRY_CLIENT_SECRET", "client-secret")
add_general_regex_sanitizer(regex=client_secret, value="client-secret")
tenant_id = os.environ.get("CONTAINERREGISTRY_TENANT_ID", "tenant-id")
add_general_regex_sanitizer(regex=tenant_id, value="00000000-0000-0000-0000-000000000000")
# Remove the following sanitizers since certain fields are needed in tests and are non-sensitive:
# - AZSDK3493: $..name
# - AZSDK2003: Location
remove_batch_sanitizers(["AZSDK3493", "AZSDK2003"])
|
