feat: streamed results

neotest_python/__init__.py

@@ -3,6 +3,8 @@ import json
 from enum import Enum
 from typing import List
 
+from neotest_python.base import NeotestResult
+
 
 class TestRunner(str, Enum):
     PYTEST = "pytest"
@@ -29,12 +31,24 @@ parser.add_argument(
     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))
-    results = adapter.run(args.args)
+    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)