mirror of
https://github.com/BigBodyCobain/Shadowbroker.git
synced 2026-08-16 15:40:26 +02:00
fix(store): preserve non-mapping native layer entries (#508)
* fix(store): preserve non-mapping native layer entries * test(store): cover mixed native injection cleanup
This commit is contained in:
@@ -141,7 +141,11 @@ def inject_layer_data(
|
||||
existing = list(current) if isinstance(current, list) else []
|
||||
|
||||
if mode == "replace":
|
||||
existing = [e for e in existing if not e.get("_injected")]
|
||||
existing = [
|
||||
entry
|
||||
for entry in existing
|
||||
if not (isinstance(entry, dict) and entry.get("_injected"))
|
||||
]
|
||||
|
||||
# Readers can hold references to published layer lists after releasing
|
||||
# _data_lock. Build a fresh list and swap it atomically rather than
|
||||
@@ -170,7 +174,11 @@ def clear_injected_data(layer: str = "") -> dict[str, Any]:
|
||||
if not isinstance(existing, list):
|
||||
continue
|
||||
before = len(existing)
|
||||
latest_data[lyr] = [e for e in existing if not e.get("_injected")]
|
||||
latest_data[lyr] = [
|
||||
entry
|
||||
for entry in existing
|
||||
if not (isinstance(entry, dict) and entry.get("_injected"))
|
||||
]
|
||||
removed += before - len(latest_data[lyr])
|
||||
|
||||
if removed:
|
||||
|
||||
@@ -0,0 +1,46 @@
|
||||
"""Regression coverage for mixed native entries during injected-data cleanup."""
|
||||
|
||||
from services import ai_intel_store
|
||||
from services.fetchers import _store
|
||||
|
||||
|
||||
def _publish_mixed_layer(monkeypatch):
|
||||
published = [
|
||||
"native-sentinel",
|
||||
{"id": "native"},
|
||||
{"id": "old-injected", "_injected": True},
|
||||
]
|
||||
monkeypatch.setitem(_store.latest_data, "air_quality", published)
|
||||
monkeypatch.setattr(_store, "bump_data_version", lambda: None)
|
||||
return published
|
||||
|
||||
|
||||
def test_replace_preserves_non_mapping_native_entries(monkeypatch):
|
||||
before = _publish_mixed_layer(monkeypatch)
|
||||
|
||||
result = ai_intel_store.inject_layer_data(
|
||||
"air_quality",
|
||||
[{"id": "new-injected"}],
|
||||
mode="replace",
|
||||
)
|
||||
|
||||
after = _store.latest_data["air_quality"]
|
||||
assert result["ok"] is True
|
||||
assert after is not before
|
||||
assert after[0] == "native-sentinel"
|
||||
assert [item.get("id") for item in after if isinstance(item, dict)] == [
|
||||
"native",
|
||||
"new-injected",
|
||||
]
|
||||
|
||||
|
||||
def test_clear_preserves_non_mapping_native_entries(monkeypatch):
|
||||
_publish_mixed_layer(monkeypatch)
|
||||
|
||||
result = ai_intel_store.clear_injected_data("air_quality")
|
||||
|
||||
assert result == {"ok": True, "removed": 1}
|
||||
assert _store.latest_data["air_quality"] == [
|
||||
"native-sentinel",
|
||||
{"id": "native"},
|
||||
]
|
||||
Reference in New Issue
Block a user