summaryrefslogtreecommitdiff
path: root/tests/test_report.py
blob: 1e7c07624864a5daad5a684af44fa04210e4bda9 (plain)
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
# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
# For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt

"""Tests for helpers in report.py"""

import pytest

from coverage.exceptions import CoverageException
from coverage.report import render_report
from tests.coveragetest import CoverageTest


class FakeReporter:
    """A fake implementation of a one-file reporter."""

    report_type = "fake report file"

    def __init__(self, output="", error=False):
        self.output = output
        self.error = error
        self.morfs = None

    def report(self, morfs, outfile):
        """Fake."""
        self.morfs = morfs
        outfile.write(self.output)
        if self.error:
            raise CoverageException("You asked for it!")


class RenderReportTest(CoverageTest):
    """Tests of render_report."""

    def test_stdout(self):
        fake = FakeReporter(output="Hello!\n")
        msgs = []
        render_report("-", fake, [pytest, "coverage"], msgs.append)
        assert fake.morfs == [pytest, "coverage"]
        assert self.stdout() == "Hello!\n"
        assert not msgs

    def test_file(self):
        fake = FakeReporter(output="Gréètings!\n")
        msgs = []
        render_report("output.txt", fake, [], msgs.append)
        assert self.stdout() == ""
        with open("output.txt", "rb") as f:
            assert f.read().rstrip() == b"Gr\xc3\xa9\xc3\xa8tings!"
        assert msgs == ["Wrote fake report file to output.txt"]

    def test_exception(self):
        fake = FakeReporter(error=True)
        msgs = []
        with pytest.raises(CoverageException, match="You asked for it!"):
            render_report("output.txt", fake, [], msgs.append)
        assert self.stdout() == ""
        self.assert_doesnt_exist("output.txt")
        assert not msgs