mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-10-28 00:53:45 +08:00
firestar5683
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""Explicit pytest plugin retaining identities/outcomes when a host job is interrupted."""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from tools.ci import worker_diagnostics
|
|
|
|
|
|
def pytest_sessionstart(session):
|
|
worker_diagnostics.initialize()
|
|
|
|
|
|
def pytest_collection_finish(session):
|
|
directory = os.environ.get('STARPILOT_TEST_DIAGNOSTICS')
|
|
if directory:
|
|
root = Path(directory).resolve()
|
|
root.mkdir(parents=True, exist_ok=True)
|
|
(root / 'pytest-collection.json').write_text(json.dumps([item.nodeid for item in session.items]) + '\n')
|
|
|
|
|
|
def pytest_runtest_logstart(nodeid, location):
|
|
worker_diagnostics.node_started(nodeid)
|
|
|
|
|
|
def pytest_runtest_logreport(report):
|
|
row = {'id': report.nodeid, 'phase': report.when, 'status': report.outcome, 'duration': report.duration}
|
|
if report.failed:
|
|
row['failure'] = str(report.longrepr)[:4096]
|
|
worker_diagnostics.event('pytest_report', **row)
|
|
if report.when == 'teardown':
|
|
worker_diagnostics.update(active_test=None, phase='between_tests')
|
|
|
|
|
|
def pytest_sessionfinish(session, exitstatus):
|
|
worker_diagnostics.update(active_test=None, phase='session_complete', exitstatus=int(exitstatus))
|