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
65
66
|
# 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
import pytest
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
|
