| import pytest |
|
|
| from tiny_computer_agent import ( |
| Action, Element, Observation, Policy, PolicyViolation, |
| ) |
|
|
| FRAME = Observation(elements=(Element("Terminal"), Element("Save"))) |
|
|
|
|
| def test_allows_an_ordinary_click(): |
| Policy().check(Action("click", label="Save"), FRAME) |
|
|
|
|
| def test_refuses_a_denied_key_chord(): |
| with pytest.raises(PolicyViolation, match="deny list"): |
| Policy().check(Action("key", keys="ctrl+alt+t")) |
|
|
|
|
| @pytest.mark.parametrize("text", ["sudo apt install foo", "rm -rf /", "my password is hunter2"]) |
| def test_refuses_denied_text(text): |
| with pytest.raises(PolicyViolation, match="deny list"): |
| Policy().check(Action("type", text=text)) |
|
|
|
|
| def test_refuses_clicking_a_denied_widget(): |
| with pytest.raises(PolicyViolation, match="Terminal"): |
| Policy().check(Action("click", label="Terminal"), FRAME) |
|
|
|
|
| def test_deny_label_is_case_insensitive(): |
| with pytest.raises(PolicyViolation): |
| Policy().check(Action("click", label="terminal"), FRAME) |
|
|
|
|
| def test_kill_switch_is_off_until_the_file_exists(tmp_path): |
| marker = tmp_path / "STOP" |
| policy = Policy(kill_switch_path=str(marker)) |
| assert policy.should_stop() is False |
| marker.write_text("stop") |
| assert policy.should_stop() is True |
|
|
|
|
| def test_no_kill_switch_path_means_never_stops(): |
| assert Policy().should_stop() is False |
|
|
|
|
| def test_rejects_a_useless_budget(): |
| with pytest.raises(ValueError, match="step_budget"): |
| Policy(step_budget=0) |
|
|
|
|
| def test_deny_lists_are_configurable(): |
| lax = Policy(deny_text_patterns=(), deny_labels=(), deny_keys=()) |
| lax.check(Action("type", text="sudo rm -rf /")) |
| lax.check(Action("click", label="Terminal"), FRAME) |
|
|
