mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 18:51:29 +00:00
3be09c6d74
Logging was missing from __init__.py which broke under mac. This adds that back in and remove unused imports from other files. The os check is also moved up to be the first thing we look for since is_google_host does not work with mac at the moment. Bug:326277821 Change-Id: I3f28838d367eb9583332fd33ad35d71c0851fe4b Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/6604498 Commit-Queue: Struan Shrimpton <sshrimp@google.com> Reviewed-by: Ben Pastene <bpastene@chromium.org>
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
# Copyright 2024 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
"""Test the config and anonymizer utils."""
|
|
|
|
import getpass
|
|
import re
|
|
|
|
from .proto import trace_span_pb2
|
|
from . import anonymization
|
|
|
|
|
|
def test_anonymizing_filter_to_redact_info_from_msg() -> None:
|
|
"""Test AnonymizingFilter to apply the passed anonymizer to msg."""
|
|
msg = trace_span_pb2.TraceSpan()
|
|
msg.name = "log-user-user1234"
|
|
|
|
anonymizer = anonymization.Anonymizer([(re.escape("user1234"), "<user>")])
|
|
f = anonymization.AnonymizingFilter(anonymizer)
|
|
|
|
filtered_msg = f(msg)
|
|
assert filtered_msg.name == "log-user-<user>"
|
|
|
|
|
|
def test_default_anonymizer_to_remove_username_from_path(monkeypatch) -> None:
|
|
"""Test that default Anonymizer redacts username."""
|
|
monkeypatch.setattr(getpass, "getuser", lambda: "user")
|
|
|
|
a = anonymization.Anonymizer()
|
|
output = a.apply("/home/user/docs")
|
|
|
|
assert output == "/home/<user>/docs"
|
|
|
|
|
|
def test_anonymizer_to_apply_passed_replacements() -> None:
|
|
"""Test anonymizer to apply the requested replacements."""
|
|
text = "/home/%s/docs" % getpass.getuser()
|
|
|
|
replacements = [(re.escape(getpass.getuser()), "<user>")]
|
|
a = anonymization.Anonymizer(replacements=replacements)
|
|
output = a.apply(text)
|
|
|
|
assert output == "/home/<user>/docs"
|
|
|
|
|
|
def test_anonymizer_to_apply_multiple_replacements() -> None:
|
|
"""Test anonymizer to apply the passed replacements in order."""
|
|
replacements = [(re.escape("abc"), "x"), (re.escape("xyz"), "t")]
|
|
text = "hello abcd. how is xyz. abcyz"
|
|
|
|
a = anonymization.Anonymizer(replacements=replacements)
|
|
output = a.apply(text)
|
|
|
|
assert output == "hello xd. how is t. t"
|
|
|
|
|
|
def test_default_anonymizer_skip_root(monkeypatch) -> None:
|
|
"""Test the anonymizer skips the root user."""
|
|
monkeypatch.setattr(getpass, "getuser", lambda: "root")
|
|
|
|
text = "/root/home service.sysroot.SetupBoard"
|
|
a = anonymization.Anonymizer()
|
|
output = a.apply(text)
|
|
|
|
assert output == text
|