mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""Tests for TerminalBenchTaskEnv (mocked terminal_bench dependency)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from openjarvis.evals.execution.terminalbench_env import TerminalBenchTaskEnv
|
|
|
|
# terminal_bench is an optional dep — skip all tests if unavailable
|
|
terminal_bench = pytest.importorskip(
|
|
"terminal_bench", reason="terminal_bench not installed"
|
|
)
|
|
|
|
|
|
class TestTerminalBenchTaskEnv:
|
|
def test_init(self):
|
|
metadata = {"task_id": "test-1"}
|
|
env = TerminalBenchTaskEnv(metadata)
|
|
assert env._metadata is metadata
|
|
assert env._terminal is None
|
|
|
|
def test_enter_without_task_raises(self):
|
|
metadata = {"task_id": "test-1"}
|
|
env = TerminalBenchTaskEnv(metadata)
|
|
with pytest.raises(ValueError, match="Task metadata missing"):
|
|
env.__enter__()
|
|
|
|
def test_exit_cleans_metadata(self):
|
|
metadata = {
|
|
"task_id": "test-1",
|
|
"terminal": "fake_terminal",
|
|
"session": "fake_session",
|
|
"container": "fake_container",
|
|
}
|
|
env = TerminalBenchTaskEnv(metadata)
|
|
env.__exit__(None, None, None)
|
|
assert "terminal" not in metadata
|
|
assert "session" not in metadata
|
|
assert "container" not in metadata
|
|
|
|
def test_run_tests_without_terminal(self):
|
|
metadata = {"task": "mock_task", "task_paths": "mock_paths"}
|
|
env = TerminalBenchTaskEnv(metadata)
|
|
env._terminal = None
|
|
is_resolved, results = env.run_tests()
|
|
assert is_resolved is False
|
|
assert results["error"] == "terminal_not_running"
|
|
assert metadata["is_resolved"] is False
|