fix(core): preserve adapter command contracts
This commit is contained in:
@@ -8,11 +8,8 @@ pgvector as a one-row upsert. This test pins the pure core of that behavior:
|
||||
- writer.sync is NEVER called (that is the full-resync path)
|
||||
"""
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import pytest
|
||||
|
||||
from tht.memory import MemoryRecord, memory_vector_record_for_decision, save_one_memory
|
||||
|
||||
|
||||
@@ -47,31 +44,31 @@ def test_returns_none_for_unknown_decision_seq():
|
||||
def test_save_one_calls_upsert_with_single_row_never_sync():
|
||||
records = [_record(seq=7)]
|
||||
writer = MagicMock()
|
||||
writer.upsert_records.return_value = 1
|
||||
writer.upsert.return_value = 1
|
||||
embedder = MagicMock()
|
||||
embedder.embed_documents.return_value = [[0.1] * 8]
|
||||
|
||||
upserted = save_one_memory(records, decision_seq=7, writer=writer, embedder=embedder)
|
||||
upserted = save_one_memory(records, decision_seq=7, store=writer, embedder=embedder)
|
||||
|
||||
assert upserted == 1
|
||||
writer.sync.assert_not_called() # the whole point of D11: no full resync
|
||||
writer.upsert_records.assert_called_once()
|
||||
args = writer.upsert_records.call_args
|
||||
writer.upsert.assert_called_once()
|
||||
args = writer.upsert.call_args
|
||||
# table is memory, exactly one row
|
||||
assert args[0][0] == "memory"
|
||||
rows = args[0][1]
|
||||
assert len(rows) == 1
|
||||
assert rows[0]["record_key"] == "memory:mem-0007"
|
||||
assert "embedding" in rows[0]
|
||||
assert rows[0].record.id == "memory:mem-0007"
|
||||
assert rows[0].embedding
|
||||
|
||||
|
||||
def test_save_one_no_record_for_seq_is_noop():
|
||||
records = [_record(seq=7)]
|
||||
writer = MagicMock()
|
||||
embedder = MagicMock()
|
||||
upserted = save_one_memory(records, decision_seq=42, writer=writer, embedder=embedder)
|
||||
upserted = save_one_memory(records, decision_seq=42, store=writer, embedder=embedder)
|
||||
assert upserted == 0
|
||||
writer.upsert_records.assert_not_called()
|
||||
writer.upsert.assert_not_called()
|
||||
writer.sync.assert_not_called()
|
||||
embedder.embed_documents.assert_not_called()
|
||||
|
||||
@@ -81,9 +78,9 @@ def test_save_one_uses_writer_key_for_upsert():
|
||||
Verified indirectly: save_one_memory takes the writer as its client argument."""
|
||||
records = [_record(seq=7)]
|
||||
writer = MagicMock()
|
||||
writer.upsert_records.return_value = 1
|
||||
writer.upsert.return_value = 1
|
||||
embedder = MagicMock()
|
||||
embedder.embed_documents.return_value = [[0.0] * 4]
|
||||
save_one_memory(records, decision_seq=7, writer=writer, embedder=embedder)
|
||||
save_one_memory(records, decision_seq=7, store=writer, embedder=embedder)
|
||||
# one upsert call, single row, table=memory
|
||||
assert writer.upsert_records.call_count == 1
|
||||
assert writer.upsert.call_count == 1
|
||||
|
||||
Reference in New Issue
Block a user