def test_mask_secret_simple_positive():
"""
Checking that mask_secret function works in a simple positive case.
"""
secrets = ["8bca8d2e-1cd6", "683c08d7-bc07"]
cmd = "ls -lh /tmp/8bca8d2e /tmp/683c08d7-bc07 /1cd6-4ec0-8e55"
cmd_masked_expected = "ls -lh /tmp/8bca8d2e /tmp/***** /1cd6-4ec0-8e55"
assert utils.mask_secrets(cmd, secrets) == cmd_masked_expected
def test_mask_secret_nomatch():
"""
Checking that mask_secret function works when there is no match.
"""
secrets = [
"8bca8d2e-1cd6-4ec0-8e55-9614aa01cf88",
"683c08d7-bc07-4d72-b098-46ef00b74aec",
]
assert utils.mask_secrets("ls -lh /tmp", secrets) == "ls -lh /tmp"
def test_mask_secret_nosecrets():
"""
Checking that mask_secret function doesn't change plaintext when secrets
are not specified.
"""
assert utils.mask_secrets("ls -lh /tmp", None) == "ls -lh /tmp"
def test_mask_secret_null():
"""
Checking that mask_secret function works with empty arguments.
"""
assert utils.mask_secrets("", None) == ""