WS2

Author: tzanko-matev

Justfile

@@ -67,13 +67,13 @@ coverage-python:
 
 # Build the module in release mode
 build:
-    just venv \
-    uv run --directory codetracer-python-recorder maturin build --release
+    just venv
+    uv run --directory codetracer-python-recorder maturin build --release --sdist
 
 # Build wheels for all target Python versions with maturin
 build-all:
     just venv
-    uv run --directory codetracer-python-recorder maturin build --release --interpreter {{PY_VERSIONS}}
+    uv run --directory codetracer-python-recorder maturin build --release --sdist --interpreter {{PY_VERSIONS}}
 
 # Smoke the built Rust wheels across versions using uv
 test-all:
@@ -82,3 +82,18 @@ test-all:
         file="${file[0]}"; \
         uv run -p "python3.$v" --with "${file}" --with pytest -- pytest -q; \
     done
+
+# Install a freshly built artifact and run a CLI smoke test
+smoke-wheel artifact="wheel" interpreter=".venv/bin/python":
+    just build
+    VENV_DIR="$(mktemp -d)"; \
+    trap 'rm -rf "$VENV_DIR"' EXIT; \
+    "{{interpreter}}" -m venv "$VENV_DIR"; \
+    VENV_PY="$VENV_DIR/bin/python"; \
+    if [ ! -x "$VENV_PY" ]; then \
+        VENV_PY="$VENV_DIR/Scripts/python.exe"; \
+    fi; \
+    "$VENV_PY" -m pip install --upgrade pip; \
+    FILE="$("$VENV_PY" scripts/select_recorder_artifact.py --wheel-dir codetracer-python-recorder/target/wheels --mode "{{artifact}}")"; \
+    "$VENV_PY" -m pip install "$FILE"; \
+    "$VENV_PY" -m codetracer_python_recorder --help >/dev/null

codetracer-python-recorder/tests/python/test_codetracer_api.py

@@ -33,6 +33,22 @@ def test_context_manager(self) -> None:
                 self.assertIsInstance(session, codetracer.TraceSession)
             self.assertFalse(codetracer.is_tracing())
 
+    def test_start_emits_trace_files(self) -> None:
+        with tempfile.TemporaryDirectory() as tmpdir:
+            trace_dir = Path(tmpdir)
+            codetracer.start(trace_dir)
+            # Execute a small workload to ensure callbacks fire.
+            def _workload() -> int:
+                return sum(range(5))
+
+            self.assertEqual(_workload(), 10)
+            codetracer.stop()
+
+            metadata = trace_dir / "trace_metadata.json"
+            paths = trace_dir / "trace_paths.json"
+            self.assertTrue(metadata.exists(), "expected trace_metadata.json to be created")
+            self.assertTrue(paths.exists(), "expected trace_paths.json to be created")
+
     def test_environment_auto_start(self) -> None:
         script = "import codetracer_python_recorder as codetracer, sys; sys.stdout.write(str(codetracer.is_tracing()))"
         with tempfile.TemporaryDirectory() as tmpdir:

design-docs/codetracer-python-recorder-pypi-release-implementation-plan.status.md

@@ -6,6 +6,12 @@
 - ✅ Introduced `scripts/check_recorder_version.py` and wired it into CI to enforce version parity.
 - ✅ Documented supported environments in `codetracer-python-recorder/README.md` and added a release checklist under `design-docs/`.
 
+## Workstream 2 – Build & test enhancements
+- ✅ Updated `Justfile` build targets to emit wheels and sdists and added a `smoke-wheel` recipe that installs artefacts via an isolated venv.
+- ✅ Added `scripts/select_recorder_artifact.py` to choose interpreter-compatible wheels and wired the smoke test to use it.
+- ✅ Expanded the Python API tests to cover end-to-end tracing via `start`/`stop` (`test_start_emits_trace_files`).
+- ✅ Verified `just py-test` passes with the new coverage.
+
 ## Next Tasks
-- Begin Workstream 2: enhance build/test automation (extend Just recipes, add smoke install target, ensure integration coverage).
-- Draft changes to the release workflow (`recorder-release.yml`) once Workstream 2 groundwork is ready.
+- Kick off Workstream 3: design the cross-platform release workflow (`recorder-release.yml`) and wire in TestPyPI publishing.
+- Add release-pipeline steps to invoke the new smoke install target once the workflow skeleton exists.

scripts/select_recorder_artifact.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+"""Select the appropriate recorder build artifact for a given interpreter."""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument(
+        "--wheel-dir",
+        type=Path,
+        required=True,
+        help="Directory containing maturin build outputs (wheels and/or sdists).",
+    )
+    parser.add_argument(
+        "--mode",
+        choices=("wheel", "sdist"),
+        default="wheel",
+        help="Prefer a wheel for the current interpreter or fallback to the source distribution.",
+    )
+    return parser.parse_args()
+
+
+def choose_artifact(wheel_dir: Path, mode: str) -> Path:
+    candidates = []
+    if mode == "wheel":
+        abi = f"cp{sys.version_info.major}{sys.version_info.minor}"
+        pattern = f"codetracer_python_recorder-*-{abi}-{abi}-*.whl"
+        candidates = sorted(wheel_dir.glob(pattern))
+    if not candidates:
+        candidates = sorted(wheel_dir.glob("codetracer_python_recorder-*.tar.gz"))
+    if not candidates:
+        raise FileNotFoundError(f"No build artefacts found in {wheel_dir}")
+    return candidates[-1]
+
+
+def main() -> int:
+    args = parse_args()
+    artifact = choose_artifact(args.wheel_dir, args.mode)
+    sys.stdout.write(str(artifact))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())