stack/tests/sem/test_state.py

"""Tests for sem.state — rolling state persistence."""

from __future__ import annotations

from sem.nodes import NodeKind, SemanticNode, Span
from sem.state import (
    apply_state,
    load_state,
    record_attempt,
    save_state,
    sync_state,
)


def _node(nid: str, src_hash: str = "abc123") -> SemanticNode:
    return SemanticNode(
        node_id=nid,
        kind=NodeKind.FUNCTION,
        span=Span(start_line=1, end_line=5),
        module="mod",
        source_hash=src_hash,
    )


class TestPersistence:
    def test_round_trip(self, tmp_path):
        p = tmp_path / "state.json"
        state = {"node_a": {"status": "covered", "attempts": 1}}
        save_state(state, p)
        loaded = load_state(p)
        assert loaded["node_a"]["status"] == "covered"

    def test_load_missing_file(self, tmp_path):
        assert load_state(tmp_path / "nope.json") == {}


class TestSync:
    def test_new_node_gets_fresh_entry(self):
        nodes = [_node("n1")]
        result = sync_state(nodes, {})
        assert "n1" in result
        assert result["n1"]["status"] == "unexplored"

    def test_unchanged_node_keeps_history(self):
        nodes = [_node("n1", "abc")]
        old = {"n1": {"status": "covered", "attempts": 3, "source_hash": "abc"}}
        result = sync_state(nodes, old)
        assert result["n1"]["status"] == "covered"
        assert result["n1"]["attempts"] == 3

    def test_changed_hash_resets_status(self):
        nodes = [_node("n1", "new_hash")]
        old = {"n1": {"status": "covered", "attempts": 3, "source_hash": "old_hash"}}
        result = sync_state(nodes, old)
        assert result["n1"]["status"] == "unexplored"
        assert result["n1"]["attempts"] == 0

    def test_stale_entries_removed(self):
        nodes = [_node("n1")]
        old = {"n1": {"source_hash": "abc123"}, "n_gone": {"source_hash": "x"}}
        result = sync_state(nodes, old)
        assert "n_gone" not in result


class TestApplyState:
    def test_pushes_state_to_nodes(self):
        node = _node("n1")
        state = {"n1": {"status": "partial", "attempts": 2}}
        apply_state([node], state)
        assert node.status == "partial"
        assert node.attempts == 2


class TestRecordAttempt:
    def test_increments_attempts(self):
        state = {"n1": {"status": "unexplored", "attempts": 0, "notes": []}}
        record_attempt(state, "n1", "partial", "needs None input")
        assert state["n1"]["attempts"] == 1
        assert state["n1"]["last_result"] == "partial"
        assert "needs None input" in state["n1"]["notes"]

    def test_covered_sets_status(self):
        state = {"n1": {"status": "unexplored", "attempts": 0, "notes": []}}
        record_attempt(state, "n1", "covered")
        assert state["n1"]["status"] == "covered"

    def test_new_node_creates_entry(self):
        state = {}
        record_attempt(state, "new", "partial")
        assert "new" in state
        assert state["new"]["attempts"] == 1