diff --git a/tests/test_tier3_integration.py b/tests/test_tier3_integration.py index a5098c0..a00e548 100644 --- a/tests/test_tier3_integration.py +++ b/tests/test_tier3_integration.py @@ -35,8 +35,7 @@ def test_integration_create_options_combination(mam_sandbox, mock_herdr, mock_ag "--agent", "claude", "--role", "Creator", "--onboard", - "--submit-job", "Test onboard prompt", - "--wrapper" + "--submit-job", "Test onboard prompt" ] res = subprocess.run(cmd, capture_output=True, text=True, cwd=str(tmp_path)) @@ -111,7 +110,9 @@ def test_integration_stop_purge_combination(mam_sandbox, mock_herdr, mock_agents proj_dir = tmp_path / ".claude" / "projects" / key assert proj_dir.exists(), f"Expected projects directory {proj_dir} to exist" - jsonl_file = proj_dir / f"{claude_uuid}.jsonl" + jsonls = list(proj_dir.glob("*.jsonl")) + assert len(jsonls) == 1, f"Expected exactly 1 jsonl file in {proj_dir}, found {jsonls}" + jsonl_file = jsonls[0] assert jsonl_file.exists() # Update mock herdr's state to match the running agent so the stop kill-chain works gracefully