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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
import tempfile
from functools import partial
from unittest.mock import patch
import pytest
from snakeoil.formatters import PlainTextFormatter
from pkgcheck import __title__ as project
from pkgcheck.checks.profiles import ProfileWarning
from pkgcheck.reporters import JsonStream
from pkgcheck.scripts import run
class TestPkgcheckReplay:
script = staticmethod(partial(run, project))
@pytest.fixture(autouse=True)
def _setup(self, testconfig):
self.args = [project, "--config", testconfig, "replay"]
def test_missing_file_arg(self, capsys):
with patch("sys.argv", self.args):
with pytest.raises(SystemExit) as excinfo:
self.script()
out, err = capsys.readouterr()
assert not out
err = err.strip().split("\n")
assert len(err) == 1
assert err[0] == ("pkgcheck replay: error: the following arguments are required: FILE")
assert excinfo.value.code == 2
def test_replay(self, capsys):
result = ProfileWarning("profile warning: foo")
with tempfile.NamedTemporaryFile() as file:
out = PlainTextFormatter(file)
with JsonStream(out) as reporter:
reporter.report(result)
with patch("sys.argv", self.args + ["-R", "StrReporter", file.name]):
with pytest.raises(SystemExit) as excinfo:
self.script()
out, err = capsys.readouterr()
assert not err
assert out == "profile warning: foo\n"
assert excinfo.value.code == 0
def test_corrupted_resuts(self, capsys):
result = ProfileWarning("profile warning: foo")
with tempfile.NamedTemporaryFile() as file:
out = PlainTextFormatter(file)
with JsonStream(out) as reporter:
reporter.report(result)
file.write(b"corrupted")
file.seek(0)
with patch("sys.argv", self.args + ["-R", "StrReporter", file.name]):
with pytest.raises(SystemExit) as excinfo:
self.script()
out, err = capsys.readouterr()
assert "corrupted results file" in err
assert excinfo.value.code == 2
def test_invalid_file(self, capsys):
with tempfile.NamedTemporaryFile(mode="wt") as file:
file.write("invalid file")
file.seek(0)
with patch("sys.argv", self.args + ["-R", "StrReporter", file.name]):
with pytest.raises(SystemExit) as excinfo:
self.script()
out, err = capsys.readouterr()
assert err.strip() == "pkgcheck replay: error: invalid or unsupported replay file"
assert excinfo.value.code == 2
def test_replay_pipe_stdin(self, capsys):
with tempfile.NamedTemporaryFile() as file:
out = PlainTextFormatter(file)
with JsonStream(out) as reporter:
reporter.report(ProfileWarning("profile warning: foo"))
file.seek(0)
with (
open(file.name) as stdin,
patch("sys.stdin", stdin),
patch("sys.argv", [*self.args, "-R", "StrReporter", "-"]),
pytest.raises(SystemExit) as excinfo,
):
self.script()
out, err = capsys.readouterr()
assert not err
assert out == "profile warning: foo\n"
assert excinfo.value.code == 0
|