from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import pytest  # type: ignore

from .coverage import Coverage
from typing import Dict, Text, Sequence, Any

_coverage = Coverage()
_marks = {}  # type: Dict[Text, Sequence[Any]]


def _add_mark(mark, bucket):  # type: (Any, Text) -> None
    proto = mark.args[0]
    if isinstance(proto, list):
        assert len(proto) == 1
        proto = proto[0]
    if proto is not None:
        _coverage.add_proto(proto, bucket)


def pytest_runtest_call(item):  # type: (pytest.nodes.Item) -> None
    mark = item.get_marker('onnx_coverage')
    if mark:
        assert item.nodeid not in _marks
        _marks[item.nodeid] = mark


def pytest_runtest_logreport(report):  # type: (Any) -> None
    if (report.when == 'call' and
        report.outcome == 'passed' and
            report.nodeid in _marks):
        mark = _marks[report.nodeid]
        _add_mark(mark, 'passed')


@pytest.hookimpl(trylast=True)  # type: ignore
def pytest_terminal_summary(terminalreporter, exitstatus):  # type: (pytest.terminal.TerminalReporter, int) -> None
    for mark in _marks.values():
        _add_mark(mark, 'loaded')
    _coverage.report_text(terminalreporter)
