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
|
import pytest
from cloudinit import subp
from conftest import UnexpectedSubpError
class TestDisableSubpUsage:
"""Test that the disable_subp_usage fixture behaves as expected."""
def test_using_subp_raises_assertion_error(self):
with pytest.raises(UnexpectedSubpError):
subp.subp(["some", "args"])
def test_typeerrors_on_incorrect_usage(self):
with pytest.raises(TypeError):
# We are intentionally passing no value for a parameter, so:
# pylint: disable=no-value-for-parameter
subp.subp()
def test_subp_exception_escapes_exception_handling(self):
with pytest.raises(UnexpectedSubpError):
try:
subp.subp(["some", "args"])
except Exception:
pytest.fail("Unexpected exception raised")
@pytest.mark.allow_all_subp
def test_subp_usage_can_be_reenabled(self):
subp.subp(["whoami"])
@pytest.mark.allow_subp_for("whoami")
def test_subp_usage_can_be_conditionally_reenabled(self):
# The two parameters test each potential invocation with a single
# argument
with pytest.raises(UnexpectedSubpError) as excinfo:
subp.subp(["some", "args"])
assert "allowed: whoami" in str(excinfo.value)
subp.subp(["whoami"])
@pytest.mark.allow_subp_for("whoami", "sh")
def test_subp_usage_can_be_conditionally_reenabled_for_multiple_cmds(self):
with pytest.raises(UnexpectedSubpError) as excinfo:
subp.subp(["some", "args"])
assert "allowed: whoami,sh" in str(excinfo.value)
subp.subp(["sh", "-c", "true"])
subp.subp(["whoami"])
@pytest.mark.allow_all_subp
@pytest.mark.allow_subp_for("sh")
def test_both_marks_raise_an_error(self):
with pytest.raises(UnexpectedSubpError, match="marked both"):
subp.subp(["sh"])
|
