maelstrom-pytest 0.14.0

Python Test Runner for Maelstrom.
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

#!/usr/bin/env python
import json
import os
import pytest
import sys

from _pytest.nodes import Node as PytestNode
from contextlib import redirect_stdout
from dataclasses import asdict, dataclass
from io import StringIO
from typing import Optional, Sequence, Tuple, List


class Plugin:
    def __init__(self) -> None:
        self.items: List[pytest.Item] = []

    def pytest_collection_modifyitems(
        self, session: pytest.Session, config: pytest.Config, items: List[pytest.Item]
    ) -> None:
        self.items = items


def collect_pytest_tests(args: List[str]) -> List[pytest.Item]:
    plugin = Plugin()
    output = StringIO()
    with redirect_stdout(output):
        ret = pytest.main(args=["--co"] + args, plugins=[plugin])
    if ret != 0:
        output.seek(0)
        sys.stderr.write(output.read())
        os._exit(ret)

    return plugin.items


@dataclass(frozen=True)
class TestCase:
    file: str
    name: str
    node_id: str
    markers: Sequence[str]
    skip: bool


def is_skip(marker: pytest.Mark) -> bool:
    if marker.name == "skip":
        return True
    elif marker.name == "skipif":
        return marker.args[0]
    else:
        return False


def main() -> None:
    args = sys.argv[1:]

    tests = collect_pytest_tests(args)
    for item in tests:
        (raw_file, _, name) = item.reportinfo()
        file = str(raw_file)
        markers = [m.name for m in item.own_markers]

        skip = any(is_skip(m) for m in item.own_markers)

        case = TestCase(
            file=file, name=name, node_id=item.nodeid, markers=markers, skip=skip
        )
        print(json.dumps(asdict(case)))


if __name__ == "__main__":
    main()