feat(memory): add incremental transcript compaction

tests/unit/data_engine/test_step_summary_versioning.py

@@ -0,0 +1,171 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tests for the versioned, status-carrying step summary write (M0 §6.1)."""
+
+from unittest.mock import MagicMock
+
+from artemis.context import ArtemisContext
+from artemis.data_engine.engine import DataEngine
+
+
+def _make_engine(tmp_path):
+    mock_ctx = MagicMock(spec=ArtemisContext)
+    mock_execution_setup = MagicMock()
+    mock_execution_setup.traces_path = str(tmp_path)
+    mock_ctx.execution_setup = mock_execution_setup
+    mock_ctx.device = None
+    engine = DataEngine(mock_ctx)
+    engine.start_session("summary versioning test")
+    return engine
+
+
+def _flush(engine):
+    for t in list(engine._pending_threads):
+        t.join()
+
+
+def _get_step(engine, step_id):
+    steps = engine.storage.get_steps(engine.current_session_id)
+    for step in steps:
+        if str(step.step_id) == str(step_id):
+            return step
+    return None
+
+
+def test_ready_write_persists_summary_and_metadata(tmp_path):
+    engine = _make_engine(tmp_path)
+    step_id = engine.record_step(summary="initial")
+    _flush(engine)
+
+    applied = engine.storage.update_step_summary(
+        step_id,
+        "Tapped the login button.",
+        source="visual_transition",
+        model="gemini-2.5-flash-lite",
+        status="ready",
+    )
+    assert applied is True
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "Tapped the login button."
+    assert step.extra_metadata["summary_status"] == "ready"
+    assert step.extra_metadata["summary_source"] == "visual_transition"
+    assert step.extra_metadata["summary_version"] == 1
+    assert step.extra_metadata["summary_model"] == "gemini-2.5-flash-lite"
+
+
+def test_auto_version_increments_per_write(tmp_path):
+    engine = _make_engine(tmp_path)
+    step_id = engine.record_step(summary="initial")
+    _flush(engine)
+
+    engine.storage.update_step_summary(step_id, "v1", status="ready")
+    engine.storage.update_step_summary(step_id, "v2", status="ready")
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "v2"
+    assert step.extra_metadata["summary_version"] == 2
+
+
+def test_stale_explicit_version_is_dropped(tmp_path):
+    engine = _make_engine(tmp_path)
+    step_id = engine.record_step(summary="initial")
+    _flush(engine)
+
+    assert engine.storage.update_step_summary(step_id, "newer", version=5, status="ready")
+    applied = engine.storage.update_step_summary(step_id, "older", version=3, status="ready")
+    assert applied is False
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "newer"
+    assert step.extra_metadata["summary_version"] == 5
+
+
+def test_pending_does_not_downgrade_ready(tmp_path):
+    engine = _make_engine(tmp_path)
+    step_id = engine.record_step(summary="initial")
+    _flush(engine)
+
+    engine.storage.update_step_summary(step_id, "done", status="ready")
+    applied = engine.storage.update_step_summary(step_id, None, status="pending")
+    assert applied is False
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "done"
+    assert step.extra_metadata["summary_status"] == "ready"
+
+
+def test_failed_marks_status_without_touching_summary(tmp_path):
+    engine = _make_engine(tmp_path)
+    step_id = engine.record_step(summary="original text")
+    _flush(engine)
+
+    engine.storage.update_step_summary(step_id, None, status="pending", source="visual_transition")
+    applied = engine.storage.update_step_summary(
+        step_id, None, status="failed", source="visual_transition"
+    )
+    assert applied is True
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "original text"
+    assert step.extra_metadata["summary_status"] == "failed"
+
+    # A failed write must never clobber an already-ready summary.
+    engine.storage.update_step_summary(step_id, "recovered", status="ready")
+    applied = engine.storage.update_step_summary(step_id, None, status="failed")
+    assert applied is False
+    step = _get_step(engine, step_id)
+    assert step.extra_metadata["summary_status"] == "ready"
+
+
+def test_missing_step_returns_false(tmp_path):
+    from uuid import uuid4
+
+    engine = _make_engine(tmp_path)
+    assert engine.storage.update_step_summary(uuid4(), "text", status="ready") is False
+
+
+def test_engine_level_write_carries_metadata_and_publishes_ready_only(tmp_path):
+    engine = _make_engine(tmp_path)
+    published = []
+    engine.subscribe(lambda event_type, data: published.append((event_type, data)))
+
+    step_id = engine.record_step(summary="initial")
+    _flush(engine)
+    published.clear()
+
+    # Non-ready statuses persist metadata but publish no SSE.
+    engine.update_step_summary(step_id, None, status="pending", source="visual_transition")
+    _flush(engine)
+    assert not [e for e in published if e[0] == "step_updated"]
+    step = _get_step(engine, step_id)
+    assert step.extra_metadata["summary_status"] == "pending"
+
+    engine.update_step_summary(
+        step_id,
+        "Ready text.",
+        status="ready",
+        source="visual_transition",
+        model="test-model",
+    )
+    _flush(engine)
+    ready_events = [e for e in published if e[0] == "step_updated"]
+    assert len(ready_events) == 1
+    assert ready_events[0][1]["summary"] == "Ready text."
+
+    step = _get_step(engine, step_id)
+    assert step.summary == "Ready text."
+    assert step.extra_metadata["summary_status"] == "ready"
+    assert step.extra_metadata["summary_model"] == "test-model"