2dc9c95fe9
Having `NeotestResultCollector` (a pytest plugin) as a inner local class would make the code a bit difficult to read due to quite much indentation. This commit does refactoring on NeotestResultCollector to make it a module-level class with a reference to NeotestAdapter. This refactoring would make easier adding more pytest plugins (e.g., debugger integration) in the future. There should be no changes in behaviors.
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
import argparse
|
|
import json
|
|
from enum import Enum
|
|
from typing import List
|
|
|
|
from neotest_python.base import NeotestAdapter, NeotestResult
|
|
|
|
|
|
class TestRunner(str, Enum):
|
|
PYTEST = "pytest"
|
|
UNITTEST = "unittest"
|
|
|
|
|
|
def get_adapter(runner: TestRunner) -> NeotestAdapter:
|
|
if runner == TestRunner.PYTEST:
|
|
from .pytest import PytestNeotestAdapter
|
|
|
|
return PytestNeotestAdapter()
|
|
elif runner == TestRunner.UNITTEST:
|
|
from .unittest import UnittestNeotestAdapter
|
|
|
|
return UnittestNeotestAdapter()
|
|
raise NotImplementedError(runner)
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--runner", required=True)
|
|
parser.add_argument(
|
|
"--results-file",
|
|
dest="results_file",
|
|
required=True,
|
|
help="File to store result JSON in",
|
|
)
|
|
parser.add_argument(
|
|
"--stream-file",
|
|
dest="stream_file",
|
|
required=True,
|
|
help="File to stream result JSON to",
|
|
)
|
|
parser.add_argument("args", nargs="*")
|
|
|
|
|
|
def main(argv: List[str]):
|
|
args = parser.parse_args(argv)
|
|
adapter = get_adapter(TestRunner(args.runner))
|
|
|
|
with open(args.stream_file, "w") as stream_file:
|
|
|
|
def stream(pos_id: str, result: NeotestResult):
|
|
stream_file.write(json.dumps({"id": pos_id, "result": result}) + "\n")
|
|
stream_file.flush()
|
|
|
|
results = adapter.run(args.args, stream)
|
|
|
|
with open(args.results_file, "w") as results_file:
|
|
json.dump(results, results_file)
|